Projet

Général

Profil

Télécharger (1,04 ko)

Support #11801 » code_localisation_bille.py

Ci-joint le code python pour la localisation de la bille dans le labyrinthe. - Anonyme, 03/12/2019 16:26

 
import cv2, time
import numpy as np


device = cv2.VideoCapture(1)
x = 0
y= 0
width = 800 - x
height = 1000 - y

term_criteria = (cv2.TERM_CRITERIA_EPS | cv2.TERM_CRITERIA_COUNT, 10, 1)

while True:

lower_range = np.array([90,130,130])
upper_range = np.array([105,255,255])


ret, frame = device.read()
hsv = cv2.cvtColor(frame,cv2.COLOR_BGR2HSV)
mask = cv2.inRange(hsv, lower_range, upper_range)
result = cv2.bitwise_and(frame, frame, mask=mask)
cv2.imshow("Result", result)


ret, track_window = cv2.CamShift(mask, (x, y, width, height), term_criteria)


if np.any(result > 120) :
pts = cv2.boxPoints(ret)
pts = np.int0(pts)
print(track_window)
cv2.polylines(frame, [pts], True, (255, 0, 0), 2)
else :
print("pas d'objet détécté")


cv2.imshow("mask", mask)
cv2.imshow("Frame", frame)



key=cv2.waitKey(1)

if key == ord('q'):
break


device.release()
cv2.destroyAllWindows()
    (1-1/1)