fix some stuff

cascade
Cole Deck 5 years ago
parent 5e560935fe
commit 3815a69ee6

Binary file not shown.

After

Width:  |  Height:  |  Size: 3.1 MiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 104 KiB

@ -6,6 +6,7 @@ import argparse
import imutils import imutils
import cv2 import cv2
import math import math
import time
itemw = 0 itemw = 0
itemh = 0 itemh = 0
@ -44,6 +45,11 @@ def smaller(a, b):
return a return a
else: else:
return b return b
def near(a, b, close):
if abs(a-b) < close:
return True
return False
# construct the argument parse and parse the arguments # construct the argument parse and parse the arguments
ap = argparse.ArgumentParser() ap = argparse.ArgumentParser()
ap.add_argument("-i", "--image", required=True, ap.add_argument("-i", "--image", required=True,
@ -70,16 +76,13 @@ if args2.show:
cv2.waitKey(0) cv2.waitKey(0)
gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY) gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
gray = cv2.GaussianBlur(gray, (7, 7), 0) gray = cv2.GaussianBlur(gray, (7, 7), 0)
if args2.show:
cv2.imshow("Item Sorter", gray)
cv2.waitKey(0)
# perform edge detection, then perform a dilation + erosion to # perform edge detection, then perform a dilation + erosion to
# close gaps in between object edges # close gaps in between object edges
edged = cv2.Canny(gray, 50, 100) edged = cv2.Canny(gray, 50, 100)
edged = cv2.dilate(edged, None, iterations=1) edged = cv2.dilate(edged, None, iterations=1)
edged = cv2.erode(edged, None, iterations=1) #edged = cv2.erode(edged, None, iterations=1)
if args2.show: if args2.show:
cv2.imshow("Item Sorter", edged) cv2.imshow("Item Sorter", edged)
@ -131,14 +134,20 @@ for c in cnts:
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),2)
mask = np.zeros(gray.shape,np.uint8)
if pixelsPerMetric is None and circular is True: cv2.drawContours(mask,[c],0,255,-1)
#pixelpoints = np.transpose(np.nonzero(mask))
mean_val = cv2.mean(orig,mask = mask)
#print(str(mean_val))
if pixelsPerMetric is None and circular is True and near(mean_val[0], 63, 30) is True and near(mean_val[1], 108, 30) is True and near(mean_val[2], 104, 30) is True:
pixelsPerMetric = smaller(dA, dB) / args["width"] pixelsPerMetric = smaller(dA, dB) / args["width"]
orig = image.copy() orig = image.copy()
# loop over the contours individually # loop over the contours individually
for c in cnts: for c in cnts:
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:
@ -233,7 +242,7 @@ for c in cnts:
#print(str(convexness) + " % fill") #print(str(convexness) + " % fill")
#if not cv2.isContourConvex(approx): #if not cv2.isContourConvex(approx):
#if cv2.matchShapes(hull, c, 1, 0.0) > 1: #if cv2.matchShapes(hull, c, 1, 0.0) > 1:
if defects.size > 5 and (convexness < 0.9 or boxiness < 0.75): if defects.size > 5 and (convexness < 0.9 and boxiness < 0.7):
objtype = "Screw" objtype = "Screw"
iteml = larger(dimA, dimB) iteml = larger(dimA, dimB)
#print("Screw Length (RAW): " + str(iteml)) #print("Screw Length (RAW): " + str(iteml))
@ -249,7 +258,7 @@ for c in cnts:
iteml = (radius * 2 / pixelsPerMetric + itemw) / 2 iteml = (radius * 2 / pixelsPerMetric + itemw) / 2
print(str(iteml)) #print(str(iteml))
# draw the object sizes on the image # draw the object sizes on the image
if args2.show: if args2.show:
#cv2.putText(orig, "{:.5f}in".format(itemhr), #cv2.putText(orig, "{:.5f}in".format(itemhr),
@ -258,6 +267,10 @@ for c in cnts:
cv2.putText(orig, str(objtype), cv2.putText(orig, str(objtype),
(int(trbrX + 20), int(trbrY)), cv2.FONT_HERSHEY_SIMPLEX, (int(trbrX + 20), int(trbrY)), cv2.FONT_HERSHEY_SIMPLEX,
0.65, (255, 255, 255), 2) 0.65, (255, 255, 255), 2)
if objtype == "Unknown":
cv2.putText(orig, "{:.2f}in".format(itemw) + " x {:.2f}in".format(itemh), # print axle length
(int(trbrX + 20), int(trbrY + 20)), cv2.FONT_HERSHEY_SIMPLEX,
0.65, (255, 255, 255), 2)
if objtype == "Screw": if objtype == "Screw":
cv2.putText(orig, str(iteml) + "in thread", # print screw length cv2.putText(orig, str(iteml) + "in thread", # print screw length
(int(trbrX + 20), int(trbrY + 20)), cv2.FONT_HERSHEY_SIMPLEX, (int(trbrX + 20), int(trbrY + 20)), cv2.FONT_HERSHEY_SIMPLEX,
@ -272,5 +285,5 @@ for c in cnts:
0.65, (255, 255, 255), 2) 0.65, (255, 255, 255), 2)
# show the output image # show the output image
cv2.imshow("Item Sorter", orig) cv2.imshow("Item Sorter", orig)
cv2.waitKey(0) cv2.waitKey(0)
Loading…
Cancel
Save