ezdxfを使って.dxfデータを作りたい

Question

前提・実現したいこと

pythonのcv2を使って画像のエッジを検出するプログラムを作りました。
検出したエッジを.dxfの形式で出力したいのですが、ezdxfを使って検出したエッジを作図する方法を教えて頂きたいです。

該当のソースコード

コード
import cv2
import numpy as np
import matplotlib.pyplot as plt

img = cv2.imread("/Users/mutaguchitakuma/Desktop/original.jpg")
img2 = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
plt.imshow(img2)
plt.show()

青の領域のマスクから青じゃない部分を取得

blue_min = np.array([140, 0, 0], np.uint8)
blue_max = np.array([255, 255, 255], np.uint8)

mask_blue = cv2.inRange(img2, blue_min, blue_max)
img3 = cv2.bitwise_not(mask_blue)
plt.imshow(cv2.cvtColor(mask_blue, cv2.COLOR_GRAY2RGB))

輪郭抽出

_, contours, hierarchy = cv2.findContours(img3, cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE)
contours = [np.squeeze(cnt, axis=1) for cnt in contours]

size = img.shape
h = size[0]
w = size[1]

imgnew = np.zeros((h, w, 3), np.uint8)
img4 = cv2.drawContours(imgnew, contours, 1, (0,255,0), 3)

print("contours=",len(contours), "hierarchy=",len(hierarchy) )
print(contours[1])

plt.imshow(img4)
plt.show

cv2.imwrite('out.jpg', img4)

Answer 1

以下のような感じでしょうか。
簡単な使い方についてはTutorialsを参考にしました。

# 検出処理は略

ctr=[[ 519,  269], # = Contours[1]
     [ 520,  268],
     [1099,  268],
     [1100,  269],
     [1100,  644],
     [1099,  645],
     [ 520,  645],
     [ 519,  644]]

import ezdxf

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')

Comments

[TAKUMA.py]

勉強になりました。
本当にありがとうございました！