画像から抽出した輪郭線をcsvに出力したい

Question

前提・実現したいこと

画像から抽出した輪郭線をcsvに出力したいのですがうまく書き出すことができません。
１行づつ[x,y]の形式でcsv形式にしたいです。

発生している問題・エラーメッセージ

該当のソースコード

import numpy as np
import cv2

im = cv2.imread("C://test/p5d128.bmp",1)
imgray = cv2.cvtColor(im,cv2.COLOR_BGR2GRAY)
ret,thresh = cv2.threshold(imgray,127,255,0)
image, contours, hierarchy = cv2.findContours(thresh,cv2.RETR_CCOMP,cv2.CHAIN_APPROX_SIMPLE)
print(contours)
np.savetxt("testcontour.csv", contours, delimiter=',')

試したこと

print(contours)では下記のようになるのですが、うまくいきません。

[array([[[ 0, 0]],

   [[  0, 706]],

   [[  2, 708]],

   ...,

   [[990, 707]],

   [[991, 706]],

   [[991,   0]]], dtype=int32)]

補足情報（FW/ツールのバージョンなど）

ここにより詳細な情報を記載してください。

Comments

[tiitoi]

輪郭は複数検出される場合があるので、(輪郭の数、点の数、2) の配列になりますが、CSV は2次元の配列しか保存できません。この場合、どのようにしたいのでしょうか？

[ryoucyan2000]

今回のサンプルは輪郭は１個しかないのですが、複数の輪郭が存在する場合はそれぞれファイルを作成して格納としたいです。また点の数については各ファイルの最後に記述されるのがベストです

Answer 1

こんな感じでしょうか？

import csv
import cv2
import numpy as np

# 画像読み込み
img = cv2.imread('test.png', cv2.IMREAD_GRAYSCALE)
# 輪郭抽出
_, contours, _ = cv2.findContours(img, cv2.RETR_CCOMP, cv2.CHAIN_APPROX_SIMPLE)
# 不要な次元削減 (NumPoints, 1, 2) -> (NumPoints, 2)
contours = [np.squeeze(cnt, axis=1) for cnt in contours]

# 輪郭ごとにファイルに出力する。
for i, cnt in enumerate(contours):
    np.savetxt('contour_{}.csv'.format(i), cnt, delimiter=',')

Comments

[ryoucyan2000]

目的の結果が得られました！勉強になりました。ありがとう御座います