前提・実現したいこと
こちらのサイトを参考にpix2pixを使用し、200epochで学習を終えました。
モデルはsnapshot_iter_23000.npzで保存してあります。
このモデルを使用し21枚分の画像を予測生成したく、下記プログラムをまわましました。
発生している問題・エラーメッセージ
警告なのですが、
A batch with no more than one sample has been given to F.batch_normalization. F.batch_normalization will always output a zero tensor for such batches. This could be caused by incorrect configuration in your code (suchas running evaluation while chainer.config.train=True), but could also happen in the last batch of training if non-repeating iterator is used.
と表示され、また、いざモデルを使用し画像を生成してみると何も描かれていないグレーの画像が出来上がるだけでした。
該当のソースコード
python
1# -*- coding: utf-8 -*- 2from __future__ import print_function 3import argparse 4import os 5 6import chainer 7from chainer import training 8 9from net3 import Discriminator 10from net3 import Encoder 11from net3 import Decoder 12from updater3 import FacadeUpdater 13 14from facade_dataset3 import FacadeDataset 15from facade_dataset3 import FacadeTestDataset 16from facade_visualizer3 import out_image 17 18import shutil 19 20 21def main(): 22 parser = argparse.ArgumentParser(description='chainer implementation of pix2pix') 23 parser.add_argument('--gpu', '-g', type=int, default=1, 24 help='GPU ID (negative value indicates CPU)') 25 parser.add_argument('--seed', type=int, default=0, 26 help='Random seed') 27 parser.add_argument('--model', '-m', default='./result/snapshot_iter_20000.npz', 28 help='model snapshot') 29 parser.add_argument('--input', '-i', default='sample.jpg', 30 help='input jpg') 31 args = parser.parse_args() 32 33 print('GPU: {}'.format(args.gpu)) 34 35 # Set up a neural network to train 36 enc = Encoder(in_ch=3) 37 dec = Decoder(out_ch=3) 38 dis = Discriminator(in_ch=3, out_ch=3) 39 40 if args.gpu >= 0: 41 chainer.cuda.get_device(args.gpu).use() # Make a specified GPU current 42 enc.to_gpu() # Copy the model to the GPU 43 dec.to_gpu() 44 dis.to_gpu() 45 46 # Setup an optimizer 47 def make_optimizer(model, alpha=0.0002, beta1=0.5): 48 optimizer = chainer.optimizers.Adam(alpha=alpha, beta1=beta1) 49 optimizer.setup(model) 50 optimizer.add_hook(chainer.optimizer.WeightDecay(0.00001), 'hook_dec') 51 return optimizer 52 opt_enc = make_optimizer(enc) 53 opt_dec = make_optimizer(dec) 54 opt_dis = make_optimizer(dis) 55 56 # generate_tmpってディレクトリが存在したら、消す 57 if os.path.exists('generate_tmp'): 58 shutil.rmtree('generate_tmp') 59 60 # generate_tmpっての作るお 61 os.mkdir('generate_tmp') 62 # 上のargsにあるsample.jpgをgenerate_tmpにtmp.jpgとしてコピー 63 shutil.copyfile(args.input,'generate_tmp/tmp.jpg') 64 test_d = FacadeDataset('generate_tmp/', 'generate_tmp/') 65 test_iter = chainer.iterators.SerialIterator(test_d, 1) 66 67 68 # Set up a trainer 69 updater = FacadeUpdater( 70 models=(enc, dec, dis), 71 iterator={}, 72 optimizer={'enc': opt_enc, 'dec': opt_dec,'dis': opt_dis}, 73 device=args.gpu) 74 trainer = training.Trainer(updater, (200, 'epoch'), out='generate/') 75 chainer.serializers.load_npz(args.model, trainer) 76 77 78 out_image( 79 updater, enc, dec, 80 1, 1, args.seed, 'generate/',args.input,True,test_iter)(trainer) 81 82 83if __name__ == '__main__': 84 main() 85
試したこと
実行する際に--mode testと打ったりしましたがグレー画像しか表示されず、警告も消えません。
また、chainer.config.train = Trueの場所も見つかりません。
python
1chainer.config.train = False
こちらを書いたところ、警告は消えたのですが未だグレーの画像が生成されます…。
回答1件
あなたの回答
tips
プレビュー
バッドをするには、ログインかつ
こちらの条件を満たす必要があります。