建议如何使用 Yolov4 训练预训练模型?
suggestion how to train pre trained models by using Yolov4?
我正在寻找为 YOLOv4 训练预训练模型的最佳方法
我的本地环境是 Debian 10 OS,
- GeForce RTX 2060 超级
- GeForce GTX 750 Ti
我计划根据不同大小的图像训练模型,我将使用训练好的模型作为在 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
我正在寻找为 YOLOv4 训练预训练模型的最佳方法 我的本地环境是 Debian 10 OS,
- GeForce RTX 2060 超级
- GeForce GTX 750 Ti
我计划根据不同大小的图像训练模型,我将使用训练好的模型作为在 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