Pythonで求めた円の半径と中心座標をcsvで保存するソフト

Question

下記の画像から青色の矢印の座標を特定し、最小の円を描画するソフトを作りました。
その円の中心座標と半径をcsvに保存したいのですが、以下のようなエラーが出ます。
改善点を教えていただきたいです。

   # -*- coding:utf-8 -*-
import cv2
import numpy as np
import sys



# 青色の検出
def detect_blue_color(img):
    # HSV色空間に変換
    hsv = cv2.cvtColor(img, cv2.COLOR_BGR2HSV)

    # 青色のHSVの値域1
    hsv_min = np.array([40,140,120])
    hsv_max = np.array([150,255,255])

    # 青色領域のマスク（255：赤色、0：赤色以外）    
    mask = cv2.inRange(hsv, hsv_min, hsv_max)

    # マスキング処理
    masked_img = cv2.bitwise_and(img, img, mask=mask)

    return mask, masked_img

for i in range(2, 35):
    img = cv2.imread('CoD1/filename_{0:05d}.png'.format(i))

    # 色検出（赤、緑、青）
    mask, masked_img = detect_blue_color(img)

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

    # 小さい輪郭は誤検出として削除する
    contours = list(filter(lambda x: cv2.contourArea(x) > 10, contours))


    def get_center(contour):
        """輪郭の中心を取得する。
        """
        # 輪郭のモーメントを計算する。
        M = cv2.moments(contour)
        # モーメントから重心を計算する。
        cx = M["m10"] / M["m00"]
        cy = M["m01"] / M["m00"]

        return cx, cy

   # 輪郭の中心を取り出す。
    centers = [get_center(x) for x in contours]

    # 確認用に各中心を描画
    centers = np.array(centers, dtype=int)
    for c in centers:
        img = cv2.circle( img, tuple(c), 3, (0,0,255), 1)

    # 最小外接円を求める
    cs = centers.reshape(len(centers), 1, 2).astype(int) # カタチを整える
    (x,y),radius = cv2.minEnclosingCircle(cs)
    print(x,y,radius) # 中心座標と半径
   

     # 結果を CSV で保存する。
    with open('centers2.csv','a') as f_handle: 
        np.savetxt(f_handle, (x,y), radius , delimiter=",", fmt="%.0f")

エラーコード
96.0 93.0 9.999999747378752e-05
Traceback (most recent call last):
File "Blue.py", line 67, in <module>
np.savetxt(f_handle, (x,y), radius , delimiter=",", fmt="%.0f")
File "<array_function internals>", line 4, in savetxt
TypeError: _savetxt_dispatcher() got multiple values for argument 'fmt'

Comments

[TakaiY]

そこまで書けたのであれば、「python csv」などで検索すればcsvモジュールの紹介は出けきますよね。
データは辞書形式で持っておけば順に保存できますよ。

Answer 1

下記でどうでしょうか？

with open('centers2.csv','a') as f_handle:
    np.savetxt(f_handle, (x,y,radius) , delimiter=",", fmt="%.0f")

Comments

[gigagiga64]

動くようにはなったのですが、１列目にx,y,radiusが全て格納されてしまいます。

[meg_]

> 動くようにはなったのですが、１列目にx,y,radiusが全て格納されてしまいます。
それが問題なのですか？質問にはcsvに保存する以外の要望は見受けられませんでしたが。

Answer 2

簡単な方法は３通りです。

pandasでcsvファイルの書き出し・追記（to_csv）

NumPyでCSVファイルを読み込み・書き込み（入力・出力）

csv --- CSV ファイルの読み書き

どれが向いているかは人それぞれです。

Comments

[gigagiga64]

csvファイルではNumpyを使って保存をしています。for文で書くので追記モードで書いております。x,y,radiusをcsvに保存する場合にどのように書けば正しく保存できるのでしょうか。具体的に教えていただきたいです

[ppaul]

回答に書いた
https://note.nkmk.me/python-numpy-loadtxt-genfromtxt-savetxt/
の
CSVファイルに書き込み（出力）: np.savetxt()
という部分を読んで、どこがわからないのかを質問してください。

[gigagiga64]

上記のmeg_さんの回答を踏まえ
with open('centers2.csv','a') as f_handle:
np.savetxt(f_handle, (x,y,radius) , delimiter=",", fmt="%.0f")
のように書いたところx,y,radiusが一列目に格納されてしまいます。
このx,y,radiusをcsvファイルの１列目、２列目、３列目にそれぞれ格納するにはどうすればいいかわからないという状況です。
上記のサイトは質問する前から読んでおりますが、aという１つの配列だけのでnp.savetxtで書く際にx,y,radiusという３つのデータをどのように表記すればいいかわからず質問するに至りました

[ppaul]

np.savetxt(f_handle, np.array([[x,y,radius]]) , delimiter=",", fmt="%.0f")
でよいはずです。