DOTA1.0train，test下来ap不是很高

yingkunwu / R-YOLOv4

This is a PyTorch-based R-YOLOv4 implementation which combines YOLOv4 model and loss function from R3Det for arbitrary oriented object detection.

114 stars 20 forks source link

Closed wangjiawen-123 closed 1 year ago

wangjiawen-123 commented 2 years ago

楼主好，我用的这个代码+DOTA1.0，裁剪成1024尺寸，epochs=50，AP最后的结果很低，并与您的outputs对比十分不理想，nms=0.2，最后杂框很多，我不知道是哪里出问题了 mAP

yingkunwu commented 2 years ago

可以看一下你的train.py 的 argument怎麼設的嗎

wangjiawen-123 commented 2 years ago

整体您的代码我没有怎么改动，train里面augmentation就是lib里的没动，mosaic也是，不知道什么原因，我用的这个代码大概是1月份您的版本，不知道后期有没有改进

yingkunwu commented 2 years ago

我主要是想看sample_size的部分