計算で使うデータは’*.png’で(001.png,002.png,・・・・)とPNGごとに行います。
for文で計算させています。
その後、計算結果はテキストに変換され(001.txt,002.txt,・・・・)のように出力されます。
このテキストデータを合体させる方法はありますでしょうか。
例えば001.txtは(1,1)002.txtは(2,9)のようなデータが入っているとき
出力では
1,1
2,9
となるようなテキストデータにしてほしいです。
現在のコードも載せておきます。
python
1import train 2import sys 3from PIL import Image 4import numpy as np 5import os 6 7input_dir = 'images' 8categories = [name for name in os.listdir(input_dir)] 9 10 11target_dir = sys.argv[1] if len(sys.argv) >= 2 else '.' 12 13for filename in os.listdir(target_dir): 14 path_in = os.path.join(target_dir, filename) 15 if not (os.path.isfile(path_in) and path_in.endswith('.png')): 16 continue 17 path_out = path_in.replace('.png', '.txt') 18 print('{} -> {} ...'.format(path_in, path_out)) 19 with open(path_in) as fin, open(path_out, 'w') as fout: 20 img = Image.open(filename).convert('RGB') ## Gray->L, RGB->RGB 21 img = img.resize((50, 50)) 22 x = np.array(img, dtype=np.float32) 23 x = x / 255. 24 x = x[None, ...] 25 model = train.TrainModel().train(x.shape[1:]) 26 model.load_weights('namimodel.h5') 27 predict = model.predict(x) 28 for pre in predict: 29 y = pre.argmax() 30 print("hantei : ", categories[y]) 31 fout.write(str(pre))
バッドをするには、ログインかつ
こちらの条件を満たす必要があります。