10.3

'''
Sa slike baboon.bmp izdvojiti crvene elemente. Smatrati da se boje elemenata od značaja nalaze u
kocki čija je stranica dužine 60 u RGB prostoru, a težište u tački (230, 70, 50).
'''

import plotly.express as px
from skimage import io
import numpy as np

img = io.imread('baboon.png')
fig = px.imshow(img)
fig.show()

r = np.abs(img[:,:,0].astype('float') - 230) <= 30
fig = px.imshow(r, color_continuous_scale='gray')
fig.show()

g = np.abs(img[:,:,1].astype('float') - 70) <= 30
fig = px.imshow(g, color_continuous_scale='gray')
fig.show()

b = np.abs(img[:,:,2].astype('float') - 50) <= 30
fig = px.imshow(b, color_continuous_scale='gray')
fig.show()

m = r & g & b
fig = px.imshow(m, color_continuous_scale='gray')
fig.show()

img_masked = img * np.stack((m,m,m),axis=2)
fig = px.imshow(img_masked)
fig.show()