问题

如何使用 yolov5 colab 文件中导出的“best.pt”文件在本地运行训练的权重？

我已经在 google colab 上使用 yoloV5 训练了我的模型，遵循提供的教程和为训练任何自定义模型提供的演练：用于训练您自己的自定义模型的 Colab 文件。在运行提供的链接中的最后一个单元格后，我现在有一个导出的 best.pt 文件。现在，我想利用这个训练有素的权重在任何 python 脚本上本地运行检测。这可能吗？如果是这样，我该怎么做？

python pytorch
推荐答案

您应该按照以下步骤操作：

  • 在桌面中创建一个名为 ObjectDetection 的空文件夹
  • 打开命令提示符并使用 cd ObjectDetection 将目录更改为该新文件夹。
  • 使用克隆 yolov5 存储库命令 - git clone https://github.com/ultralytics/yolov5.git。它将在 ObjectDetection 文件夹中创建一个名为 yolov5 的新文件夹。yolov5 文件夹包含名为 detect.py 的重要 python 文件，该文件负责检测对象。
  • 克隆 repo 后，使用 进入 yolov5 文件夹 cd yolov5
  • 使用 - pip install -r requirements.txt
  • 从 colab 下载 best.pt 并手动粘贴它在 yolov5 文件夹中。
  • 还将要测试的图像复制到 yolov5 文件夹中。
  • 在运行推理之前，确保 image.png、best.pt 和 detect.py 应该在 yolov5 文件夹内。
  • 然后您可以使用以下命令在 yolov5 文件夹内运行推理：python detect.py --weights best.pt --source image.png
  • 处理完成后，可以查看路径 ObjectDetection/yolov5/runs/detect/exp
    • 里面的结果