【AI-人工智能-mmdetection】ModuleNotFoundError: No module named 'mmdet.version'

在集成 mmdetection 框架时遇到这样的问题。

ModuleNotFoundError: No module named 'mmdet.version'

mmdetection 框架搭建过程很容易忽略 编译过程

root@48ea75026556:/mmdetection# python sg-pcr-client/pcr_agant_new.py
Traceback (most recent call last):
  File "sg-pcr-client/pcr_agant_new.py", line , in <module>
    from pr_handler_new import PicRecHandler
  File "/home/mmxsrt/sg-pcr-client/pr_handler_new.py", line , in <module>
    from mmdet.apis import init_detector, inference_detector, show_result
  File "/mmdetection/mmdet/__init__.py", line , in <module>
    from .version import __version__, short_version
ModuleNotFoundError: No module named 'mmdet.version'
root@48ea75026556:/mmdetection# 

其实在mmdetection-master中 有个docker目录

里面很清楚的描述了搭建过程。贴上dockerfile代码

 ARG PYTORCH="1.1.0"
 ARG CUDA="10.0"
 ARG CUDNN="7.5"

 FROM pytorch/pytorch:${PYTORCH}-cuda${CUDA}-cudnn${CUDNN}-devel

 RUN apt-get update && apt-get install -y libglib2.- libsm6 libxrender-dev libxext6

 # Install mmdetection
 RUN conda install cython -y
 RUN git clone https://github.com/open-mmlab/mmdetection.git /mmdetection
 WORKDIR /mmdetection
 RUN pip install -e .

最后一步使我们本次解决问题的关键 安装编译

RUN pip install -e .

编译成功后

Installing collected packages: mmdet
  Found existing installation: mmdet .0rc0+unknown
    Uninstalling mmdet-.0rc0+unknown:
      Successfully uninstalled mmdet-.0rc0+unknown
  Running setup.py develop for mmdet
Successfully installed mmdet

此时表示已经完成安装和环境准备工作。可以调取模型和接下来的训练，推理，验证的过程了。