数据集地址:
https://github.com/Sharpiless/gobang-object-detection-dataset
相关大作业进度的博客:
北理BIT人工智能大作业,写脚本收集了黑/白棋子检测数据集
数据集为pygame游戏界面截图:
这里使用PaddleX提供的YOLOv3目标检测算法。
同时由于目标比较好识别,所以使用轻量级的MobileNet作为主干网络。
写代码不易,求个star~
关注我的公众号:
感兴趣的同学关注我的公众号——可达鸭的深度学习教程: