尝试使用新类重新训练初始v3模型的最后一层。我已经准备了自己的一组图像,并存储在C:\tmp中。当我运行代码时
C:\tmp\image_retrain\tfClassifier\image_classification>python retrain.py --model_dir=C:\tmp\image_retrain\tfClassifier\image_classification\inception --image_dir=C:\tmp\image_retrain\tfClassifier\image_classification\styles --output_graph=C:\tmp\image_retrain\tfClassifier\image_classification\output_dir\all_ouput_stored_here.pb --how_many_training_steps 500它开始分类并创建瓶颈文件,但是我得到了这个错误,我不知道出了什么问题
File "retrain.py", line 967, in <module>
tf.app.run(main=main, argv=[sys.argv[0]] + unparsed)
File "C:\Users\aamo8\Anaconda3\envs\tensorflow-gpu\lib\site-packages\tensorflow\python\platform\app.py", line 48, in run
_sys.exit(main(_sys.argv[:1] + flags_passthrough))
File "retrain.py", line 750, in main
train_writer = tf.train.SummaryWriter(FLAGS.summaries_dir + '/train',
AttributeError: module 'tensorflow.python.training.training' has no attribute 'SummaryWriter'发布于 2017-12-14 21:06:35
SummaryWriter在TensorFlow 1.0中已被废弃。使用tensorflow.summary.FileWriter代替。
https://stackoverflow.com/questions/47821890
复制相似问题