pytorch绘制并显示loss曲线和acc曲线,LeNet5识别图像准确率
我用的是Anaconda3 ,用spyder编写pytorch的代码,在Anaconda3中新建了一个pytorch的虚拟环境。如下图所示:3.安装完成之后,先给大家看一下我的文件夹,如下图:假设用LeNet5框架识别图像的准确率,LeNet.py代码如下:示网络结构如下图:训练代码如下:样本图像是pytorch官网中介绍迁移学习时用到的,蚂蚁与蜜蜂的二分类图像,图像大小不一。LeNet5 的输入图像是32*32,所以进行分类时会损失一定的图像像素,导致识别率较低。在LeNet_train_test.py中, writer.add_scalar和writer.add_scalar,这两行代码就是生成train数据集的loss和acc曲线,同理测试数据集亦是如此。
用户评论