from imageio import imread, imsave
from numpy import shape, zeros, uint8
from matplotlib.pyplot import imshow,show
from math import sqrt
import os
Lseuil = os.listdir("Test2/") #Liste des images seuillées
Limg = os.listdir("VID1/") #Liste des images à seuiller
def seuillageC(cR, cG, cB, s): #Classiquement 75,63,65,20
"""cR, cG, cB : couleur de l'élément qu'on souhaite repérer
On seuille tout un dossier et on stocke dans un nouveau"""
for k in Limg:
img = imread("VID1/" +k)
taille = shape(img)
nlig = taille[0]
ncol = taille[1]
imgSEUIL = zeros((nlig,ncol),uint8)
for i in range(180,220):
for j in range(520,580):
d = sqrt((img[i,j,0]-cR)**2+(img[i,j,1]-cG)**2+(img[i,j,2]-cB)**2)
if d<s : imgSEUIL[i,j] = 0
else : imgSEUIL[i,j] = 255
for i in range(450,550):
for j in range(510,520):
d = sqrt((img[i,j,0]-cR)**2+(img[i,j,1]-cG)**2+(img[i,j,2]-cB)**2)
if d<s : imgSEUIL[i,j] = 0
else : imgSEUIL[i,j] = 255
imsave("Test1/imgSeuil" + k, imgSEUIL)