CustomDataset 輸入參數問題

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 Ian-Shih closed 1 month ago

Ian-Shih commented 1 month ago

想請教一下，為何我看其他的Dataset的輸入輸出都差不多，唯獨CustomDataset完全不一樣，輸入參數跟輸出參數完全對不上另外為何訓練時best.pth第一個epoch儲存後就沒有更新了，明明last.pth的效果明顯比best.pth還要好，判斷模型好壞的部分是不是有些bug存在？