ファルダ名を引数にして、ファルダ内のすべてのファイルにプログラムを実行させるように修正したい

Question

やりたいこと
①プログラム実行時に引数をファルダ名にする、もしくはプログラム内でフォルダを指定
②ファルダ内すべてのpngファイルにプログラムを実行

下のコードはグレースケールで画像を読み込んで、ハミング距離を出力するプログラムです。
このプログラムの実行は

入力：＄python test.py
出力：[0 1 0 1 0 . . . 0]
　　　　　　　・
　　　　　　　・
　　　　　　　・
ハミング距離:7159
ハミング距離:19430

となります。
これをフォルダの引数をとって、

入力：
python test.py fold1 fold2 fold3

で実行できるようにしたいです。
もしくはプログラム内でフォルダを指定して、ファルダ内のファイルすべてにプログラムを実行したいです。

ファイルの引数、ファイル取得の関数はできたのですが、#11をどのように書きかえればいいのか分かりません。引数で指定したフォルダ内のファイルを１つ１つループで読み込み、ハミング距離を出力したいです。
どなたか詳しい方、ご助力お願いいたします。

#11---関数の呼び出し
ahash = average_hash('test1_2D_out/000002_OUT.png')
ahash1 = average_hash('test1_2D_out/000003_OUT.png')
ahash2 = average_hash('test1_mask_bara/2D/Target/000002.png')
hamming_ab,hamming_ac = hamming_dist(ahash,ahash1,ahash2)

test.py

from PIL import Image
import numpy as np
import os, re
import argparse
import sys
import pathlib

def get_filenames_of_path(path: pathlib.Path, ext: str = "*"):
    """Returns a list of files in a directory/path. Uses pathlib."""
    filenames = [file for file in path.glob(ext) if file.is_file()]
    filenames.sort()
    return filenames
  
def get_args():
    parser = argparse.ArgumentParser(description='Evaluate result',
                                     formatter_class=argparse.ArgumentDefaultsHelpFormatter)
    parser.add_argument('target_dir', type=str)
    parser.add_argument('result_dir', type=str)

    return parser.parse_args()
args = get_args()

result_dir = pathlib.Path.cwd() / args.result_dir
target_dir = pathlib.Path.cwd() / args.target_dir

#1---Average Hashの取得
def average_hash(fname, size = 256):
#2---画像データを開く
    img_data = Image.open(fname)
#3---グレースケールに変換
    img_data = img_data.convert('L')
#4---リサイズ
    img_data = img_data.resize((size, size), Image.ANTIALIAS)
#5---ピクセルデータを得る
    pixel_data = img_data.getdata()
#6---Numpyに変換
    pixels = np.array(pixel_data)
#7---二次元の配列に変換
    pixels = pixels.reshape((size, size))
#8---平均算出
    average = pixels.mean()
#9---値を1と0に変換
    out_pix = 1 * (pixels > average)
    return out_pix
#10---ハミング距離の算出
def hamming_dist(a, b, c):
    aa = a.reshape(1, -1)
    ab = b.reshape(1, -1)
    ac = c.reshape(1,-1)
    dist = (aa != ab).sum()
    dist1 = (aa != ac).sum()
    return dist,dist1
#11---関数の呼び出し
ahash = average_hash('test4_out_bara/a/02/00353_00354_01.png')
ahash1 = average_hash('test4_out_bara/a/02/00353_00354_01.png')
ahash2 = average_hash('test4_mask_bara/02/Target/out48_02.png')
hamming_ab,hamming_ac = hamming_dist(ahash,ahash1,ahash2)

#12---出力結果
print(ahash)
print("*****")
print(ahash1)
print("*****")
print(ahash2)
print("ハミング距離は",hamming_ab)
print("ハミング距離は",hamming_ac)

Comments

[TakaiY]

質問は何ですか？

sys.argvを使えばできますので、検索したり書籍を読んだりして試してみればよいと思います。試してみて、できなければそのつまづいたところについて質問しましょう。

[tiroha]

回答ありがとうございます。
質問を修正しました。調べたら、sys.argvを使えばできる気がするのですが、使い方がよく分からなくて。。

Answer 1

公式ドキュメント argparse --- コマンドラインオプション、引数、サブコマンドのパーサー
Pythonでコマンドライン引数を扱う方法（sys.argv, argparse）
を読んで、まずできるところまでご自身でやってください。

その上でわからないところがあるなら改めて質問してください。

Answer 2

sys.argv 使い方 - Google 検索

まずはこれをどうぞ。
これを読んでわからない所があれば、そのページを提示して聞いてください

---

1. 引数のディレクトリからファイルのリストを取得
2. ファイルリストからパスを取ってきて処理
3. リストからファイルが尽きるまでループ
4. おしまい

という手順になると思われますが、これのどれがわからないんでしょうか

Comments

[tiroha]

回答ありがとうございます。
質問したプログラムに追加しました。
ディレクトリの引数とファイル取得の関数は作成したのですが、＃11をどのように書きかえればいいのか分かりません。

[y_waiwai]

それをどうしたいのかを書かないとだれも答えれませんよ

[tiroha]

質問を修正しました。

[y_waiwai]

追記しました

[tiroha]

回答ありがとうございます。
手順をプログラムに落とし込むのに苦労しています。