Shua-Kang / ros_pytorch_yolov5

23 stars 8 forks source link

请问如何使用自己训练的权重 #14

Closed 1solati0n closed 1 year ago

1solati0n commented 1 year ago

你好,我想使用自己训练的权重来进行识别,但是我不知道都需要修改哪些地方,请问该怎么去修改呢

Shua-Kang commented 1 year ago

I think you can 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.
1solati0n commented 1 year ago

I tried but yolo.py have a lot of errors ,I tried to solve one by one and now it works successfully

Shua-Kang commented 1 year ago

OK. Maybe it is because different yolo versions have different yolo.py.