diff --git a/main.py b/main.py index 5668174..bee7732 100644 --- a/main.py +++ b/main.py @@ -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