Как использовать экспортированный файл best.pt из файла yolov5 colab для локального запуска обученных весов?

Я обучил свою модель с помощью yoloV5 в Google Colab, следуя предоставленному руководству и пошаговому руководству для обучения любой пользовательской модели: Colab-файл для обучения вашей собственной модели. Теперь у меня есть экспортированный файл best.pt после запуска последней ячейки в предоставленной ссылке. Теперь я хочу использовать этот обученный вес для локального запуска обнаружения на любом скрипте Python. Это возможно? Если да, то как мне это сделать?


python pytorch object-detection google-colaboratory yolov5
person Zachary    schedule 09.06.2021    source источник

Ответы (2)


arrow_upward
0
arrow_downward

Вам следует выполнить этот шаг:

  • Создайте пустую папку на рабочем столе
  • Gitbash в этой папке и клонировать репозиторий yolov5 с помощью - git clone https://github.com/ultralytics/yolov5.git
  • После клонирования репо войдите в папку yolov5 и откройте командную строку по этому пути.
  • В командной строке установите все требования с помощью - pip install -r requirements.txt.
  • Скачайте best.pt из colab и вставьте в папку yolov5.
  • Также скопируйте изображение, которое вы хотите протестировать, в папку yolov5.
  • Затем вы можете выполнить вывод в командной строке с помощью этой команды: python detect.py --weights best.pt --source image.png
  • После завершения процесса вы можете проверить результат внутри пути run / detect / exp
person Prakash Dahal    schedule 09.06.2021
comment
Хорошо, когда я запускаю этап установки, я получаю сообщение об ошибке при создании колеса для pycocotools. Как мне это исправить? Или я игнорирую ошибку - person Zachary; 09.06.2021
comment
Вы используете windows или linux? - person Prakash Dahal; 09.06.2021
comment
Используйте эту команду отдельно pip3 install pycocotools - person Prakash Dahal; 09.06.2021

arrow_upward
0
arrow_downward

Использовать библиотеку файлов colab

from google.colab import files
files.download('/content/yolov5/runs/train/yolov5s_results/weights/best.pt')

Если путь отличается, вы можете изменить путь.

person Prabhat Kumar Sahu    schedule 09.06.2021