AttributeError: 'OrientedRepPointsHead' object has no attribute 'show_points'

[hukaixuan19970627]

I have splited my DOTA dataset by using prepare_dota1_ms.py，then i tried to test OrientedRepPoints. But something doesnot work，can you help me?

---

root@train-orientedreppoints-0:/data1/OrientedRepPoints# python tools/test.py configs/dota/orientedrepoints_r50_demo.py work_dirs/orientedreppoints_r50_demo/epoch_40.pth --out work_dirs/orientedreppoints_r50_demo/results.pkl loading annotations into memory... Done (t=0.00s) creating index... index created! [ ] 0/4, elapsed: 0s, ETA:Traceback (most recent call last): File "tools/test.py", line 170, in main() File "tools/test.py", line 148, in main outputs = single_gpu_test(model, data_loader, args.show) File "/data1/OrientedRepPoints/mmdet/apis/test.py", line 19, in single_gpu_test result = model(return_loss=False, rescale=not show, data) File "/usr/local/miniconda3/lib/python3.7/site-packages/torch/nn/modules/module.py", line 532, in call result = self.forward(*input, *kwargs) File "/usr/local/miniconda3/lib/python3.7/site-packages/torch/nn/parallel/data_parallel.py", line 150, in forward return self.module(inputs[0], kwargs[0]) File "/usr/local/miniconda3/lib/python3.7/site-packages/torch/nn/modules/module.py", line 532, in call result = self.forward(*input, *kwargs) File "/data1/OrientedRepPoints/mmdet/core/fp16/decorators.py", line 49, in new_func return old_func(args, kwargs) File "/data1/OrientedRepPoints/mmdet/models/detectors/base.py", line 149, in forward return self.forward_test(img, img_metas, kwargs) File "/data1/OrientedRepPoints/mmdet/models/detectors/base.py", line 130, in forward_test return self.simple_test(imgs[0], img_metas[0], *kwargs) File "/data1/OrientedRepPoints/mmdet/models/detectors/orientedreppoints_detector.py", line 38, in simple_test bbox_list = self.bbox_head.get_bboxes(bbox_inputs) File "/data1/OrientedRepPoints/mmdet/models/anchor_heads/orientedreppoints_head.py", line 644, in get_bboxes scale_factor, cfg, rescale, nms) File "/data1/OrientedRepPoints/mmdet/models/anchor_heads/orientedreppoints_head.py", line 717, in get_bboxes_single cfg.max_per_img, multi_reppoints=mlvl_reppoints if self.show_points else None) File "/usr/local/miniconda3/lib/python3.7/site-packages/torch/nn/modules/module.py", line 576, in getattr type(self).name, name)) AttributeError: 'OrientedRepPointsHead' object has no attribute 'show_points'

[hukaixuan19970627]

BTW ： My DOTA dataset just have 4 images

[LiWentomng]

@hukaixuan19970627 I update the script of orientedreppoints_head.py . Please redownload this script, and try again.

[hukaixuan19970627]

@hukaixuan19970627 I update the script of orientedreppoints_head.py . Please redownload this script, and try again.

Thank you！That new file works. Could you please upload another ‘show_learning_points.py’ to visualize the detect-results?

[LiWentomng]

@hukaixuan19970627 OK, I will soon upload the code to visualize the learning points of detection results.

[LiWentomng]

@hukaixuan19970627 The visualization code is to get the learning points and boxes of detection results.

[hukaixuan19970627]

@hukaixuan19970627 The visualization code is to get the learning points and boxes of detection results.

Thanks for your code!!!!!