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数据处理:定义自己的数据集合实例-创新互联
本文链接：https://www.cdcxhl.com/article22/dpipjc.html

成都网站建设公司_创新互联，为您提供软件开发、网站设计、品牌网站建设、网站维护、定制开发、自适应网站

声明：本网站发布的内容（图片、视频和文字）以用户投稿、用户转载内容为主，如果涉及侵权请尽快告知，我们将会在第一时间删除。文章观点不代表本网站立场，如需处理请联系客服。电话：028-86922220；邮箱：631063699@qq.com。内容未经允许不得转载，或转载时需注明来源：创新互联

猜你还喜欢下面的内容