当前位置: 首页 > news >正文

DataLoader

import torchvision
from torch.utils.data import DataLoader
from torch.utils.tensorboard import SummaryWriter# 准备的测试数据集  数据放在了CIFAR10文件夹下test_data = torchvision.datasets.CIFAR10("./CIFAR10",train=False, transform=torchvision.transforms.ToTensor())
test_loader = DataLoader(dataset=test_data, batch_size=4, 
shuffle=True, num_workers=0, drop_last=False)# 测试数据集中第一张图片及target
img, target = test_data[0]
print(img.shape)
print(target)# 在定义test_loader时,设置了batch_size=4,表示一次性从数据集中取出4个数据
for data in test_loader:imgs, targets = dataprint(imgs.shape)print(targets)# 在定义test_loader时,设置了batch_size=4,表示一次性从数据集中取出4个数据
writer = SummaryWriter("logs")
for epoch in range(2):step = 0for data in test_loader:imgs, targets = datawriter.add_images("Epoch: {}".format(epoch), imgs, step)step = step + 1
writer.close()

 把CIFAR10做成一个数据集,然后得到迭代器

每个迭代器包括图像和标签

下面是tensorboard的用法

 

http://www.lryc.cn/news/312127.html

相关文章:

  • 持续集成(CICD)- Jenkins+Git+gogs综合实战(笔记二)
  • VUE:key属性的作用
  • linux的通信方案(SYSTEM V)
  • VUE 入门及应用 ( 路由 router )
  • SpringBoot集成RocketMQ
  • 【Web】关于FastJson反序列化开始前的那些前置知识
  • 工业镜头的重要参数之视场、放大倍率、芯片尺寸--51camera
  • 基于java springboot+redis网上水果超市商城设计和实现以及文档
  • 3. 在Go语言项目中使用Zap日志库
  • 想要节省成本,哪个品牌的https证书值得考虑?
  • R语言及其开发环境简介
  • 部署DNS解析服务
  • 2024新算法:鹅算法优化VMD参数,五种适应度函数任意切换,最小包络熵、样本熵、信息熵、排列熵、排列熵/互信息熵...
  • 自定义注解校验
  • 由数据范围反推算法复杂度以及算法内容
  • js监听F11触发全屏事件
  • Seata 2.x 系列【1】专栏导读
  • fly-barrage 前端弹幕库(3):滚动弹幕的设计与实现
  • Mysql面试总结
  • 【深圳五兴科技】Java后端面经
  • 画图(ccf201409-2)解题思路
  • 蓝桥杯刷题(一)
  • 设计模式:策略模式 ⑥
  • 数据结构从入门到精通——顺序表
  • 001-CSS-水平垂直居中布局
  • 【[STM32]标准库-自定义BootLoader】
  • Spring Boot项目中不使用@RequestMapping相关注解,如何动态发布自定义URL路径
  • Vue中有哪些优化性能的方法?
  • Python pandas遍历行数据的2种方法
  • Spring之@Transactional源码解析