我在运行train.py时遇到问题,感到困惑

时间:2020-01-24 20:58:03

标签: python python-3.x tensorflow

我已经运行并成功编译了该网站的所有代码: Medium article 我到了培训点,然后运行命令:

python train.py --logtostderr --train_dir=training/ --pipeline_config_path=training/ssdlite_mobilenet_v2_coco.config

但是我收到一个错误,它找不到train.py,我读到它已经过时了,我应该运行model_main.py,当我这样做时,我得到这个错误

2020-01-24 15:43:27.488352:我tensorflow / stream_executor / platform / default / dso_loader.cc:44]成功打开动态库cudart64_101.dll 追溯(最近一次通话): 文件“ model_main.py”,第24行,在 从object_detection导入model_hparams ModuleNotFoundError:没有名为“ object_detection”的模块

我在使用python 3.7的Windows 10 x64计算机上。我已经尝试过将我的工作目录放在models文件夹之外的对象检测所在的位置,而将其直接放在所有文件所在的object-detection文件夹内。如果有人可以帮助,那就太好了!谢谢!!!

1 个答案:

答案 0 :(得分:1)

在Windows上运行tensorflow并不容易,但请查看https://github.com/tensorflow/models/issues/2031

上的说明

TLDR:

  • 转到模型/研究目录
  • 运行以下命令

     python setup.py build
     python setup.py install
    
  • 如果您正在使用Anaconda,请在同一目录中运行它们

     conda develop .
     conda develop slim/