|
|
|
@ -50,6 +50,13 @@ def near(a, b, close):
|
|
|
|
|
if abs(a-b) < close:
|
|
|
|
|
return True
|
|
|
|
|
return False
|
|
|
|
|
|
|
|
|
|
def swap(a, b):
|
|
|
|
|
tmp = a
|
|
|
|
|
a = b
|
|
|
|
|
b = tmp
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
# construct the argument parse and parse the arguments
|
|
|
|
|
ap = argparse.ArgumentParser()
|
|
|
|
|
ap.add_argument("-i", "--image", required=True,
|
|
|
|
@ -210,7 +217,7 @@ for c in cnts:
|
|
|
|
|
rectangular = False
|
|
|
|
|
if boxiness > circleness:
|
|
|
|
|
rectangular = True
|
|
|
|
|
cv2.drawContours(orig, [box.astype("int")], -1, (0, 255, 0), 2)
|
|
|
|
|
#cv2.drawContours(orig, [box.astype("int")], -1, (0, 255, 0), 2)
|
|
|
|
|
else:
|
|
|
|
|
circular = True
|
|
|
|
|
cv2.circle(orig,(int(x),int(y)),int(radius),(0,255,0),2)
|
|
|
|
@ -260,7 +267,22 @@ for c in cnts:
|
|
|
|
|
objtype = "Axle"
|
|
|
|
|
iteml = (radius * 2 / pixelsPerMetric + itemw) / 2
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
rows,cols = orig.shape[:2]
|
|
|
|
|
[vx,vy,xx,yy] = cv2.fitLine(c, cv2.DIST_L2,0,0.01,0.01)
|
|
|
|
|
lefty = int((-xx*vy/vx) + yy)
|
|
|
|
|
righty = int(((cols-xx)*vy/vx)+yy)
|
|
|
|
|
#cv2.line(orig,(cols-1,righty),(0,lefty),(0,255,0),2)
|
|
|
|
|
slope = (lefty - righty) / (1 - cols)
|
|
|
|
|
angle = math.atan(slope)
|
|
|
|
|
xpos = x - math.cos(angle) * radius
|
|
|
|
|
ypos = y - math.sin(angle) * radius
|
|
|
|
|
xpos2 = x + math.cos(angle) * radius
|
|
|
|
|
ypos2 = y + math.sin(angle) * radius
|
|
|
|
|
if xpos > xpos2:
|
|
|
|
|
swap(xpos, xpos2)
|
|
|
|
|
swap(ypos, ypos2)
|
|
|
|
|
if rectangular:
|
|
|
|
|
cv2.line(orig,(int(xpos),int(ypos)),(int(xpos2), int(ypos2)),(0,255,0),2)
|
|
|
|
|
#print(str(iteml))
|
|
|
|
|
# draw the object sizes on the image
|
|
|
|
|
if args2.show:
|
|
|
|
@ -268,23 +290,23 @@ for c in cnts:
|
|
|
|
|
# (int(trbrX + 20), int(trbrY)), cv2.FONT_HERSHEY_SIMPLEX,
|
|
|
|
|
# 0.65, (255, 255, 255), 2)
|
|
|
|
|
cv2.putText(orig, str(objtype),
|
|
|
|
|
(int(trbrX + 20), int(trbrY)), cv2.FONT_HERSHEY_SIMPLEX,
|
|
|
|
|
(int(xpos2 + 10), int(ypos2 + 20)), cv2.FONT_HERSHEY_SIMPLEX,
|
|
|
|
|
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,
|
|
|
|
|
(int(xpos2 + 10), int(ypos2 + 40)), cv2.FONT_HERSHEY_SIMPLEX,
|
|
|
|
|
0.65, (255, 255, 255), 2)
|
|
|
|
|
if objtype == "Screw":
|
|
|
|
|
cv2.putText(orig, str(iteml) + "in thread", # print screw length
|
|
|
|
|
(int(trbrX + 20), int(trbrY + 20)), cv2.FONT_HERSHEY_SIMPLEX,
|
|
|
|
|
cv2.putText(orig, str(iteml) + "in", # print screw length
|
|
|
|
|
(int(xpos2 + 10), int(ypos2 + 40)), cv2.FONT_HERSHEY_SIMPLEX,
|
|
|
|
|
0.65, (255, 255, 255), 2)
|
|
|
|
|
if objtype == "Standoff":
|
|
|
|
|
cv2.putText(orig, str(iteml) + "in", # print standoff length
|
|
|
|
|
(int(trbrX + 20), int(trbrY + 20)), cv2.FONT_HERSHEY_SIMPLEX,
|
|
|
|
|
(int(xpos2 + 10), int(ypos2 + 40)), cv2.FONT_HERSHEY_SIMPLEX,
|
|
|
|
|
0.65, (255, 255, 255), 2)
|
|
|
|
|
if objtype == "Axle":
|
|
|
|
|
cv2.putText(orig, "{:.2f}in".format(iteml), # print axle length
|
|
|
|
|
(int(trbrX + 20), int(trbrY + 20)), cv2.FONT_HERSHEY_SIMPLEX,
|
|
|
|
|
(int(xpos2 + 10), int(ypos2 + 40)), cv2.FONT_HERSHEY_SIMPLEX,
|
|
|
|
|
0.65, (255, 255, 255), 2)
|
|
|
|
|
|
|
|
|
|
# show the output image
|
|
|
|
|