I used opencv to detect contours in Python. I wrote the code for saving this contour as a dxf file as below through a search.
cap = EasyPySpin.VideoCapture(0)
_, frame = cap.read()
frame = cv2.cvtColor(frame, cv2.COLOR_GRAY2BGR)
frame = cv2.resize(frame, dsize=(0,0), fx=0.26, fy=0.26, interpolation=cv2.INTER_AREA)
frame_blr = cv2.GaussianBlur(frame, (3, 3), 0)
canny = cv2.Canny(frame_blr, 255, 0)
contours, hierarchy=cv2.findContours(canny, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_NONE)
contours_draw = cv2.drawContours(frame, contours, -1, (0, 255, 0), 1)
cv2.imshow('contour', contours_draw)
contours = [np.squeeze (cnt, axis = 1) for cnt in contours]
ctr = contours[1]
dwg = ezdxf.new ('R2010') # create a new DXF R2010 drawing, official DXF version name: 'AC1024'
msp = dwg.modelspace () # add new entities to the model space
dwg.layers.new (name = 'MyLines', dxfattribs = {'color': 3}) # 3 = Green
for i in range (len (ctr)):
n = i + 1
if n>= len (ctr):
n = 0
msp.add_line (ctr [i], ctr [n], dxfattribs = {'layer': 'MyLines'}) # add a LINE entity
print (ctr [i], '->', ctr [n])
dwg.saveas ('line.dxf')
But the detected contour and the line drawn in the dxf file are different. Also, a slightly different line is drawn in the dxf file every time it runs.I don't know why.
Contour Image > enter image description here DXF file > enter image description here
And I don't know what this code means.
contours = [np.squeeze (cnt, axis = 1) for cnt in contours]
ctr = contours[1]
If I delete this code and edit the crt into contours, the following error occurs.
Traceback (most recent call last):
File "c:/Users/MyPc/KHS/p40.contour detection, (x,y).py", line 38, in <module>
msp.add_line (contours [i], contours [n], dxfattribs = {'layer': 'MyLines'}) # add a LINE entity
File "C:\Users\MyPc\AppData\Local\Programs\Python\Python38\lib\site-packages\ezdxf\graphicsfactory.py", line 121, in add_line
dxfattribs["start"] = Vec3(start)
File "src\ezdxf\acc\vector.pyx", line 417, in ezdxf.acc.vector.Vec3.__cinit__
TypeError: invalid argument count
>>> import cv2
... import numpy as np
... frame = cv2.imread("cyan quick_3_MAD.png")
>>> canny = cv2.Canny(frame, 255, 0)
>>> contours, hierarchy = cv2.findContours(canny, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_NONE)
>>> contours_draw = cv2.drawContours(frame, contours, -1, (0, 255, 0), 1)
>>> cv2.imwrite("out.png", contours_draw)
output:
np.squeeze
?For this example, contours looks like this:
>>> print(str(contours)[:400])
[array([[[629, 568]],
[[629, 569]],
[[629, 570]],
...
>>> len(contours)
188
>>> contours[0].shape
(59, 1, 2)
After using np.squeeze
, unnecessary 1-sized axis is removed.
squeezed = [np.squeeze(cnt, axis=1) for cnt in contours]
>>> len(squeezed)
188
>>> squeezed[0].shape
(59, 2)
To quote from document np.squeeze
numpy.squeeze(a, axis=None)
Remove axes of length one from a.
You used ctr = contours[1]
, which means you Only draw first contour in list among multiple contours.
Instead, you need to iterate thru contours and draw every contour it has.
>>> import ezdxf
... dwg = ezdxf.new("R2010")
... msp = dwg.modelspace()
... dwg.layers.new(name="greeny green lines", dxfattribs={"color": 3})
>>> for ctr in squeezed:
... for n in range(len(ctr)):
... if n >= len(ctr) - 1:
... n = 0
... try:
... msp.add_line(ctr[n], ctr[n + 1], dxfattribs={"layer": "greeny green lines", "lineweight": 20})
... except IndexError:
... pass
>>> dwg.saveas("output.dxf")
output:
To flip the image, we can just simply invert y values.
Since squeezed
was list of np arrays that contains countours, we can simply multiply vector [1, -1]
to it to flip values in y axis.
>>> inverted_squeezed = [arr * [1, -1] for arr in squeezed]
>>> for ctr in inverted_squeezed:
... for n in range(len(ctr)):
... if n >= len(ctr) - 1:
... n = 0
... try:
... msp.add_line(ctr[n], ctr[n + 1], dxfattribs={"layer": "greeny green lines", "lineweight": 20})
... except IndexError:
... pass
>>> dwg.saveas("output.dxf")