以下のようなファイルがあります。
import ic_module as ic import os.path as op i = 0 for filename in ic.FileNames : # ディレクトリ名入力 while True : dirname = input(">>「" + ic.ClassNames[i] + "」の画像のあるディレクトリ : ") if op.isdir(dirname) : break print(">> そのディレクトリは存在しません!") # 関数実行 ic.PreProcess(dirname, filename, var_amount=3) i += 1
ここでは、いちいちデータセットのあるディレクトリを入力しなくてはいけないので次のように変更できないものかなと思っております。
dir = ["いぬ","ねこ","さる"] dirname = input(">>「" + ic.ClassNames[i] + "」の画像のあるディレクトリ : ") if op.isdir("c\user\"+dir) : break print(">> そのディレクトリは存在しません!")
if op.isdir("c:\user"+dir[0])
こうすると、ディレクトリの指定エラーが出ます。
どのように書けば、正しくディレクトリを指定できるのでしょうか。
ic.classNamesはClassNames = ["うさぎ", "いぬ", "ねこ"]となっております。
初心者の質問で申し訳ないのですが、どうか教えていただけると幸いです。
よろしくお願いいたします。
読み込み部分
def PreProcess(dirname, filename, var_amount=3): num = 0 arrlist = [] files = glob.glob(dirname + "/*.jpg") for imgfile in files: img = load_img(imgfile, target_size=(hw["height"], hw["width"])) # 画像ファイルの読み込み array = img_to_array(img) / 255 # 画像ファイルのnumpy化 arrlist.append(array) # numpy型データをリストに追加 for i in range(var_amount-1): arr2 = array arr2 = random_rotation(arr2, rg=360) arrlist.append(arr2) # numpy型データをリストに追加 num += 1 nplist = np.array(arrlist) np.save(filename, nplist) print(">> " + dirname + "から" + str(num) + "個のファイル読み込み成功")
回答1件
あなたの回答
tips
プレビュー