任务描述:
(这是基础课的作业就TM离谱)
目前已经开源:https://github.com/Sharpiless/gobang-object-detection-dataset
构建棋盘:
这里使用PyQt5开发(GIthub有源码,稍作改动),效果如图:
收集数据:
由于棋子比较密集而且较小,自己标注的话进度很慢,所以我写了个小脚本,在下棋的过程中自动记录棋子的位置坐标和类别,并写入xml文件生成目标检测训练集。
训练模型:
这里使用PaddleX提供的YOLOv3目标检测算法。
同时由于目标比较好识别,所以使用轻量级的MobileNet作为主干网络。
检测效果:
总结:
第一天完成了目标检测任务,后面的话打算写个简单的UI界面可视化一下检测结果。
有需求的大佬欢迎加入我的接单群,需求详情请群里戳群主
关注我的公众号:
感兴趣的同学关注我的公众号——可达鸭的深度学习教程: