Files
bildverarbeitungss18-hausau…/assignments/01_assignment_sample_solution.ipynb
2018-05-02 18:30:24 +02:00

423 lines
11 KiB
Plaintext

{
"cells": [
{
"cell_type": "markdown",
"metadata": {},
"source": [
"# Image Processing SS 16 - Assignment - 01\n",
"\n",
"### Deadline is 27.4.2016 at 16:00\n",
"\n",
"Please solve the assignments together with a partner.\n",
"I will run every notebook. Make sure the code runs through, when clicked on `Kernel` -> `Restart & Run All`.\n"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"## Introduction to Python / Numpy\n",
"\n",
"* [Learn Python in 15 minutes](https://learnxinyminutes.com/docs/python/)\n",
"* [Numpy for Matlab Users](https://docs.scipy.org/doc/numpy-dev/user/numpy-for-matlab-users.html#general-purpose-equivalents)\n",
"* [Numpy Quickstart](https://docs.scipy.org/doc/numpy-dev/user/quickstart.html)\n",
"\n",
"## Libraries\n",
"\n",
"We will use the following libraries:\n",
"\n",
"* matplotlib\n",
"* numpy\n",
"* scipy\n",
"* skimage\n"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"# Exercise 0 - Setup Development Enviroment - [1 Point]\n",
"\n",
"Find a partner, follow the steps in the [README](https://github.com/) and paste a link to your repository, names and matriculation numbers into the KVV assignment box.\n",
"You do not need to upload any files to the KVV. I will clone your repository. "
]
},
{
"cell_type": "code",
"execution_count": 1,
"metadata": {},
"outputs": [],
"source": [
"# display the plots inside the notebook\n",
"%matplotlib inline"
]
},
{
"cell_type": "code",
"execution_count": 2,
"metadata": {},
"outputs": [],
"source": [
"import numpy as np\n",
"import matplotlib.pyplot as plt\n",
"import pylab\n",
"import copy\n",
"pylab.rcParams['figure.figsize'] = (12, 12) # This makes the plot bigger"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"The [skimage](http://scikit-image.org/) library comes with multiple useful test images. Let's start with an image of an astronaut. "
]
},
{
"cell_type": "code",
"execution_count": 3,
"metadata": {},
"outputs": [],
"source": [
"from skimage.data import astronaut"
]
},
{
"cell_type": "code",
"execution_count": 4,
"metadata": {},
"outputs": [],
"source": [
"img = astronaut() # Get the image\n",
"print(img.shape) # the dimension of the image\n",
"print(img.dtype) # the image type"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"We have a `(512, 512, 3)` array of unsigned bytes. At `img[x, y]` there are three values for R,G and B."
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"We will always work with floating point arrays between 0 and 1. "
]
},
{
"cell_type": "code",
"execution_count": 5,
"metadata": {},
"outputs": [],
"source": [
"img = img / 255."
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"Lets display the image."
]
},
{
"cell_type": "code",
"execution_count": 6,
"metadata": {},
"outputs": [],
"source": [
"plt.imshow(img)\n",
"plt.show()"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"This is [Eileen Collins](https://en.wikipedia.org/wiki/Eileen_Collins). She was the first astronaut \n",
" to fly the Space Shuttle through a complete 360-degree pitch maneuver. What an inspiring woman."
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"## Exercise 1 - Plot - [1 Point]\n",
"\n",
"Plot the R, G and B channels separately."
]
},
{
"cell_type": "code",
"execution_count": 7,
"metadata": {},
"outputs": [],
"source": [
"img_red = copy.deepcopy(img)\n",
"img_red[:,:,1] = 0\n",
"img_red[:,:,2] = 0\n",
"plt.imshow(img_red)\n",
"plt.show()\n",
"\n",
"img_green = copy.deepcopy(img);\n",
"img_green[:,:,0] = 0\n",
"img_green[:,:,2] = 0\n",
"plt.imshow(img_green)\n",
"plt.show()\n",
"\n",
"img_blue = copy.deepcopy(img);\n",
"img_blue[:,:,0] = 0\n",
"img_blue[:,:,1] = 0\n",
"plt.imshow(img_blue)\n",
"plt.show()"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"## Exercise 2 - RGB to HSV [6 Points]\n",
"\n",
"Implement the `rgb_to_hsv` and `hsv_to_rgb` functions. Don't use any color conversion functions from a library.\n"
]
},
{
"cell_type": "code",
"execution_count": 8,
"metadata": {
"collapsed": true
},
"outputs": [],
"source": [
"def rgb_to_hsv(x):\n",
" \"\"\"\n",
" Converts the numpy array `x` from RGB to the HSV. \n",
" \"\"\"\n",
" hsv=[]\n",
" for line in x:\n",
" newLine=[]\n",
" for pixel in line:\n",
" r= pixel[0]/255\n",
" g= pixel[1]/255\n",
" b= pixel[2]/255\n",
" cMax = np.amax([r,g,b])\n",
" cMin = min([r,g,b])\n",
" delta = cMax-cMin\n",
" v = cMax\n",
" h=0\n",
" s=0\n",
" if v > 0:\n",
" s = delta/cMax\n",
" if s>0 :\n",
" if r == cMax:\n",
" h= ((g-b)/delta)%6\n",
" elif g==cMax:\n",
" h= ((b-r)/delta)+2\n",
" elif b==cMax:\n",
" h=((r-g)/delta)+4\n",
" h=h*60\n",
" if h<0:\n",
" h=h+360\n",
" newLine.append([h,s,v])\n",
" hsv.append(newLine)\n",
" return hsv\n",
" "
]
},
{
"cell_type": "code",
"execution_count": 9,
"metadata": {
"collapsed": true
},
"outputs": [],
"source": [
"def hsv_to_rgb(x):\n",
" \"\"\"\n",
" Converts the numpy array `x` from HSV to the RGB. \n",
" \"\"\"\n",
" rgb = []\n",
" for line in x:\n",
" newLine=[]\n",
" for pixel in line:\n",
" h= pixel[0]\n",
" s= pixel[1]\n",
" v= pixel[2]\n",
" c = v*s\n",
" x=c*(1-abs(((h/60)%2)-1))\n",
" m= v-c\n",
" tmpPixel= [0,0,0]\n",
" if h >= 0 and h <60:\n",
" tmpPixel=[c,x,0]\n",
" elif h >= 60 and h <120:\n",
" tmpPixel=[x,c,0]\n",
" elif h >= 120 and h <180:\n",
" tmpPixel=[0,c,x]\n",
" elif h >= 180 and h <240:\n",
" tmpPixel=[0,x,c]\n",
" elif h >= 240 and h <300:\n",
" tmpPixel=[x,0,c]\n",
" elif h >= 300 and h <360:\n",
" tmpPixel=[c,0,x]\n",
" newLine.append([(tmpPixel[0]+m)*255,(tmpPixel[1]+m)*255,(tmpPixel[2]+m)*255])\n",
" rgb.append(newLine)\n",
" return rgb"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"Plot the saturation of the astronaut image"
]
},
{
"cell_type": "code",
"execution_count": 10,
"metadata": {},
"outputs": [],
"source": [
"img_as_hsv = rgb_to_hsv(img)\n",
"\n",
" # your code\\n\",\n",
"\n",
"img_saturation=copy.deepcopy(np.array(img_as_hsv))\n",
"img_saturation[:,:,0]=0\n",
"img_saturation[:,:,2]=0\n",
"plt.imshow(img_saturation[:, :, 1], cmap='gray')\n",
"plt.show()"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"Increase the saturation by a factor of 2, convert it back to RGB and plot the result."
]
},
{
"cell_type": "code",
"execution_count": 11,
"metadata": {},
"outputs": [],
"source": [
"img_as_hsv = rgb_to_hsv(img)\n",
"# your code\\n\",\n",
"img_saturation = []\n",
"for line in img_as_hsv:\n",
" newLine=[]\n",
" for pixel in line:\n",
" newPixel=[]\n",
" newPixel.append(pixel[0]) \n",
" newSaturation = pixel[1]*2\n",
" if newSaturation >1:\n",
" newSaturation=1\n",
"\n",
" newPixel.append(newSaturation)\n",
" newPixel.append(pixel[2])\n",
" newLine.append(newPixel)\n",
" img_saturation.append(newLine)\n",
"\n",
"img_as_rgb= hsv_to_rgb(img_saturation)\n",
"plt.imshow(img_as_rgb)\n",
"plt.show()"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"# Exercise 3 - Callculation [2 Points]\n",
"\n",
"In the figure below you can see the [CIE-XYZ](https://de.wikipedia.org/wiki/CIE-Normvalenzsystem) color space.\n",
"![](https://upload.wikimedia.org/wikipedia/commons/4/49/CIE-Normfarbtafel.png)\n",
"\n",
"What are the approximate x,y,z values for the following Adobe RGB colors:\n",
"* `(0, 0.5, 0.5)`\n",
"* `(0.33, 0.33, 0.33)`\n",
"\n",
"A sodium-vapor lamp shines with double the intensity of a mercury-vapor lamp\n",
". The light from the sodium lamp only contains \n",
"the spectral line at `589,00nm` and the light from the mercury lamp only the\n",
"spectral line at `435,83 nm`.\n",
"\n",
"What color does a human experience? What are the approximate x,y,z values? \n"
]
},
{
"cell_type": "code",
"execution_count": 12,
"metadata": {},
"outputs": [],
"source": [
"T=np.matrix([[2.04159, -0.56501, -0.34473],[-0.96924, 1.87597, 0.04156],[0.01344, -0.11836, 1.01517]])"
]
},
{
"cell_type": "code",
"execution_count": 13,
"metadata": {},
"outputs": [],
"source": [
"ARGB1 = np.array([0, 0.5, 0.5])\n",
"ARGB2 = np.array([0.33, 0.33, 0.33])\n",
"\n",
"XYZ1 = (T.I).dot(ARGB1)\n",
"print(XYZ1)\n",
"\n",
"XYZ2 = (T.I).dot(ARGB2)\n",
"print(XYZ2)"
]
},
{
"cell_type": "code",
"execution_count": 14,
"metadata": {
"collapsed": true
},
"outputs": [],
"source": [
"# For a sodium-vapor lamp and a mercury-vapor lamp a human does experience a bright red.\n",
"# The approximate x,y,z values are 0.5, 0.32, 0.18\n",
"# You can read the values directly from the XYZ-diagram. Connect the two spectral lines with a line.\n",
"# The resulting color is on this line. \n",
"# As the two colors mix and the sodium lamp has double intensity, the color that a human experience is nearer to \n",
"# the sodium lamp.\n",
"# It is exactly 1/3 of the total line length from the sodium spectral line away. "
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {
"collapsed": true
},
"outputs": [],
"source": []
}
],
"metadata": {
"kernelspec": {
"display_name": "Python 3",
"language": "python",
"name": "python3"
},
"language_info": {
"codemirror_mode": {
"name": "ipython",
"version": 3
},
"file_extension": ".py",
"mimetype": "text/x-python",
"name": "python",
"nbconvert_exporter": "python",
"pygments_lexer": "ipython3",
"version": "3.5.2"
}
},
"nbformat": 4,
"nbformat_minor": 1
}