search

bubbliiiing / yolov7-pytorch

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

kmeans聚类anchor失败 #66

Open neuhsm opened 1 year ago

neuhsm commented 1 year ago

我在DroneVehicle数据集上聚类新的anchor，但是不知道为什么一直出问题nan image

bubbliiiing commented 1 year ago

鹅……目标太小？

neuhsm commented 1 year ago

应该

鹅……目标太小？

应该是的，我找了一个别的代码聚类出来了 image 这样的

neuhsm commented 1 year ago

然后我试着按照YOLOv7作者代码里面加focalloss的方式加了，开始发现效果还不如之前的，我参考你YOLO4里面加了10倍的放大，效果好像有一点点，但是测试map的时候太慢了，其中一个类别的中间文件1.2G，快一个小时都没测完，验证集数据量小很多能测出来，验证集只有1469张图片，这个txt几十MB

image

neuhsm commented 1 year ago

我好像知道focal loss为什么不行了，我发现SimOTA里面有类似的操作了。。。。 image 我再去搞搞别的地方

bubbliiiing commented 1 year ago

呀哈0 0