Google Colab 中的 Tensorflow Slim：缺少“train_dir”

问题描述

这个问题是我在运行 train.py 而不是 model_main_tf2.py 时遇到的替代错误。这是 model_main_tf2.py 出现的错误：

Tensorflow 2 in Google Colab: Invoked with: None,524288

此项目正在 Windows 10 中运行。

我正在学习在 Google Colab 中使用 Tensorflow 2 制作我自己的自定义模型的教程：https://medium.com/analytics-vidhya/custom-tensorflow-object-detection-model-on-google-colab-4402b8e7ee7e

我调用了这个函数：

python train.py -- logtostderr -- train_dir=research/ -- research/

我收到此错误：

021-07-07 17:02:40.026148: I tensorflow/stream_executor/platform/default/dso_loader.cc:53] 成功打开动态库 libcudart.so.11.0 警告:tensorflow:From /usr/local/lib/python3.7/dist-packages/absl/app.py:251: main (from ma​​in) 已弃用，将在未来版本中删除. 更新说明： 使用 object_detection/model_main.py。 W0707 17:02:53.310447 140362781439872 deprecation.py:336] 来自 /usr/local/lib/python3.7/dist-packages/absl/app.py:251：main（来自 ma​​in）是已弃用，将在未来版本中删除。 更新说明： 使用 object_detection/model_main.py。 回溯（最近一次调用最后一次）： 文件“train.py”，第 186 行，在 tf.app.run() 运行中的文件“/usr/local/lib/python3.7/dist-packages/tensorflow/python/platform/app.py”，第 40 行 _run(main=main，argv=argv，flags_parser=_parse_flags_tolerate_undef) 运行中的文件“/usr/local/lib/python3.7/dist-packages/absl/app.py”，第 303 行 _run_main(main,args) 文件“/usr/local/lib/python3.7/dist-packages/absl/app.py”，第 251 行，在 _run_main sys.exit(main(argv)) 文件“/usr/local/lib/python3.7/dist-packages/tensorflow/python/util/deprecation.py”，第337行，在new_func 返回 func(*args,**kwargs) 文件“train.py”，第 91 行，在主目录中 断言 FLAGS.train_dir，缺少“train_dir”。 断言错误：train_dir 丢失。

我在模型文件夹中放置了火车文件夹、火车和测试记录的副本。它没有帮助。使用下面的代码将 PythonPath 设置为模型以避免模块“官方”未找到错误：

导入操作系统 os.environ['PYTHONPATH'] += ":/content/gdrive/MyDrive/models" 导入系统 sys.path.append("/content/gdrive/MyDrive/models")

我在网站上查过这个，但有用户建议我改用 model_main_tf2.py。我不能让 train.py 和 model_main_tf2.py 工作。感谢您的阅读，我希望能尽快解决这两个错误。

解决方法

暂无找到可以解决该程序问题的有效方法，小编努力寻找整理中！

如果你已经找到好的解决方法，欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@）