在google colab中训练对象检测模型时没有模块名称部署错误

时间:2018-05-25 10:55:10

标签: tensorflow google-colaboratory

我是Malathi,我尝试使用google colab训练对象检测模型。在训练此模型时,我得到了错误没有模块名称部署。但我在tensorflow-gpu中训练这个相同的文件它会正常工作。

/usr/local/lib/python3.6/dist-packages/h5py/__init__.py:36: FutureWarning: Conversion of the second argument of issubdtype from `float` to `np.floating` is deprecated. In future, it will be treated as `np.float64 == np.dtype(float).type`.
  from ._conv import register_converters as _register_converters
Traceback (most recent call last):
  File "train.py", line 49, in <module>
    from object_detection import trainer
  File "/content/my_drive/tut_kaggle/zero/models/research/object_detection/trainer.py", line 33, in <module>
    from deployment import model_deploy
ModuleNotFoundError: No module named 'deployment'

这是培训代码

!unset DISPLAY XAUTHORITY
!xvfb-run python3 train.py --logtostderr --train_dir=training/ --pipeline_config_path=training/ssd_mobilenet_v1_pets.config

1 个答案:

答案 0 :(得分:0)

尝试此操作以在PYTHONPATH中设置纤细。它应该工作。

%set_env PYTHONPATH=/content/models/research:/content/models/research/slim