You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
I wanna share two things:
1.in train.py, for example, solov2_r50_fpn_8gpu_3x.py
model->pretrained and load_from are different, model->pretrained is the pretrained weights of backbone, for example, resnet.
url path: torchvision://resnet50.
load_from is pretrained weights of SOLO, for example, SOLOv2_R50_3x.pth which can be downloaded at SOLO readme.
2.Label error. mistake is :
if your class is ('A', 'B', 'C'), when you use test_ins_vis.py to get the result, you find you can get 'B' instead of 'A'(but the real label is 'A').
get 'C' instead of 'B'(but the real label is 'B'), but 'C' can't be predicted.
I use
and
as labels. then i get the good performance.(bbox_head->num_class=5 in train.py)
if you meet this problem, you can try this.
The text was updated successfully, but these errors were encountered:
if you get this label error, your train data may not have 'background'.
you may use CLASSES = ('A','B','C') or you add 'background' class in train data.
you can follow :
https://yuanwenjie.blog.csdn.net/article/details/123592452
and
https://blog.csdn.net/qq_37134388/article/details/115625744
I wanna share two things:
1.in train.py, for example, solov2_r50_fpn_8gpu_3x.py
model->pretrained and load_from are different, model->pretrained is the pretrained weights of backbone, for example, resnet.
url path: torchvision://resnet50.
load_from is pretrained weights of SOLO, for example, SOLOv2_R50_3x.pth which can be downloaded at SOLO readme.
2.Label error. mistake is :
if your class is ('A', 'B', 'C'), when you use test_ins_vis.py to get the result, you find you can get 'B' instead of 'A'(but the real label is 'A').
get 'C' instead of 'B'(but the real label is 'B'), but 'C' can't be predicted.
I use
and
as labels. then i get the good performance.(bbox_head->num_class=5 in train.py)
if you meet this problem, you can try this.
The text was updated successfully, but these errors were encountered: