tensorflow - 如何将 Yolov5 模型转换为 tensorflow.js

Question

是否可以将 YOLOv5 PyTorch 模型转换为 Tensorflow.js 模型？

我正在开发一个对象检测网络应用程序。所以我已经使用 Yolov5 训练了数据，但现在我正在寻找将模型转换为 Tf.js 的正确方法。

标签： tensorflowmachine-learningtensorflow.jsyolov5

Answer 1

我认为这就是您要寻找的东西https://github.com/zldrobit/tfjs-yolov5-example

在YoloV5 存储库中，运行export.py命令。

python export.py --weights yolov5s.pt --include tfjs

然后 cd 进入上面链接的 repo 并将 weights 文件夹复制到公共：

cp ./yolov5s_web_model public/web_model

不要忘记，您必须更改名称数组src/index.js以匹配您的自定义模型。

但不幸的是，它在大约 1-2 秒时似乎非常缓慢。我认为我无法让 WebGL 正常工作。