yolov4-pytorch icon indicating copy to clipboard operation
yolov4-pytorch copied to clipboard

博主大佬,yolov4单小目标检测怎么找指标最好的权重,是根据F1指标还是?

Open ChenMaolong opened this issue 2 years ago • 6 comments

ChenMaolong avatar Sep 16 '22 02:09 ChenMaolong

map不可以嘛

bubbliiiing avatar Sep 18 '22 13:09 bubbliiiing

地图不可以嘛

是一个一个用验证集试吗(能通过totalloss和valloss吗),怎么快速找最好的权重?

ChenMaolong avatar Sep 23 '22 08:09 ChenMaolong

我有提供训练时计算map

bubbliiiing avatar Sep 26 '22 14:09 bubbliiiing

我有提供训练时计算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最低的对吗?)

ChenMaolong avatar Sep 28 '22 05:09 ChenMaolong

如果没有difficult参数,这两个应该差距不会很大

bubbliiiing avatar Oct 01 '22 08:10 bubbliiiing

那是以训练时的ap为准,还是get_map.py得到的ap为准呢,测试了多个,发现后者比前者的数据值要大(应该来说更准确)。。。顺便问一下,这两个跑的不一样是因为权重文件的不同吗,一个是train时用的你提供的,一个是训练后得到的。

yusiyiyusiyi avatar Apr 23 '23 13:04 yusiyiyusiyi