pytorch
泰勒几天不想展开
多多互相学习
展开
-
pytorch构建卷积自编码器实现图片的压缩的功能,图片检索(以图搜图)功能
使用Pytorch构建卷积自编码器,尝试对MNIST数据集进行压缩。利用上述自编码器完成手写数字的“以图搜图”的功能。要求有构建与训练自编码器的代码和训练过程,以及以图搜图的结果(在测试集中随机拿出10张类别不同的图片,查找其中每个图片的最接近的5-10张图片)。代码实现:import numpy as npimport torchimport torchvisionfrom t...原创 2020-04-10 10:18:36 · 4294 阅读 · 1 评论 -
Mnist数据集下载慢的问题。
一般下载mnist数据集代码会这样下载。train_data = mnist.MNIST('./data', train=True,download=True, transform=transform)test_data = mnist.MNIST('./data', train=False,download=True ,transform=transform)因为这个网站的数据集在美国,...原创 2020-04-10 09:54:29 · 2357 阅读 · 1 评论