search

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

数据集txt标签问题 #20

Closed jiayaohua closed 1 year ago

jiayaohua commented 2 years ago

iii 这个模型好像必须使用这种13列的标签，我试了用rolabelimg2制作的5列标签，他就会报错，错误如下： ppp

jiayaohua commented 2 years ago

这个13列的标签，各列都代表啥呀。实在弄不明白，只能再次麻烦您了

yingkunwu commented 2 years ago

你必須要更改load.py內的程式碼才能夠讀取客製化的資料集。根據這個問題我已經做了update，現在只需要在train.py後面多加上一個--custom_dataset的flag，應該就可以成功的執行你客製化的程式碼。希望有幫助到你。

jiayaohua commented 2 years ago

1 这种数据集标签格式，大佬你知道是如何制作的吗？

yingkunwu commented 2 years ago

我不知道欸，這是官方製作的數據集。 如果你需要用自己客製化的資料集做訓練的話，你可以去我寫的Readme.md中看要如何訓練。