やりたいこと
①プログラム実行時に引数をファルダ名にする、もしくはプログラム内でフォルダを指定
②ファルダ内すべてのpngファイルにプログラムを実行
下のコードはグレースケールで画像を読み込んで、ハミング距離を出力するプログラムです。
このプログラムの実行は
入力:$python test.py 出力:[0 1 0 1 0 . . . 0] ・ ・ ・ ハミング距離:7159 ハミング距離:19430
となります。
これをフォルダの引数をとって、
入力: python test.py fold1 fold2 fold3
で実行できるようにしたいです。
もしくはプログラム内でフォルダを指定して、ファルダ内のファイルすべてにプログラムを実行したいです。
ファイルの引数、ファイル取得の関数はできたのですが、#11をどのように書きかえればいいのか分かりません。引数で指定したフォルダ内のファイルを1つ1つループで読み込み、ハミング距離を出力したいです。
どなたか詳しい方、ご助力お願いいたします。
#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)
まだ回答がついていません
会員登録して回答してみよう