search

csuhan / s2anet

Official code of the paper "Align Deep Features for Oriented Object Detection"
376 stars 90 forks source link

No module named 'mmdet.ops.box_iou_rotated_diff.sort_vertices_cuda' #121

Closed adhirajghosh closed 2 years ago

adhirajghosh commented 2 years ago

While running tools/train.py, I run into an error in the following command: from mmdet.apis import (get_root_logger, init_dist, set_random_seed, train_detector) Kindly suggest a fix to the bug.

The following is the error sequence: *File "train.py", line 15, in from mmdet.apis import (get_root_logger, init_dist, set_random_seed, train_detector) File "/workspace/s2anet/mmdet/apis/init.py", line 2, in from .inference import (inference_detector, init_detector, show_result, File "/workspace/s2anet/mmdet/apis/inference.py", line 13, in from mmdet.datasets.pipelines import Compose File "/workspace/s2anet/mmdet/datasets/init.py", line 2, in from .cityscapes import CityscapesDataset File "/workspace/s2anet/mmdet/datasets/cityscapes.py", line 1, in from .coco import CocoDataset File "/workspace/s2anet/mmdet/datasets/coco.py", line 4, in from .custom import CustomDataset File "/workspace/s2anet/mmdet/datasets/custom.py", line 7, in from .pipelines import Compose File "/workspace/s2anet/mmdet/datasets/pipelines/init.py", line 7, in from .transforms import (Albu, Expand, MinIoURandomCrop, Normalize, Pad, File "/workspace/s2anet/mmdet/datasets/pipelines/transforms.py", line 11, in from mmdet.core.evaluation.bbox_overlaps import bbox_overlaps File "/workspace/s2anet/mmdet/core/init.py", line 1, in from .anchor import # noqa: F401, F403 File "/workspace/s2anet/mmdet/core/anchor/init.py", line 3, in from .anchor_target import anchor_inside_flags, anchor_target, unmap, images_to_levels File "/workspace/s2anet/mmdet/core/anchor/anchor_target.py", line 3, in from ..bbox import PseudoSampler, assign_and_sample, build_assigner, build_bbox_coder File "/workspace/s2anet/mmdet/core/bbox/init.py", line 2, in from .assigners import AssignResult, BaseAssigner, MaxIoUAssigner File "/workspace/s2anet/mmdet/core/bbox/assigners/init.py", line 1, in from .approx_max_iou_assigner import ApproxMaxIoUAssigner File "/workspace/s2anet/mmdet/core/bbox/assigners/approx_max_iou_assigner.py", line 3, in from .max_iou_assigner import MaxIoUAssigner File "/workspace/s2anet/mmdet/core/bbox/assigners/max_iou_assigner.py", line 6, in from ..iou_calculators import build_iou_calculator File "/workspace/s2anet/mmdet/core/bbox/iou_calculators/init.py", line 3, in from .iou2d_calculator_rotated import BboxOverlaps2D_rotated, bbox_overlaps_rotated File "/workspace/s2anet/mmdet/core/bbox/iou_calculators/iou2d_calculator_rotated.py", line 1, in from mmdet.ops.box_iou_rotated import box_iou_rotated File "/workspace/s2anet/mmdet/ops/init.py", line 15, in from .box_iou_rotated_diff import box_iou_rotated_differentiable File "/workspace/s2anet/mmdet/ops/box_iou_rotated_diff/init.py", line 1, in from .box_iou_rotated_diff import box_iou_rotated_differentiable File "/workspace/s2anet/mmdet/ops/box_iou_rotated_diff/box_iou_rotated_diff.py", line 6, in from .box_intersection_2d import oriented_box_intersection_2d File "/workspace/s2anet/mmdet/ops/box_iou_rotated_diff/box_intersection_2d.py", line 10, in from .sort_vertices_cuda import sort_vertices_forward ModuleNotFoundError: No module named 'mmdet.ops.box_iou_rotated_diff.sort_vertices_cuda'**

csuhan commented 2 years ago

I suggest you to re-compile the cuda module mmdet.ops.box_iou_rotated_diff. Also, check the compatibility of your pytorch & cuda version.