search

Shua-Kang / ros_pytorch_yolov5

23 stars 8 forks source link

Choosing between different model types #8

Open jaiswalharsh opened 2 years ago

jaiswalharsh commented 2 years ago

Hi! I have been trying to understand the wrapper's working and can't seem to find how to choose a model type. We want to use yolov5n(nano) and don't know what the default config is. Any help regarding how to choose a model would be great. Thank you!

Shua-Kang commented 2 years ago

Try:

  1. put your model file to ros_pytorch_yolov5/yolov5/weights
  2. modify line 3 of [ros_pytorch_yolov5/launch/detector.launch] to your model file name.