Train models and datasets

[Phance]

@mgonzs13 Can I create my own data set and then train the model, and finally run the trained model with this?How can I specifically train my own model?I hope you can give me some advice and some starting points. Thank you very much!

[mgonzs13]

Hi @Phance, to train a custom model you have to use ultralytics. Take a look at its docs https://docs.ultralytics.com/es/modes/train/#usage-examples.

[Phance]

@mgonzs13 Hi ! After training the model with my custom dataset, how can I use this yolo_ros to run and perform detection? What format should I use to save the final model? I am really looking forward to your reply, thank you very much.

[mgonzs13]

@Phance you can use pt format. Then, pass the file path to the launch using the model argument.

[liamcarlos]

Hey @Phance did you figure out how to use you custom model set? I've trained my own model and successfully used it with this package and found it as an incredibly useful tool for my project, so feel free to ask me anything if you need help!