shark/item-sort
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

imx135 support

Browse Source
This commit is contained in:
Cole Deck 2019-12-13 08:30:14 -06:00
parent bfa501b6db
commit 72d0015d8f
6 changed files with 18 additions and 24 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

BIN
__pycache__/detect.cpython-38.pyc
View File

Binary file not shown.

BIN
build/temp.linux-x86_64-3.8/detect.o
View File

Binary file not shown.

16
detect.c
View File

@ -5741,7 +5741,7 @@ static PyObject *__pyx_pf_6detect_14detect(CYTHON_UNUSED PyObject *__pyx_self, P
* pixelsPerMetric = smaller(dA, dB) / calibration_width * pixelsPerMetric = smaller(dA, dB) / calibration_width
* continue # <<<<<<<<<<<<<< * continue # <<<<<<<<<<<<<<
* *
* pixelsPerMetric = 25 * #pixelsPerMetric = 25
*/ */
goto __pyx_L16_continue; goto __pyx_L16_continue;
@ -5765,19 +5765,9 @@ static PyObject *__pyx_pf_6detect_14detect(CYTHON_UNUSED PyObject *__pyx_self, P
} }
__Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0; __Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
/* "detect.py":181
* continue
*
* pixelsPerMetric = 25 # <<<<<<<<<<<<<<
* orig = image.copy()
* objtype = "Unknown"
*/
__Pyx_INCREF(__pyx_int_25);
__Pyx_DECREF_SET(__pyx_v_pixelsPerMetric, __pyx_int_25);
/* "detect.py":182 /* "detect.py":182
* *
* pixelsPerMetric = 25 * #pixelsPerMetric = 25
* orig = image.copy() # <<<<<<<<<<<<<< * orig = image.copy() # <<<<<<<<<<<<<<
* objtype = "Unknown" * objtype = "Unknown"
* objname = "" * objname = ""
@ -5803,7 +5793,7 @@ static PyObject *__pyx_pf_6detect_14detect(CYTHON_UNUSED PyObject *__pyx_self, P
__pyx_t_5 = 0; __pyx_t_5 = 0;
/* "detect.py":183 /* "detect.py":183
* pixelsPerMetric = 25 * #pixelsPerMetric = 25
* orig = image.copy() * orig = image.copy()
* objtype = "Unknown" # <<<<<<<<<<<<<< * objtype = "Unknown" # <<<<<<<<<<<<<<
* objname = "" * objname = ""

BIN
detect.cpython-38-x86_64-linux-gnu.so
View File

Binary file not shown.

4
detect.py
View File

@ -178,7 +178,7 @@ def detect(calibration_width, img_file, show, quick):
pixelsPerMetric = smaller(dA, dB) / calibration_width pixelsPerMetric = smaller(dA, dB) / calibration_width
continue continue
pixelsPerMetric = 25 #pixelsPerMetric = 25
orig = image.copy() orig = image.copy()
objtype = "Unknown" objtype = "Unknown"
objname = "" objname = ""
@ -187,7 +187,7 @@ def detect(calibration_width, img_file, show, quick):
#orig = image.copy() #orig = image.copy()
num += 1 num += 1
# if the contour is not sufficiently large, ignore it # if the contour is not sufficiently large, ignore it
if cv2.contourArea(c) < 100: # or pixelsPerMetric is None: if cv2.contourArea(c) < 100 or pixelsPerMetric is None:
continue continue
# compute the rotated bounding box of the contour # compute the rotated bounding box of the contour

22
run_detect.py
View File

@ -1,6 +1,7 @@
import detect import detect
import timeit import timeit
import cv2 import cv2
from imutils.video import FPS
calibration_width = 0.75 calibration_width = 0.75
image = "img7.jpg" image = "img7.jpg"
images = ("img.jpg", "img2.jpg", "img3.jpg", "img4.jpg", "img5.jpg", "img6.jpg", "img7.jpg", "img8.jpg") images = ("img.jpg", "img2.jpg", "img3.jpg", "img4.jpg", "img5.jpg", "img6.jpg", "img7.jpg", "img8.jpg")
@ -14,19 +15,22 @@ if not video:
elapsed_time = timeit.timeit(go, number=3)/3 elapsed_time = timeit.timeit(go, number=3)/3
print(elapsed_time) print(elapsed_time)
else : else :
capture = cv2.VideoCapture('tcpclientsrc host=192.168.86.108 port=5001 ! gdpdepay ! rtph264depay ! avdec_h264 ! videoconvert ! appsink', cv2.CAP_GSTREAMER) capture = cv2.VideoCapture('udpclientsrc host=192.168.43.152 port=5001 ! gdpdepay ! rtph264depay ! avdec_h264 ! videoconvert ! videorate ! video/x-raw,framerate=5/1 ! appsink', cv2.CAP_GSTREAMER)
# server command for pi camera /opt/vc/bin/raspivid -t 0 -w 1920 -h 1080 -hf -fps 5 -o - | gst-launch-1.0 -v fdsrc ! h264parse ! rtph264pay config-interval=1 pt=96 ! gdppay ! tcpserversink host=192.168.86.108 port=5001 # server command for imx135 camera ./video2stdout | gst-launch-1.0 -v fdsrc ! h264parse ! rtph264pay config-interval=1 pt=96 ! gdppay ! tcpserversink host=192.168.43.152 port=5001
ret,frame = capture.read() #ret,frame = capture.read()
detect.detect(calibration_width, "img7.jpg", True, False) #detect.detect(calibration_width, "img7.jpg", True, False)
detect.detect(calibration_width, frame, True, True) #detect.detect(calibration_width, frame, True, True)
x = 0
while True: while True:
ret,frame = capture.read() ret = capture.grab()
x+=1
if not ret: if not ret:
print('empty frame') print('empty frame')
break break
#print('frame') #print('frame')
if x > 0:
cv2.imshow('Item Sorter', detect.detect(calibration_width, frame, True, True)) ret,frame = capture.retrieve()
cv2.imshow('Item Sorter', detect.detect(calibration_width, frame, True, True))
x = 0
if cv2.waitKey(1)&0xFF == ord('q'): if cv2.waitKey(1)&0xFF == ord('q'):
break break