search

Epiphqny / SOLOv2

SOLOv2: Dynamic, Faster and Stronger, achives 39.5mAP on coco test-dev (36 epochs result)
https://arxiv.org/abs/2003.10152
Apache License 2.0
244 stars 37 forks source link

inference error #25

Closed deep-practice closed 4 years ago

deep-practice commented 4 years ago

I use the pretrained weight to infer,but got the following error:

Traceback (most recent call last): File "inference_demo.py", line 15, in show_result(img, result, model.CLASSES, score_thr=0.25, out_file="demo_out.jpg") File "/home/work/deep_learning/instance_segmentation/SOLOv2/mmdet/apis/inference.py", line 154, in show_result for i, bbox in enumerate(bbox_result) File "/home/work/deep_learning/instance_segmentation/SOLOv2/mmdet/apis/inference.py", line 154, in for i, bbox in enumerate(bbox_result) AttributeError: 'tuple' object has no attribute 'shape'

the inference code I use is modified from inference_demo.ipynb pls,I can train normally.

Epiphqny commented 4 years ago

@deep-practice The code of demo is copied from the SOLO v1 code and I haven't used the code, you could run the test code to see if it is correct.

deep-practice commented 4 years ago

Test is ok.