You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
item-sort/run_detect.py

28 lines
765 B
Python

import detect
import timeit
import cv2
calibration_width = 0.75
image = "img7.jpg"
images = ("img.jpg", "img2.jpg", "img3.jpg", "img4.jpg", "img5.jpg", "img6.jpg", "img7.jpg", "img8.jpg")
show = False
video = False
def go():
#for file in images:
detect.detect(calibration_width, "img7.jpg", show)
if not video:
elapsed_time = timeit.timeit(go, number=3)/3
print(elapsed_time)
else :
capture = cv2.VideoCapture('tcpclientsrc host=192.168.43.152 port=5000 ! queue ! decodebin ! appsink', cv2.CAP_GSTREAMER)
while True:
ret,frame = capture.read()
if not ret:
print('empty frame')
break
print('frame')
cv2.imshow('receive', frame)
if cv2.waitKey(1)&0xFF == ord('q'):
break