网络视频直播系统 VBA linux创建文件 authentication web post button split vue组件开发 郑州小程序公司 list获取最后一个元素 oracle取第一条数据 android调试工具 cmd清空命令 mysql插入 后台管理网站模板 python返回值 java获取月份 java中数据类型 java替换字符 java连接sql数据库 java文件读取 linux文档 霜之祝福 超星网课助手 选择模拟位置信息应用 groupy 彩虹岛小草黑暗之矛 疯狂java讲义 深渊碎片 qq飞车刷车 winsxs可以删除吗 android开发案例 屏幕录像机 拍照姿势的摆法女 小米账号怎么退出 su镜像 谷歌浏览器升级 pcmark10 cad周长快捷键
当前位置: 首页 > 学习教程  > python

torchvision.utils.make_grid() 拼接图片 mnist数据集(新手)

2021/2/6 22:30:38 文章标签: 测试文章如有侵权请发送至邮箱809451989@qq.com投诉后文章立即删除

参考:博客园 import torch from torchvision import datasets, transforms from torch.utils.data import DataLoader import torchvision import matplotlib.pyplot as pltdef image_show(images):images images.numpy()images images.transpose((1, 2, 0))pri…

参考:博客园 

import torch
from torchvision import datasets, transforms
from torch.utils.data import DataLoader
import torchvision
import matplotlib.pyplot as plt

def image_show(images):
    images = images.numpy()
    images = images.transpose((1, 2, 0))
    print(images.shape)
    plt.imshow(images)
    plt.show()

def main():
    train_dataset = datasets.MNIST(root='./datasets', train=False, download=False,
                                   transform=transforms.ToTensor())
    train_loader = DataLoader(train_dataset, batch_size=32, shuffle=False)

    device = torch.device('cuda:0')
    # for batch_idx, (inputs, targets) in enumerate(train_loader):
    #     inputs = inputs.to(device)
    #     print(inputs.shape)
    inputs, targets = next(iter(train_loader))
    print(inputs.shape)
    print(targets.shape)

    images = torchvision.utils.make_grid(inputs)
    print(f'images.shape:{images.shape}')
    image_show(images)




if __name__=='__main__':
    main()

 


本文链接: http://www.dtmao.cc/news_show_2000138.shtml

附件下载

相关教程

    暂无相关的数据...

共有条评论 网友评论

验证码: 看不清楚?