训练 tensorflow model 时出错：在“Tensorflow”中找不到“__main__”模块

Question

我正在尝试在 jupyter 笔记本中的 windows 上训练 tensorflow model。 环境和object检测API安装正确。

我在配置路径后运行此命令：

print("""python {}/research/object_detection/model_main_tf2.py --model_dir={}/{} --pipeline_config_path={}/{}/pipeline.config --num_train_steps=5000""".format(APIMODEL_PATH, MODEL_PATH,CUSTOM_MODEL_NAME,MODEL_PATH,CUSTOM_MODEL_NAME))

并得到这个 output：

python Tensorflow / models / research / object_detection / model_main_tf2.py --model_dir = Tensorflow / workspace / models / my_ssd_mobnet --pipeline_config_path = Tensorflow / workspace / models / my_ssd_mobnet / pipeline.config --num_train_steps = 3000

但是，当我实际尝试使用相同的 output 在我的终端中训练 model 时，它给了我错误：

C:\ProgramData\Anaconda3\python.exe：在“Tensorflow”中找不到“ main ”模块

有人可以提供一些解决此错误的方法吗？

Answer 1

您需要 go 到此路径“tensorflow/models/research”并将此行放在命令行中：

cd /path-to/models-master/research/
python setup.py build
python setup.py install

然后这样做以避免额外的错误

cd /path-to/models-master/research/slim
python setup.py build
python setup.py install

问候。