Mmdetection: ImportError: cannot import name 'deform_conv_cuda' from 'mmdet.ops.dcn'

Created on 7 May 2019  路  6Comments  路  Source: open-mmlab/mmdetection

uge@zhuge-All-Series:~/hs/MobilenetV2-Retina-Pytorch-master/tools$ python train.py '/home/zhuge/hs/MobilenetV2-Retina-Pytorch-master/configs/pascal_voc/shuffle_retina_voc.py' --work_dir '/home/zhuge/hs/MobilenetV2-Retina-Pytorch-master/weights/shuffle_retina/1.0/640' --seed 1314
Traceback (most recent call last):
File "train.py", line 7, in
from mmdet.datasets import get_dataset
File "/home/zhuge/hs/MobilenetV2-Retina-Pytorch-master/mmdet/datasets/__init__.py", line 1, in
from .custom import CustomDataset
File "/home/zhuge/hs/MobilenetV2-Retina-Pytorch-master/mmdet/datasets/custom.py", line 11, in
from .extra_aug import ExtraAugmentation
File "/home/zhuge/hs/MobilenetV2-Retina-Pytorch-master/mmdet/datasets/extra_aug.py", line 5, in
from mmdet.core.evaluation.bbox_overlaps import bbox_overlaps
File "/home/zhuge/hs/MobilenetV2-Retina-Pytorch-master/mmdet/core/__init__.py", line 6, in
from .post_processing import * # noqa: F401, F403
File "/home/zhuge/hs/MobilenetV2-Retina-Pytorch-master/mmdet/core/post_processing/__init__.py", line 1, in
from .bbox_nms import multiclass_nms
File "/home/zhuge/hs/MobilenetV2-Retina-Pytorch-master/mmdet/core/post_processing/bbox_nms.py", line 3, in
from mmdet.ops.nms import nms_wrapper
File "/home/zhuge/hs/MobilenetV2-Retina-Pytorch-master/mmdet/ops/__init__.py", line 1, in
from .dcn import (DeformConv, DeformRoIPooling, DeformRoIPoolingPack,
File "/home/zhuge/hs/MobilenetV2-Retina-Pytorch-master/mmdet/ops/dcn/__init__.py", line 1, in
from .functions.deform_conv import deform_conv, modulated_deform_conv
File "/home/zhuge/hs/MobilenetV2-Retina-Pytorch-master/mmdet/ops/dcn/functions/deform_conv.py", line 5, in
from .. import deform_conv_cuda
ImportError: cannot import name 'deform_conv_cuda' from 'mmdet.ops.dcn' (/home/zhuge/hs/MobilenetV2-Retina-Pytorch-master/mmdet/ops/dcn/__init__.py)

Most helpful comment

PyTorch version 1.1 mmdetection should be installed with "python setup.py develop or pip install -v -e ." ! !
PyTorch-0.4.1 version mmdetection is compiled with "./compile.sh" and installed with python setup.py install!

All 6 comments

I forgot to run ./compile. Sh.

Hallo 237014845 i have the same problem.
Where i must run ./compile. Sh.?

Hallo 237014845 i have the same problem.
Where i must run ./compile. Sh.?

Try running python3 setup.py install in the project root directory.

PyTorch version 1.1 mmdetection should be installed with "python setup.py develop or pip install -v -e ." ! !
PyTorch-0.4.1 version mmdetection is compiled with "./compile.sh" and installed with python setup.py install!

@ash11sh Where can I get pytorch-0.4.1 version mmdetection?

@ash11sh Where can I get pytorch-0.4.1 version mmdetection?

I mean here two versions of pytorch had different installation setup

Was this page helpful?
0 / 5 - 0 ratings