search

bubbliiiing / yolov7-pytorch

这是一个yolov7的库，可以用于训练自己的数据集。
GNU General Public License v3.0
861 stars 150 forks source link

同样的条件下再次训练get_map之后的结果差异较大 #71

Open SCcdet opened 1 year ago

SCcdet commented 1 year ago

B导你好，请问下我没采用预训练权重，先用模型训练了300轮，得到一个权重，之后用这个代码上不做任何修改重新训练300轮，两次训练后get_map的结果有差异，差了将近5%的map，请问下这是什么原因呢？

bubbliiiing commented 1 year ago

啊？你的意思是第一次的基础上第二次预训练吗

SCcdet commented 1 year ago

第二次也是从0开始训练，没有用任何预训练权重，就是两次一模一样条件下的复现结果不一样，差异较大。

bubbliiiing commented 1 year ago

ooo，数据集比较小是这样的，需要加seed 但是有人反馈加了seed还是不太一样0 0，可以先试试

SCcdet commented 1 year ago

噢噢这样啊，明白了谢谢b导

bubbliiiing commented 1 year ago

加成功了吗

1992232054 commented 1 year ago

你才5%吗，我两次之间差了一半以上，第一次92%，第二次43%，这是什么导致的啊

bubbliiiing commented 1 year ago

0 0………………你这个离谱了，数据集太小了吧？

SCcdet commented 1 year ago

没有，还是会有一点差异。后来扩充了下数据集差距好像也不是很大。

SCcdet commented 1 year ago

你才5%吗，我两次之间差了一半以上，第一次92%，第二次43%，这是什么导致的啊

你这个差的确实有点太大了，是不是两次采用的评价指标不一样了啊？