本科毕设阶段第一次开始使用深度学习模型,并没有怎么系统的学习,于是找到B站上的一个up主上传的教程稍微跟着学了以下yolov5,最后做了一点小改进。
原github仓库为:https://github.com/bubbliiiing/faster-rcnn-pytorch。
项目的主要内容为使用yolov5对Android游戏界面截图进行控件识别,基础模型为Yolov5,对neck部分做了一点小变动,如下图。
项目的图片及截图资源在final_dataset文件夹中,做了一些数据增强处理。