Pytorch demo(三)之蚂蚁和蜜蜂

模型

resnet18

在这里插入图片描述

18 = 17个卷积层 + 1个全连接层

流程

在这里插入图片描述

加载数据集并训练

此demo是区分蚂蚁和蜜蜂。
在这里插入图片描述

分类测试

【1】蚂蚁

在这里插入图片描述
在这里插入图片描述

在这里插入图片描述

【2】蜜蜂

在这里插入图片描述
在这里插入图片描述
在这里插入图片描述

猜你喜欢

转载自blog.csdn.net/FemaleHacker/article/details/108682075
今日推荐