findcontoursの結果をcsvに出力

Question

画像から抽出した複数の輪郭の各面積を１つのcsvファイルに出力したいです。csvの書き込みイメージは、
A B
1 輪郭番号 面積
2 輪郭番号 面積
…
のような感じです。
printでは上手く輪郭番号と面積は出力されましたが、それをcsvに出力する方法が分かりません。
コードは以下になります。まだ初心者の為、不要なコードも含まれているかもしれませんがご了承下さい。是非ともお力添えををお願い致します。
for分の下が投稿するとタブ空白が消えて左つめになってしまっていますが、areaとprintはタブ空白あります。

import cv2
import numpy as np
from matplotlib import pyplot as plt
import sys
import csv

img = cv2.imread('dot.jpg')
b = cv2.bitwise_not(img)

c=cv2.cvtColor(b,cv2.COLOR_BGR2GRAY)
d=cv2.threshold(c,50,255,cv2.THRESH_BINARY_INV)[1]

e = cv2.bitwise_not(d)

image, contours, hierarchy = cv2.findContours(e,cv2.RETR_TREE,cv2.CHAIN_APPROX_SIMPLE)

for i, cnt in enumerate(contours):
area = cv2.contourArea(cnt)
print('contour: {}, area: {}'.format(i, area))

Answer 1

回答

csv モジュールをお使いください。
使い方については下記リンクを参照ください。

Python - CSV の読み込み/書き込み まとめ - Pynote

コード

入力画像

import cv2
import csv

# 画像を読み込む。
img = cv2.imread("test.png")

# グレースケールに変換する。
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)

# 輪郭を抽出する。
image, contours, hierarchy = cv2.findContours(gray, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)

data = []
for i, cnt in enumerate(contours):
    area = cv2.contourArea(cnt)
    data.append({"No": i, "Area": area})

with open("output.csv", "w", newline="") as f:
    writer = csv.DictWriter(f, fieldnames=["No", "Area"])
    writer.writerows(data)

0,4900.0
1,3362.0
2,10743.0
3,3784.0
4,1923.0

追記

csv モジュールを使わない方法

import cv2

# 画像を読み込む。
img = cv2.imread("test.png")

# グレースケールに変換する。
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)

# 輪郭を抽出する。
image, contours, hierarchy = cv2.findContours(
    gray, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE
)

data = []
for cnt in contours:
    area = cv2.contourArea(cnt)
    data.append(area)

with open("output.csv", "w", newline="") as f:
    f.write("No,Area\n".format(i, area))
    for i, area in enumerate(data):
        f.write("{},{}\n".format(i, area))

参考文献

• Python - CSV の読み込み/書き込み まとめ - Pynote

Comments

[mukai1174]

ご回答ありがとうございます。回答者様のサンプル画像とコードで試したみましたら以下のようなエラーがかえってきてしまいました。何が問題だと思われるでしょうか？

runfile('C:/Users/xxx/abc.py', wdir='C:/Users/xxx')
Reloaded modules: csv
Traceback (most recent call last):

File "<ipython-input-16-e2ca45ba9c88>", line 1, in <module>
runfile('C:/Users/xxx/abc.py', wdir='C:/Users/xxx')

File "C:\Users\xxx\Anaconda3\envs\openCV\lib\site-packages\spyder_kernels\customize\spydercustomize.py", line 827, in runfile
execfile(filename, namespace)

File "C:\Users\xxx\Anaconda3\envs\openCV\lib\site-packages\spyder_kernels\customize\spydercustomize.py", line 110, in execfile
exec(compile(f.read(), filename, 'exec'), namespace)

File "C:/Users/xxx/abc.py", line 18, in <module>
contours, hierarchy = cv2.findContours(gray, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)

ValueError: too many values to unpack (expected 2)

[tiitoi]

OpenCV3 をお使いの場合、findContours() の返り値は image, contours, hierarchy の3つになります。
質問のコードを修正したので、これでどうでしょうか？

[mukai1174]

ご対応ありがとうございます。再度実行してみましたが下記エラーが出てしまいました。自分でもエラー修正を試みましたが分からずご協力して頂けると大変ありがたいです。

runfile('C:/Users/xxx/abc.py', wdir='C:/Users/xxx')
Reloaded modules: csv
Traceback (most recent call last):

File "<ipython-input-2-e2ca45ba9c88>", line 1, in <module>
runfile('C:/Users/xxx/abc.py', wdir='C:/Users/xxx')

File "C:\Users\xxx\Anaconda3\envs\openCV\lib\site-packages\spyder_kernels\customize\spydercustomize.py", line 827, in runfile
execfile(filename, namespace)

File "C:\Users\xxx\Anaconda3\envs\openCV\lib\site-packages\spyder_kernels\customize\spydercustomize.py", line 110, in execfile
exec(compile(f.read(), filename, 'exec'), namespace)

File "C:/Users/xxx/abc.py", line 26, in <module>
writer = csv.DictWriter(f, fieldnames=["No", "Area"])

AttributeError: module 'csv' has no attribute 'DictWriter'

[tiitoi]

'csv' has no attribute 'DictWriter' となる理由はわかりませんが、csv モジュールを使わない方法を追記しました。

[mukai1174]

上手く出力されました。ご協力して頂きまして大変助かりました。ありがとうございます。