object-detection - 建议如何使用 Yolov4 训练预训练模型？

Question

我正在寻找为 YOLOv4 训练预训练模型的最佳方法我有我的本地环境 Debian 10 OS，

• GeForce RTX 2060 超级
• GeForce GTX 750 钛

我计划根据不同大小的图像训练模型，并将训练后的模型用作基于 java 或 python 开发的微服务的一部分。

我应该使用任何第三方服务，如 Google colab 吗？第二个问题用什么框架比较好？（pytorch、张量流等）

谢谢你的建议

标签： object-detectionyolo

Answer 1

你可以使用这个 repo，https://github.com/AlexeyAB/darknet。它具有自定义训练、迁移学习以及 colab 训练、推理脚本的所有说明。

您可以使用提供的卷积层权重在小型数据集上更快地改进结果。不同维度的图像可用于训练。

定制培训说明，

https://github.com/AlexeyAB/darknet#how-to-train-to-detect-your-custom-objects

在 colab 中进行训练和推理，

https://colab.research.google.com/drive/1_GdoqCJWXsChrOiY8sZMr_zbr_fH-0Fg