当前位置:网站首页>Pytorch load data
Pytorch load data
2022-07-06 03:19:00 【Up and down black】
One 、 Load data
- Load a single image in the dataset
from PIL import Image
img_path="D:\\pycharm\\PycharmProjects\\learn_torch\\hymenoptera_data\\train\\ants\\0013035.jpg"
img = Image.open(img_path)
img.show()
- Load dataset list
import os
dir_path = "hymenoptera_data/train/ants"
img_path_list = os.listdir(dir_path)
stay pycharm Medium Python console( controller ) Run in , The effect is more intuitive .
Two 、 Application pytorch Load data
torch.utils yes torch Common toolbox .
Want to according to idx Get the corresponding picture , Get the picture address first list( obtain dataset All under data).
from torch.utils.data import Dataset
- Definition MyData class
class MyData(Dataset):
''' Define global variables '''
def __init__(self, root_dir, label_dir):
self.root_dir = root_dir # root_dir Is the directory of the dataset
self.label_dir = label_dir # label_dir Is the catalog of labels
self.path = os.path.join(self.root_dir, self.label_dir)
self.img_path = os.listdir(self.path)
''' get data '''
def __getitem__(self, idx):
img_name = self.img_path[idx]
img_item_path = os.path.join(self.root_dir, self.label_dir, img_name)
img = Image.open(img_item_path)
label = self.label_dir
return img, label
''' The length of the data set '''
def __len__(self):
return len(self.img_path)
- Test defined MyData class
root_dir = "hymenoptera_data/train"
ants_label_dir = "ants"
bees_label_dir = "bees"
ants_dataset = MyData(root_dir, ants_label_dir)
bees_dataset = MyData(root_dir, bees_label_dir)
Method 1 :
Can be found in Python Medium test :
ants_dataset[0]
Method 2 :
img, label = ants_dataset[0]
img.show()
Add :
train_dataset = ants_dataset + bees_dataset # The whole training set ( A collection of ant data sets and bee data sets )
When the data set is insufficient , Data sets can be supplemented in this way
边栏推荐
- Explore pointers and pointer types in depth
- 手写数据库客户端
- 适合程序员学习的国外网站推荐
- 【RISC-V】外部中断
- [risc-v] external interrupt
- #PAT#day10
- Zhang Lijun: penetrating uncertainty depends on four "invariants"
- 【指针训练——八道题】
- [network security interview question] - how to penetrate the test file directory through
- Derivation of anti Park transform and anti Clarke transform formulas for motor control
猜你喜欢
随机推荐
js 正则过滤和增加富文本中图片前缀
BUUCTF刷题笔记——[极客大挑战 2019]EasySQL 1
[network security interview question] - how to penetrate the test file directory through
I sorted out a classic interview question for my job hopping friends
ERA5再分析资料下载攻略
3857 Mercator coordinate system converted to 4326 (WGS84) longitude and latitude coordinates
Idea push rejected solution
[padding] an error is reported in the prediction after loading the model weight attributeerror: 'model' object has no attribute '_ place‘
Selenium share
Distributed service framework dobbo
Game theory matlab
Polymorphic day02
Mysqldump data backup
mysqldump数据备份
电机控制反Park变换和反Clarke变换公式推导
OCR文字识别方法综述
Classic interview question [gem pirate]
【RISC-V】外部中断
1.16 - 校验码
【指针训练——八道题】