WXinlong / SOLO

SOLO and SOLOv2 for instance segmentation, ECCV 2020 & NeurIPS 2020.
Other
1.69k stars 307 forks source link

bugs in your soloV2 solution #194

Open SylvainArd opened 3 years ago

SylvainArd commented 3 years ago

Hello, I am trying to test SOLOv2 in https://github.com/WXinlong/SOLO I use my own dataset with only one class : "foliole_0" I have done training but : I have the following bug when I launch

python tools/test.py configs/solov2/solov2_r50_fpn_8gpu_3x.py demo/epoch_36.pth --json_out output --eval segm
[>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>] 30/30, 5.1 task/s, elapsed: 6s, ETA: 0sTraceback (most recent call last):
File “tools/test.py”, line 282, in
main()
File “tools/test.py”, line 273, in main
results2json(dataset, outputs, args.json_out)
File “c:\solo-master\mmdet\core\evaluation\coco_utils.py”, line 224, in results2json
json_results = det2json(dataset, results)
File “c:\solo-master\mmdet\core\evaluation\coco_utils.py”, line 153, in det2json
for i in range(bboxes.shape[0]):
AttributeError: ‘NoneType’ object has no attribute ‘shape’

and when I run the demo I have always 0 instances Please help me ! Best regards

SylvainArd commented 3 years ago

I have instances but always only one. That I want is to segment the leaflets of a leaf but if the leaf is simple it works but if the leaf is compound all the leaflets are in the same instance, an idea ?

SylvainArd commented 3 years ago

I have fixed the only instance bug by take a resnet101 abnd 4 images per gpu vs a resnet 50 and 2 images per gpu