- 博客(1)
- 收藏
- 关注
原创 将MNIST的手写数字识别转化为使用yolov5训练的模型
把MNIST数据集转化为VOC数据集基本上大功就快告成了,后面就是划分数据集、修改数据集路径使用train.py训练模型、使用测试数据通过运行detect.py来测试训练好的模型。但是由于是直接从MNIST数据集转化为VOC数据集的,图片都是没有画框的,而且在这些图片的对应的.txt文件中xy的中心点都是0,框的长度也都是1,这很明显是会出现下面的结果。而yolov5又是当前较为流行的算法,刚好有个作业是要改进上面的模型,于是使用yolov5算法来训练手写数字识别的模型。这也说明了模型的鲁棒性不够强。
2024-01-23 20:06:27 850 2
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人