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

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

python
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)