2.5. 训练模型
在任务界面中点击提交训练任务,就可以开始训练模型,因为我们这次准备的数据不多,所以有一个提示,点击继续训练就好:
运行开始后可以看到本任务的状态为运行中:
可以点击日志,查看运行情况:
训练时间与数据量大小有关,1000张图片可能需要几个小时训练,不过本评测案例因为只有不到40张图,所以速度很快。运行成功有界面如图:
2.6. 校验模型效果
可通过模型评估报告或模型校验了解模型效果:
选择一张测试图:
感觉效果还可以,虽然BoundingBox有点偏差,不过主要是因为训练集太小以及采用了追求速度的YOLO算法。如果增加训练集以及采用FastRCNN会有不小的提升。