【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 9, in <module>
    from pr_handler_new import PicRecHandler
  File "/home/mmxsrt/sg-pcr-client/pr_handler_new.py", line 4, in <module>
    from mmdet.apis import init_detector, inference_detector, show_result
  File "/mmdetection/mmdet/__init__.py", line 1, 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.0-0 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 1.0rc0+unknown
    Uninstalling mmdet-1.0rc0+unknown:
      Successfully uninstalled mmdet-1.0rc0+unknown
  Running setup.py develop for mmdet
Successfully installed mmdet

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