建议如何使用 Yolov4 训练预训练模型?

suggestion how to train pre trained models by using Yolov4?

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

我计划根据不同大小的图像训练模型,我将使用训练好的模型作为在 java 或 python 上开发的微服务的一部分。

我应该使用 Google colab 等任何第三方服务吗? 第二个问题什么框架更好用? (pytorch、tensorflow 等)

感谢您的建议

您可以使用此存储库,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