search

LiWentomng / BoxInstSeg

A toolbox for box-supervised instance segmentation.
Apache License 2.0
407 stars 38 forks source link

No module named 'mmdet.ops.pairwise.pairwise_ext' #4

Open XJTU-Haolin opened 1 year ago

XJTU-Haolin commented 1 year ago

(openmmlab) @amax:/data0/ZHL/project/DET2D/rope2D/BoxInstSeg$ CUDA_VISIBLE_DEVICES=5 python tools/train.py configs/boxinst/boxinst_r50_fpn_1x_coco.py Traceback (most recent call last): File "tools/train.py", line 17, in from mmdet.apis import init_random_seed, set_random_seed, train_detector File "/data0/ZHL/project/DET2D/rope2D/BoxInstSeg/mmdet/apis/init.py", line 2, in from .inference import (async_inference_detector, inference_detector, File "/data0/ZHL/project/DET2D/rope2D/BoxInstSeg/mmdet/apis/inference.py", line 13, in from mmdet.datasets import replace_ImageToTensor File "/data0/ZHL/project/DET2D/rope2D/BoxInstSeg/mmdet/datasets/init.py", line 9, in from .utils import (NumClassCheckHook, get_loading_pipeline, File "/data0/ZHL/project/DET2D/rope2D/BoxInstSeg/mmdet/datasets/utils.py", line 11, in from mmdet.models.roi_heads.mask_heads import FusedSemanticHead File "/data0/ZHL/project/DET2D/rope2D/BoxInstSeg/mmdet/models/init.py", line 7, in from .dense_heads import * # noqa: F401,F403 File "/data0/ZHL/project/DET2D/rope2D/BoxInstSeg/mmdet/models/dense_heads/init.py", line 4, in from .condinst_head import CondInstBoxHead, CondInstSegmHead, CondInstMaskBranch, CondInstMaskHead File "/data0/ZHL/project/DET2D/rope2D/BoxInstSeg/mmdet/models/dense_heads/condinst_head.py", line 14, in from mmdet.ops.pairwise import pairwise_nlog File "/data0/ZHL/project/DET2D/rope2D/BoxInstSeg/mmdet/ops/pairwise/init.py", line 1, in from .pairwise import pairwise_nlog File "/data0/ZHL/project/DET2D/rope2D/BoxInstSeg/mmdet/ops/pairwise/pairwise.py", line 3, in from .pairwise_ext import pairwise_nlog_forward, pairwise_nlog_backward ModuleNotFoundError: No module named 'mmdet.ops.pairwise.pairwise_ext'

Someone meet same problem? Where is pairwise_ext?

LiWentomng commented 1 year ago

@XJTU-Haolin
Hello, the pairwise is a CUDA operation used for boxinst, which is needed to be compliled by the followings:

cd mmdet/ops/pairwise #compiling for the opeartion using in boxinst
python setup.py build develop 
cd mmdet/ops/tree_filter #compiling for the opeartion using in boxlevelset
python setup.py build develop

Or

bash setup.sh

XJTU-Haolin commented 1 year ago

@XJTU-Haolin Hello, the pairwise is a CUDA operation used for boxinst, which is needed to be compliled by the followings:

cd mmdet/ops/pairwise #compiling for the opeartion using in boxinst
python setup.py build develop 
cd mmdet/ops/tree_filter #compiling for the opeartion using in boxlevelset
python setup.py build develop

Or

bash setup.sh

It is nice of your help! I have built but still met some other problems. I'll check later.

amarnath-asteria commented 1 year ago

any solution to the problem? i am getting the same error