c99774的博客

原创 将MNIST的手写数字识别转化为使用yolov5训练的模型

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