python 画像を保存する複数の関数を任意で実行したい

Question

悩んでいること、分からないこと

以下のコードは、画像処理を行って処理後の画像を確認するために、保存しています。しかし、画像を保存するのには、僅かでも時間がかかります。それに、gray.jpg画像なんってめったに確認しません。出来れば、一か所にまとめて、「保存する/しない」を切り替えられるようにしたいのですが、自分の堅い頭ではどうコードを書くか分かりません。皆さんは、画像処理する時にどのように書いているのでしょうか。

# 余分な部分を黒色で塗りつぶす
img_rgb = cv2.rectangle(self.img_RGB, (0, 0), (130, 110), (0, 0, 0), -1)
cv2.imwrite("./image/img_rgb.jpg", img_rgb)
# グレースケール化
img_gray = cv2.cvtColor(img_rgb, cv2.COLOR_BGR2GRAY)
cv2.imwrite("./image/gray.jpg", img_gray)

# 白色を抽出する2値化を行う。閾値135で二値化
ret_val, img_thresh = cv2.threshold(img_gray, 135, 255, 0)
cv2.imwrite("./image/img_thresh.jpg", img_thresh)

# 最外輪郭を抽出
img_contours, contours, hierarchy = cv2.findContours(img_thresh, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
cv2.imwrite("./image/img_contours.jpg", img_contours)

Answer 1

一括で保存する／しないなら、次のように書けばいいです。

img_rgb = cv2.rectangle(self.img_RGB, (0, 0), (130, 110), (0, 0, 0), -1)
img_gray = cv2.cvtColor(img_rgb, cv2.COLOR_BGR2GRAY)
ret_val, img_thresh = cv2.threshold(img_gray, 135, 255, 0)
img_contours, contours, hierarchy = cv2.findContours(img_thresh, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)

if need_to_save:
    cv2.imwrite("./image/img_rgb.jpg", img_rgb)  
    cv2.imwrite("./image/gray.jpg", img_gray)
    cv2.imwrite("./image/img_thresh.jpg", img_thresh)
    cv2.imwrite("./image/img_contours.jpg", img_contours)

あとは、need_to_saveの指定の仕方を考えるだけです。
・ コード内に埋め込んでしまう
・ 保存するかどうか(y/n)入力を求める
・ オプションをコマンドラインで設定できるようにする

上にいくほど簡単／愚直で、下にいくほど柔軟／面倒な設計です。

---

一応、コマンドライン引数のサンプルを置いておきます。

# hoge.py

import argparse

parser = argparse.ArgumentParser()
parser.add_argument('-s', '--save', nargs='?', const='./image')
args = parser.parse_args()

if args.save:
    print(args.save)

"""実行例
>python hoge.py

>python hoge.py --save
./image

>python hoge.py --save "./result"
./result
"""

ついでに、画像保存時のフォーマットも指定できるようにしてみたり。

import argparse

parser = argparse.ArgumentParser()
parser.add_argument('-s', '--save', nargs='?', const='./image')
parser.add_argument('-f', '--format', default='jpg')
args = parser.parse_args()

if args.save:
    print(args.save)

print(args.format)

"""実行例
>python hoge.py --save "./result"
./result
jpg

>python hoge.py --save "./result" --format png
./result
png
"""

追記

かなり愚直な方法ではありますが、一個一個オプションを作るのが簡単でしょう。

import argparse

parser = argparse.ArgumentParser()
parser.add_argument('--s_rgb', action='store_true')
parser.add_argument('--s_gray', action='store_true')
parser.add_argument('--s_thresh', action='store_true')
parser.add_argument('--s_contours', action='store_true')
parser.add_argument('--s_all', action='store_true')
args = parser.parse_args()

if args.s_all or args.s_rgb:
    print('save rgb')
if args.s_all or args.s_gray:
    print('save gray')
if args.s_all or args.s_thresh:
    print('save thresh')
if args.s_all or args.s_contours:
    print('save contours')

"""実行例
>python hoge.py

>python hoge.py --s_all
save rgb
save gray
save thresh
save contours

>python hoge.py --s_rgb --s_contours
save rgb
save contours
"""

集合形式で引数を与えた方が綺麗そうですね。なんかいい方法あるのかな...
↓　ありました

さらに追記

こいつでどうだ。

import argparse

parser = argparse.ArgumentParser()
parser.add_argument('-s', '--save_set', default={})
args = parser.parse_args()

if 'r' in args.save_set:
    print('save rgb')
if 'g' in args.save_set:
    print('save gray')
if 't' in args.save_set:
    print('save thresh')
if 'c' in args.save_set:
    print('save contours')

"""実行例
>python hoge.py -s rc
save rgb
save contours

>python hoge.py

>python hoge.py -s r
save rgb
"""

こういう風にすれば、『全部保存』という指定も出来ます。...1文字しか減ってませんが。

parser.add_argument('-s', '--save_set', type=(lambda s: 'rgtc' if s == 'all' else s), default='')

Comments

[退会済みユーザー]

ご回答ありがとうございます。if文で書く方法が一番単純そうですね。

・ 保存するかどうか(y/n)入力を求める
これは、実行の途中で標準入力にユーザー入力するという意味でしょうか？
・ オプションをコマンドラインで設定できるようにする
これは、自分が初心者だからか初めて見ました。因みにもしimg_rgbのみを保存するとしたら、どのようにコマンドで実行すべきでしょうか?

[LouiS0616]

> これは、実行の途中で標準入力にユーザー入力するという意味でしょうか？
その通りです。
> もしimg_rgbのみを保存するとしたら、どのようにコマンドで実行すべきでしょうか?
うーん...
img_rgbだけを保存するオプションを追加するのが一番愚直です。
集合形式で保存する画像を指定するのもありですね。

[退会済みユーザー]

ご回答ありがとうございます。悩んだ結果、簡単なif文の方法に致します。