我在尝试按照TensorFlow用于诗人的ML图像分类指南进行操作时,无法重新训练我的数据集,因为遇到了以下错误:
Traceback (most recent call last): File "C:\Python\Python37\lib\runpy.py", line 193, in _run_module_as_main "__main__", mod_spec) File "C:\Python\Python37\lib\runpy.py", line 85, in _run_code exec(code, run_globals) File "C:\Users\I352778\tensorflow-for-poets-2\tensorflow-for-poets-2\scripts\retrain.py", line 1326, in <module> tf.app.run(main=main, argv=[sys.argv[0]] + unparsed)AttributeError: module 'tensorflow' has no attribute 'app'
我理解这是由于使用了v1功能而导致的问题,我现在使用的是TensorFlow v2,但仍不确定如何解决这个问题。我使用的训练代码如下:
C:\Users\I352778\tensorflow-for-poets-2\tensorflow-for-poets-2>python -m scripts.retrain ^More? --bottleneck_dir=tf_files/bottlenecks ^More? --model_dir=tf_files/models/%ARCHITECTURE% ^More? --summaries_dir=tf_files/training_summaries/%ARCHITECTURE% ^More? --output_graph=tf_files/retrained_graph.pb ^More? --output_labels=tf_files/retrained_labels.txt ^More? --architecture=%ARCHITECTURE% ^More? --image_dir=tf_files/C:\Users\I352778\Pictures\dataset-resized
为了克服我在尝试实现v1功能时遇到的问题,我应该输入什么代码呢?
回答:
根据TensorFlow 2.0的文档,”app“属性已被移除:https://www.tensorflow.org/guide/effective_tf2
“在TF 2.0中,许多API要么被移除,要么被移动。主要变化包括移除tf.app、tf.flags和tf.logging…”。要解决这个问题,可以卸载TensorFlow 2并安装版本1,或者更改你的代码片段。我不确定绕过这个问题是否对你有很大帮助,因为TensorFlow是一个非常敏感的环境,特别是在与Python一起使用时,尤其是在版本方面。所以我认为稍微更改代码会更好。