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

fix compiled version, improve contrast

Browse Source
This commit is contained in:
Cole Deck 2019-12-13 10:13:50 -06:00
parent af0410be8e
commit 129ad2a762
6 changed files with 349 additions and 325 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.

653
detect.c
View File

File diff suppressed because it is too large Load Diff

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

Binary file not shown.

12
detect.py
View File

@ -332,12 +332,12 @@ def detect(calibration_width, img_file, show, quick):
cv2.putText(orig, str(objtype), cv2.putText(orig, str(objtype),
(int(x - 25), int(y + radius + 20) (int(x - 25), int(y + radius + 20)
), cv2.FONT_HERSHEY_SIMPLEX, ), cv2.FONT_HERSHEY_SIMPLEX,
0.55, (0, 0, 255), 2) 0.6, (50, 50, 220), 2)
else: else:
cv2.putText(orig, str(objtype), cv2.putText(orig, str(objtype),
(int(xpos2 + 10), int(ypos2 + 20) (int(xpos2 + 10), int(ypos2 + 20)
), cv2.FONT_HERSHEY_SIMPLEX, ), cv2.FONT_HERSHEY_SIMPLEX,
0.55, (0, 0, 255), 2) 0.6, (50, 50, 220), 2)
output = "" output = ""
objname = objtype; objname = objtype;
if objtype == "Unknown": if objtype == "Unknown":
@ -351,14 +351,14 @@ def detect(calibration_width, img_file, show, quick):
print(objname) print(objname)
if circular: if circular:
cv2.putText(orig, output, # print data cv2.putText(orig, output, # print data
(int(x - 25), int(y + radius + 35) (int(x - 25), int(y + radius + 40)
), cv2.FONT_HERSHEY_SIMPLEX, ), cv2.FONT_HERSHEY_SIMPLEX,
0.5, (0, 0, 255), 1) 0.5, (50, 50, 220), 1)
else: else:
cv2.putText(orig, output, # print data cv2.putText(orig, output, # print data
(int(xpos2 + 10), int(ypos2 + 35) (int(xpos2 + 10), int(ypos2 + 40)
), cv2.FONT_HERSHEY_SIMPLEX, ), cv2.FONT_HERSHEY_SIMPLEX,
0.5, (0, 0, 255), 1) 0.5, (50, 50, 220), 1)
# show the output image # show the output image
if show: if show:

9
run_detect.py
View File

@ -5,14 +5,13 @@ 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")
show = False video = False
video = True
def go(): def go():
#for file in images: for file in images:
detect.detect(calibration_width, "img7.jpg", show) detect.detect(calibration_width, file, True, False)
if not video: if not video:
elapsed_time = timeit.timeit(go, number=3)/3 elapsed_time = timeit.timeit(go, number=1)/1
print(elapsed_time) print(elapsed_time)
else : else :
#tcp capture = cv2.VideoCapture('udpsrc port=5001 ! gdpdepay ! rtph264depay ! avdec_h264 ! videoconvert ! videorate ! video/x-raw,framerate=5/1 ! appsink', cv2.CAP_GSTREAMER) #tcp capture = cv2.VideoCapture('udpsrc port=5001 ! gdpdepay ! rtph264depay ! avdec_h264 ! videoconvert ! videorate ! video/x-raw,framerate=5/1 ! appsink', cv2.CAP_GSTREAMER)