如何利用Yolov4训练预训练模型的建议？

Question

我正在寻找最佳的方式来训练YOLOv4的预训练模型我有我的本地环境Debian10OS，

• GeForce RTX 2060超级
• GeForce GTX 750 Ti

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

我应该使用像Google这样的第三方服务吗？第二个问题是什么框架更好的使用？(火把、丹瑟尔流等)

谢谢你的建议

Answer 1

你可以用这个回购，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