數(shù)據(jù)處理

創(chuàng)新互聯(lián)-云計算及IDC服務提供商，涵蓋公有云、IDC機房租用、托管服務器、等保安全、私有云建設等企業(yè)級互聯(lián)網(wǎng)基礎服務，聯(lián)系熱線：18982081108

版本1

#數(shù)據(jù)處理
import os
import torch
from torch.utils import data
from PIL import Image
import numpy as np

#定義自己的數(shù)據(jù)集合
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數(shù)據(jù)處理:定義自己的數(shù)據(jù)集合實例-創(chuàng)新互聯(lián)
本文鏈接：http://chinadenli.net/article22/dpipjc.html

成都網(wǎng)站建設公司_創(chuàng)新互聯(lián)，為您提供軟件開發(fā)、網(wǎng)站設計、品牌網(wǎng)站建設、網(wǎng)站維護、定制開發(fā)、自適應網(wǎng)站

廣告

聲明：本網(wǎng)站發(fā)布的內容（圖片、視頻和文字）以用戶投稿、用戶轉載內容為主，如果涉及侵權請盡快告知，我們將會在第一時間刪除。文章觀點不代表本網(wǎng)站立場，如需處理請聯(lián)系客服。電話：028-86922220；郵箱：631063699@qq.com。內容未經(jīng)允許不得轉載，或轉載時需注明來源： 創(chuàng)新互聯(lián)