墙体破损检测

Question

我打算使用 YOLOv5 和我的自定义图像数据集 (125) 创建一个实时墙壁损坏检测器 [划痕、裂缝]。

1. 你认为我可以做迁移学习吗，还是不可能，因为 coco 数据集 类 不相似？
2. 您认为我需要增加数据集大小吗？

现在我只是想做一个概念验证。想提前计划我的步骤。

Answer 1

如果你的数据集很小（就像你的情况），与从头开始训练相比，迁移学习几乎总能得到更好的结果。至于你的第二个问题，是的。您获得的数据越多，您的模型就能够更好地学习和执行。考虑到这是一个与 Yolo-V5 最初训练的任务相对不同的任务，请尝试获取尽可能多的图像