没有合适的资源?快使用搜索试试~ 我知道了~
使用PyTorch将文件夹下的图片分为训练集和验证集实例
22 下载量 90 浏览量
2020-12-23
05:52:14
上传
评论 1
收藏 32KB PDF 举报
温馨提示
试读
2页
PyTorch提供了ImageFolder的类来加载文件结构如下的图片数据集: root/dog/xxx.png root/dog/xxy.png root/dog/xxz.png root/cat/123.png root/cat/nsdf3.png root/cat/asd932_.png 使用这个类的问题在于无法将训练集(training dataset)和验证集(validation dataset)分开。我写了两个类来完成这个工作。 import os import torch from torch.utils.data import Dataset, DataLoader f
资源推荐
资源评论
资源评论
weixin_38663193
- 粉丝: 8
- 资源: 951
上传资源 快速赚钱
- 我的内容管理 展开
- 我的资源 快来上传第一个资源
- 我的收益 登录查看自己的收益
- 我的积分 登录查看自己的积分
- 我的C币 登录后查看C币余额
- 我的收藏
- 我的下载
- 下载帮助
安全验证
文档复制为VIP权益,开通VIP直接复制
信息提交成功