Pyhtonを使って大量の画像を輪郭抽出するプログラムを組みたい。

Question

Canny検出器を使ったプログラムを作ったのですが、1枚ずつしか処理できず、効率よく繰り返し文などを使って大量の写真を処理したいと考えています。しかし、どこに繰り返し文を入れればよいかわからない状況です。

import numpy as np

def main():
    img = cv2.imread(r"C:\Users\niras\Desktop\NNC sotuken\2021-04-10\data\full.py\02.jpg")

    gray = cv2.cvtColor(img, cv2.COLOR_RGB2GRAY)

    edge2 = cv2.Canny(gray, 100,200)

    cv2.imwrite(r"C:\Users\niras\Desktop\2021-07-21\data\full.py\02.jpg", edge2)

    cv2.imshow('gray',edge2)
    cv2.waitKey(0)


if __name__ == "__main__":
 main()
コード

現状のコードはこちらになります。

Comments

[TakaiY]

このソースはこのままでは動きません。ということは、実際に動かしているソースだとは思えません。
実際のソースに差し替えをお願いします。

[k.nirasawa]

import cv2
import numpy as np

def main():
img = cv2.imread(r"C:\Users\niras\Desktop\NNC sotuken\2021-04-10\data\full.py\02.jpg")

gray = cv2.cvtColor(img, cv2.COLOR_RGB2GRAY)

edge2 = cv2.Canny(gray, 100,200)

cv2.imwrite(r"C:\Users\niras\Desktop\2021-07-21\data\full.py\02.jpg", edge2)

cv2.imshow('gray',edge2)
cv2.waitKey(0)

if __name__ == "__main__":
main()

すみません。こちらでした。

[jbpb0]

正しいコードはここに書くのではなく、質問のコードを編集してください

[k.nirasawa]

すいません。変更しました。

Answer 1

• 処理するファイルの一覧を取得するには、pathlib.glob や glob.globを使います。

• 出力するファイルの名前を入力するファイルから生成する手段を考えます。 パスの操作はpathlibが便利です。

• 処理するファイルのパスと出力するファイルのパスを受け取って、処理後のファイルを生成する関数を作ります。基本的には現在のmain関数と同じでいいでしょう。

• main処理として、処理するファイルの取得と、そのリストでループして、出力ファイル名の生成と、生成した関数の呼び出しをする処理を作ります。

Answer 2

TakaiYさんの回答で終わっていると思うのですが、全体構成がわからないようですので補足します。

def convert_image(in_filename, out_filename):
    '''    in_filename: str such as r"C:\Users\niras\Desktop\NNC sotuken\2021-04-10\data\full.py\02.jpg"
    out_filename: str such as r"C:\Users\niras\Desktop\2021-07-21\data\full.py\02.jpg"'''
#    何らかの処理

def convert_images(filelist):
    '''    filelist: whose each element is a list such as (in_filename, out_filename)'''
    for in_filename, out_filename in filelist:
        convert_image(in_filename, out_filename)

def get_in_filelist(input_dir, extension='jpg'):
    '''    input_dir: directory of input files
    Returns
    -------
    in_filelist: list of in_filenames'''
    from glob import glob
#    何らかの処理

def get_filelist(in_filelist, output_dir):
    '''    in_filelist: list of in_filenames
    input_dir: directory of input files
    Returns
    -------
    filelist: whose each element is a list such as (in_filename, out_filename)'''
    from os.path import basename, join
#    何らかの処理

def main(input_dir, output_dir):
    in_filelist = get_in_filelist(input_dir)
    filelist = get_filelist(in_filelist, output_dir)
    convert_images(filelist)

これを埋めていけば目的のコードになります。

Comments

[k.nirasawa]

回答ありがとうございました。やってみます。