yolov4-pytorch
yolov4-pytorch copied to clipboard
博主大佬,yolov4单小目标检测怎么找指标最好的权重,是根据F1指标还是?
map不可以嘛
地图不可以嘛
是一个一个用验证集试吗(能通过totalloss和valloss吗),怎么快速找最好的权重?
我有提供训练时计算map
我有提供训练时计算map
我用的3.1版本的训练时候五个epoch计算一次map,比如第300epoch Average Precision (AP) @[ IoU=0.50 | area= all | maxDets=100 ] = 0.621,然后我用getmap.py用的验证集(参数什么的都没改3.1版本是啥样就是啥样)跑了后 76.80% = nodule AP || score_threhold=0.5 : F1=0.78 ; Recall=75.63% ; Precision=80.36%他们两个map相差0.147这是因为什么?(还有best_epoch_weights是跑的所有权重1-300epoch中valloss最低的对吗?)
如果没有difficult参数,这两个应该差距不会很大
那是以训练时的ap为准,还是get_map.py得到的ap为准呢,测试了多个,发现后者比前者的数据值要大(应该来说更准确)。。。顺便问一下,这两个跑的不一样是因为权重文件的不同吗,一个是train时用的你提供的,一个是训练后得到的。