pytorch数据处理:定义自己的数据集合实例-创新互联

数据处理

创新互联-云计算及IDC服务提供商，涵盖公有云、IDC机房租用、托管服务器、等保安全、私有云建设等企业级互联网基础服务，联系热线：18982081108

版本1

#数据处理
import os
import torch
from torch.utils import data
from PIL import Image
import numpy as np

#定义自己的数据集合
class DogCat(data.Dataset):

  def __init__(self,root):
    #所有图片的绝对路径
    imgs=os.listdir(root)

    self.imgs=[os.path.join(root,k) for k in imgs]

  def __getitem__(self, index):
    img_path=self.imgs[index]
    #dog-> 1 cat ->0
    label=1 if 'dog' in img_path.split('/')[-1] else 0
    pil_img=Image.open(img_path)
    array=np.asarray(pil_img)
    data=torch.from_numpy(array)
    return data,label

  def __len__(self):
    return len(self.imgs)

dataSet=DogCat('./data/dogcat')

print(dataSet[0])

当前名称：pytorch数据处理:定义自己的数据集合实例-创新互联
网址分享：http://www.wtcwzsj.com/article/dpipjc.html