【深度学习】3.2 使用pytorch搭建AlexNet并训练花分类数据集【能学到如何搭建自己的数据集】

最新推荐文章于 2024-07-09 01:48:08 发布

算法研究员

最新推荐文章于 2024-07-09 01:48:08 发布

阅读量389

点赞数

分类专栏： # 【经典网络pytorch实现】霹雳吧啦Wz

本文链接：https://blog.csdn.net/weixin_43154149/article/details/116102591

版权

【经典网络pytorch实现】霹雳吧啦Wz 专栏收录该内容

16 篇文章 6 订阅

订阅专栏

文章目录

前言
model.py
- dropout
- flatten
train.py
- 加载数据集实用！（顶）
- 查看数据集
predict.py

前言

视频链接
 配合食用 - pytorch图像分类篇：3.搭建AlexNet并训练花分类数据集

model.py

在这里插入图片描述

dropout

在这里插入图片描述

flatten

在这里插入图片描述

train.py

加载数据集实用！（顶）

在这里插入图片描述

查看数据集

在这里插入图片描述

test_data_iter = iter(validate_loader)
test_image, test_label = test_data_iter.next()

def imshow(img):
    img = img / 2 + 0.5     # unnormalize
    npimg = img.numpy()
    plt.imshow(np.transpose(npimg, (1, 2, 0)))
    plt.show()

# print labels
print(' '.join('%5s' % cla_dict[test_label[j].item()] for j in range(4)))
# show images
imshow(utils.make_grid(test_image))