我如何使用从 yolov5 colab 文件导出的“best.pt”文件到 运行 在本地训练的权重？
How do I use the exported 'best.pt" file from yolov5 colab file to run the trained weights locally?
我已经在 google colab 上使用 yoloV5 训练了我的模型，遵循为训练任何自定义模型提供的教程和演练：Colab file for training your own custom model。在 运行 提供的 link 中的最后一个单元格之后，我现在有一个导出的 best.pt 文件。现在，我想利用这个经过训练的权重 运行 在任何 python 脚本上进行本地检测。这可能吗？如果是这样，我该怎么做？
使用 colab 文件库
from google.colab import files
files.download('/content/yolov5/runs/train/yolov5s_results/weights/best.pt')
如果路径不同，您可以更改路径。
您应该按照以下步骤操作：
- 在桌面上创建一个名为 ObjectDetection
的空文件夹
- 打开命令提示符并使用以下命令将目录更改为该新文件夹
cd ObjectDetection.
- 使用命令克隆 yolov5 存储库 -
git clone https://github.com/ultralytics/yolov5.git。它将在 ObjectDetection 文件夹中创建一个名为 yolov5 的新文件夹。
yolov5 文件夹包含名为 detect.py 的重要 python 文件，它负责检测对象。
- 克隆repo后，使用
cd yolov5 进入yolov5文件夹
- 使用 -
pip install -r requirements.txt 安装所有必要的要求
- 从 colab 下载 best.pt 并手动将其粘贴到 yolov5 文件夹中。
- 同时将要测试的图像复制到 yolov5 文件夹中。
- 在运行推理之前，确保image.png、best.pt和detect.py应该在[=里面18=]yolov5 文件夹.
- 然后您可以 运行 使用以下命令在 yolov5 文件夹中进行推理：
python detect.py --weights best.pt --source image.png
- 处理完成后，您可以在路径
ObjectDetection/yolov5/runs/detect/exp 中查看结果
我已经在 google colab 上使用 yoloV5 训练了我的模型，遵循为训练任何自定义模型提供的教程和演练：Colab file for training your own custom model。在 运行 提供的 link 中的最后一个单元格之后，我现在有一个导出的 best.pt 文件。现在，我想利用这个经过训练的权重 运行 在任何 python 脚本上进行本地检测。这可能吗？如果是这样，我该怎么做？
使用 colab 文件库
from google.colab import files
files.download('/content/yolov5/runs/train/yolov5s_results/weights/best.pt')
如果路径不同，您可以更改路径。
您应该按照以下步骤操作：
- 在桌面上创建一个名为 ObjectDetection 的空文件夹
- 打开命令提示符并使用以下命令将目录更改为该新文件夹
cd ObjectDetection. - 使用命令克隆 yolov5 存储库 -
git clone https://github.com/ultralytics/yolov5.git。它将在 ObjectDetection 文件夹中创建一个名为 yolov5 的新文件夹。
yolov5 文件夹包含名为 detect.py 的重要 python 文件，它负责检测对象。 - 克隆repo后，使用
cd yolov5进入yolov5文件夹
- 使用 -
pip install -r requirements.txt 安装所有必要的要求
- 从 colab 下载 best.pt 并手动将其粘贴到 yolov5 文件夹中。
- 同时将要测试的图像复制到 yolov5 文件夹中。
- 在运行推理之前，确保image.png、best.pt和detect.py应该在[=里面18=]yolov5 文件夹.
- 然后您可以 运行 使用以下命令在 yolov5 文件夹中进行推理：
python detect.py --weights best.pt --source image.png - 处理完成后，您可以在路径
ObjectDetection/yolov5/runs/detect/exp中查看结果