XinzeLee / RotateObjectDetection

This repository is based on Ultralytics/yolov5, with adjustments to enable rotate prediction boxes.
113 stars 21 forks source link

It is possible to use other models: medium, large, x..? #29

Open mamanso opened 9 months ago

mamanso commented 9 months ago

Hi Xinze I have created an python envirom to probe your code and it works fine with the examples. Crongratulations.

I am trying to train with my own data, 512x512 image tiles labeling bridges over roads, train, and rivers.

My question is related to how can i create a more complex model (medium, large, extra-large) to train it with my labeled data? I can create de config file with de template for that models, and i can modify it but the starting weigths file, how can i create it? If a do no specify it download yolov5s.pt model and do no have expected information. Same when specify other models like medium or large.

Thank you