我计划使用YOLOv5和我的自定义图像数据集(125张)创建一个实时墙体损伤检测器(划痕、裂缝)。
- 你认为我可以进行迁移学习吗,还是因为coco数据集的类别不相似而无法实现?
- 你认为我需要增加数据集的大小吗?
目前我只是在尝试做一个概念验证。希望提前规划好步骤。
回答:
如果你的数据集很小(就像你的情况),与从头开始训练相比,迁移学习几乎总是能得到更好的结果。至于你的第二个问题,是的。数据越多,你的模型学习和表现得就越好。考虑到这是一个与Yolo-V5最初训练的任务相对不同的任务,尽可能多地获取图像吧。