采集了接近4000张图像,包括
[‘qiao’, ‘zonglvshu’, ‘renxinghengdao’, ‘xiaofangshuang’, ‘honglvdeng’, ‘zixingche’, ‘xiaojiaoche’, ‘yancong’, ‘louti’, ‘jidongche’, ‘tuolaji’, ‘bus’, ‘school_bus’, ‘motuo’, ‘chuzuche’, ‘tingchejishiqi’, ‘chuan’, ‘shan’] 共18类
打标训练
训练了两种模型yolov5n及yolov5s
识别率还可以,里面包含安装环境的说明
数据集、代码及训练好的模型下载
里面包括训练代码,测试代码以及近4000张代标注数据集
测试代码 输入图片可以直接返回 小图片坐标。且测试代码是单独的工程,只包括加载模型及预测(可以省去很多其他的环境安装)
识别结果展示: