getmap以及NMS问题

[cx330a]

B导我想请教一下我利用代码train跑出的map是0.86左右，但是getmap的是0.16，为什么会出现这种情况呢；关于先验框的问题我利用predict检测图片发现目标上有多个框但是只有一个物体，是我的NMS设置有问题还是出现了误检的问题呀

[bubbliiiing]

1、验证集和测试集差很大吗？
2、测试用的权重是？

[cx330a]

差距不大的，我进行了几次随机划分都是这个样子，测试权重是训练的得到的best_epoch_weight,我是在yolo.py里面设置的，然后运行getmap.py.

[cx330a]

还有一个问题是kmean聚类的类别数更改之后是dataloader那里也要修改吗，我更改了聚类数之后显示dataloader里面的存在维度超出的问题

[bubbliiiing]

best_epoch_weight是根据loss来的，不是根据mAP来的0 0，你可以试试不要用best_epoch_weight
kmean聚类的类别数不太懂是什么