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

video capture and detection working now

Browse Source
This commit is contained in:
Cole Deck 2019-12-13 09:30:11 -06:00
parent c40d84705f
commit 574bf7050c
2 changed files with 3 additions and 3 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.

6
detect.py
View File

@ -164,7 +164,7 @@ def detect(calibration_width, img_file, show, quick):
cv2.drawContours(orig, [box.astype("int")], -1, (0, 255, 0), 2) cv2.drawContours(orig, [box.astype("int")], -1, (0, 255, 0), 2)
else: else:
circular = True circular = True
cv2.circle(orig, (int(x), int(y)), int(radius), (0, 255, 0), 2) cv2.circle(orig, (int(x), int(y)), int(radius), (0, 255, 0), 3)
mask = np.zeros(gray.shape, np.uint8) mask = np.zeros(gray.shape, np.uint8)
cv2.drawContours(mask, [c], 0, 255, -1) cv2.drawContours(mask, [c], 0, 255, -1)
#pixelpoints = np.transpose(np.nonzero(mask)) #pixelpoints = np.transpose(np.nonzero(mask))
@ -246,7 +246,7 @@ def detect(calibration_width, img_file, show, quick):
#cv2.drawContours(orig, [box.astype("int")], -1, (0, 255, 0), 2) #cv2.drawContours(orig, [box.astype("int")], -1, (0, 255, 0), 2)
else: else:
circular = True circular = True
cv2.circle(orig, (int(x), int(y)), int(radius), (0, 255, 0), 1) cv2.circle(orig, (int(x), int(y)), int(radius), (255, 0, 0), 2)
objtype = "Unknown" objtype = "Unknown"
itemw = larger(dimA, dimB) itemw = larger(dimA, dimB)
@ -321,7 +321,7 @@ def detect(calibration_width, img_file, show, quick):
swap(ypos, ypos2) swap(ypos, ypos2)
if rectangular: if rectangular:
cv2.line(orig, (int(xpos), int(ypos)), cv2.line(orig, (int(xpos), int(ypos)),
(int(xpos2), int(ypos2)), (0, 255, 0), 1) (int(xpos2), int(ypos2)), (255, 127, 0), 2)
# print(str(iteml)) # print(str(iteml))
# draw the object sizes on the image # draw the object sizes on the image
if show or True: if show or True: