【问题标题】:suggestion how to train pre trained models by using Yolov4?建议如何使用 Yolov4 训练预训练模型?
【发布时间】:2020-11-14 14:20:14
【问题描述】:

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

  • GeForce RTX 2060 SUPER
  • GeForce GTX 750 Ti

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

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

谢谢你的建议

【问题讨论】:

    标签: object-detection yolo


    【解决方案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

    【讨论】:

      猜你喜欢
      • 2021-01-09
      • 2017-08-19
      • 2021-01-26
      • 1970-01-01
      • 2023-01-03
      • 1970-01-01
      • 1970-01-01
      • 2019-08-16
      • 2019-04-10
      相关资源
      最近更新 更多