Google Cloud vertex AI 上的 Yolo v5 模型部署

问题描述 投票:0回答:1

我试图在 Vertex AI 上训练和部署我的 yolov5 模型,并正在关注这篇文章 -> https://blog.ml6.eu/how-to-deploy-a-yolov4-darknet-model-to-vertex-ai-a8df1b783341.

本教程旨在部署yolov4暗网模型,但问题是我的权重文件是.pt扩展名,是否可以将我的.pt模型文件转换为.weights扩展名。

我尝试在互联网上找到多种解决方案,但似乎都不起作用。

我也看过这个问题:https://github.com/ultralytics/ultralytics/issues/941

但即使建议某种方法将 .pt 文件转换为 .weight 文件扩展名,我也无法理解。

我的主要疑问是,是否可以进行这样的转换?

python computer-vision yolov5 yolov4
1个回答
0
投票

这个 repo 有一个脚本,可以接受 .pt 权重并将其转换为所需的 Onyx 格式。

这是脚本 -> https://github.com/marcoslucianops/DeepStream-Yolo/blob/master/utils/export_yoloV5.py

© www.soinside.com 2019 - 2024. All rights reserved.