facebookresearch / detectron2

Detectron2 is a platform for object detection, segmentation and other visual recognition tasks.
https://detectron2.readthedocs.io/en/latest/
Apache License 2.0
30.55k stars 7.49k forks source link

rotated rcnn error #3834

Closed rlaehddls9 closed 2 years ago

rlaehddls9 commented 2 years ago

Hello, I am running rotated faster rcnn with detectron2 in radar dataset called radiate. radiate: https://github.com/marcelsheeny/radiate_sdk And I used this code. (https://github.com/marcelsheeny/radiate_sdk/blob/master/vehicle_detection/train.py) Also yaml file is below. (faster_rrcnn_R_50_FPN_3x.yaml) I think they called rotated rcnn to rrcnn.

BASE: "Base-RCNN-FPN.yaml" MODEL: PIXEL_MEAN: [103.530, 116.280, 123.675] # Pixel mean of imagenet dataset R/G/B? WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-50.pkl" MASK_ON: False RESNETS: DEPTH: 50 PROPOSAL_GENERATOR: NAME: "RRPN" RPN: HEAD_NAME: "StandardRPNHead" BBOX_REG_WEIGHTS: (1.0, 1.0, 1.0, 1.0, 1.0) ANCHOR_GENERATOR: NAME: "RotatedAnchorGenerator" ANGLES: [[0, 30, 60, 90]] ASPECT_RATIOS: [[0.5, 1.0, 2.0]] ROI_HEADS: NAME: "RROIHeads" NUM_CLASSES: 1 # number of foreground classes ROI_BOX_HEAD: POOLER_TYPE: "ROIAlignRotated" BBOX_REG_WEIGHTS: (10.0, 10.0, 5.0, 5.0, 1.0) SOLVER: STEPS: (210000, 250000)

When inference is made based on the learning result (AP50: 23), a bounding box offset like the image below occurs.

https://user-images.githubusercontent.com/79086973/148163487-fd64186f-6949-406d-a54e-c32992b9cbfc.mp4

It was confirmed that the label of the dataset itself was well done. When running rotated rcnn on detectron2 code, is the offset reflected?

Please advise on how to solve the problem.

github-actions[bot] commented 2 years ago

You've chosen to report an unexpected problem or bug. Unless you already know the root cause of it, please include details about it by filling the issue template. The following information is missing: "Instructions To Reproduce the Issue and Full Logs"; "Your Environment";