问题描述
我已经在 google colab 上使用 yolov5 训练了我的模型，遵循为训练任何自定义模型提供的教程和演练：Colab file for training your own custom model。在运行提供的链接中的最后一个单元格后，我现在有一个导出的 best.pt 文件。现在，我想利用这个经过训练的权重在任何 python 脚本上本地运行检测。这可能吗？如果是这样，我该怎么做？
解决方法
使用 colab 文件库
from google.colab import files
files.download('/content/yolov5/runs/train/yolov5s_results/weights/best.pt')
如果路径不同，您可以更改路径。
,您应该按照以下步骤操作：
- 在桌面创建一个空文件夹
- 该文件夹中的 Gitbash 并使用 - git clone https://github.com/ultralytics/yolov5.git 克隆 yolov5 存储库
- 克隆 repo 后，进入 yolov5 文件夹并在该路径中打开命令提示符。
- 在命令提示符下，使用 - pip install -r requirements.txt 安装所有要求
- 从 colab 下载
best.pt并将其粘贴到 yolov5 文件夹中。 - 同时将要测试的图像复制到 yolov5 文件夹中。
- 然后，您可以使用以下命令在命令提示符下运行推理： python detect.py --weights best.pt --source image.png
- 该过程完成后，您可以在路径runs/detect/exp中查看结果