WXinlong / SOLO

SOLO and SOLOv2 for instance segmentation, ECCV 2020 & NeurIPS 2020.
Other
1.69k stars 307 forks source link

how to compile cuda file #152

Closed lushaoan closed 3 years ago

lushaoan commented 3 years ago

how to compile .cpp and .cu file in mmdet/ops/

I got this error when i running test_ins_vis.py

Traceback (most recent call last): File "tools/test_ins_vis.py", line 15, in from mmdet.core import coco_eval, results2json, wrap_fp16_model, tensor2imgs, get_classes File "/media/lsa/MobileDisk3/paper/mechine learning/instance segmentation/solo/SOLO/mmdet/core/init.py", line 6, in from .post_processing import * # noqa: F401, F403 File "/media/lsa/MobileDisk3/paper/mechine learning/instance segmentation/solo/SOLO/mmdet/core/post_processing/init.py", line 1, in from .bbox_nms import multiclass_nms File "/media/lsa/MobileDisk3/paper/mechine learning/instance segmentation/solo/SOLO/mmdet/core/post_processing/bbox_nms.py", line 3, in from mmdet.ops.nms import nms_wrapper File "/media/lsa/MobileDisk3/paper/mechine learning/instance segmentation/solo/SOLO/mmdet/ops/init.py", line 2, in from .dcn import (DeformConv, DeformConvPack, DeformRoIPooling, File "/media/lsa/MobileDisk3/paper/mechine learning/instance segmentation/solo/SOLO/mmdet/ops/dcn/init.py", line 1, in from .deform_conv import (DeformConv, DeformConvPack, ModulatedDeformConv, File "/media/lsa/MobileDisk3/paper/mechine learning/instance segmentation/solo/SOLO/mmdet/ops/dcn/deform_conv.py", line 10, in from . import deform_conv_cuda ImportError: cannot import name 'deform_conv_cuda'

kexinhandsome commented 3 years ago

i have this problem, too. have you solved this error?how?