从头开始或微调或转移学习的模型

Question

我想使用 Yolov4 和 Yolov3，但是 Github 上由 AlexejAB Repo 提供的在 ImageNet 上训练的暗网预训练模型有 类 狗，猫，...

我的小数据集是关于区分电缆的，它由 4 类（总共 240 张图片）组成。

由于我不认为存在具有类似数据集的预训练模型，我应该从头开始训练还是在我的案例中使用迁移学习/微调大型预训练模型是否有优势？

Answer 1

我认为您应该首先简单地微调数据集上的现有模型，因为从头开始训练的成本会有点高。