我是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
答案 0 :(得分:0)
尝试此操作以在PYTHONPATH中设置纤细。它应该工作。
%set_env PYTHONPATH=/content/models/research:/content/models/research/slim