Projet

Général

Profil

Feature #8914 » detection_fleche.py

Anonyme, 19/11/2017 15:57

 
# -*- coding: utf-8 -*-
"""
Created on Sun Nov 19 11:34:53 2017

@author: Rémi
"""

import cv2
import numpy as np

print("Entrez le nom de l'image")
nom = input("exemple : template.png \n")
image = cv2.imread(nom)

traitement = cv2.cvtColor(image, cv2.COLOR_BGR2HSV)
imgray = cv2.cvtColor(image,cv2.COLOR_BGR2GRAY)

#permet de trouver la couleur verte
for a in range( np.shape( traitement)[0]):
for b in range( np.shape( traitement)[1]):
if ( traitement[ a, b, 0] > 40 and traitement[ a, b, 0] < 70 and traitement[ a, b, 1] > 90):
imgray[ a, b]= 255
else:
imgray[ a, b]= 0

imgray = cv2.GaussianBlur( imgray, (7, 7), 0)

#permet de detecter les contours
ret,thresh = cv2.threshold(imgray,127,255,0)
image1, contours, hierarchy = cv2.findContours(thresh,cv2.RETR_TREE,cv2.CHAIN_APPROX_SIMPLE)

#permet de dessiner les contours
cv2.drawContours(image, contours, -1, (0,255,0), 9)

#permet de cree l'image resultat
cv2.imwrite('resultat.png',image)
(1-1/3)