pytorch 實(shí)現(xiàn)將自己的圖片數(shù)據(jù)處理成可以訓(xùn)練的圖片類(lèi)型
更新時(shí)間：2020年01月08日 15:11:29 作者：冰菓(笑)
今天小編就為大家分享一篇pytorch 實(shí)現(xiàn)將自己的圖片數(shù)據(jù)處理成可以訓(xùn)練的圖片類(lèi)型，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧
為了使用自己的圖像數(shù)據(jù)，需要仿照pytorch數(shù)據(jù)輸入創(chuàng)建新的類(lèi)，其中數(shù)據(jù)格式為numpy.ndarray。
將自己的圖片保存到numpy.ndarray中，然后創(chuàng)建類(lèi)
from torch.utils.data import Dataset
import numpy as np
class Dataset(Dataset):
def __init__(self, path_img, path_target, transforms=None):
self.train = path_img
self.targets = path_target
self.transforms = transforms
def __len__(self):
return len(self.train)
def __getitem__(self, idx):
img = self.train[idx]
target = self.targets[idx]
if self.transforms:
img = self.transforms(img)
target = self.transforms(target)
return img, target
使用方法和Mnist數(shù)據(jù)一樣的使用方法
isbi = Dataset(imgs_train, imgs_mask_train,
transforms=transform)
dataload=torch.utils.data.DataLoader(isbi,batch_size=4,shuffle=True)
for i, data in enumerate(dataload, 1):
img,label=data
print img.shape
print img.shape
print 10*'*'
以上這篇pytorch 實(shí)現(xiàn)將自己的圖片數(shù)據(jù)處理成可以訓(xùn)練的圖片類(lèi)型就是小編分享給大家的全部?jī)?nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。
相關(guān)文章
uwsgi+nginx部署Django項(xiàng)目操作示例
這篇文章主要介紹了uwsgi+nginx部署Django項(xiàng)目操作,結(jié)合實(shí)例形式簡(jiǎn)單介紹了uwsgi的概念、原理、安裝、項(xiàng)目創(chuàng)建、配置、調(diào)試運(yùn)行等相關(guān)操作技巧,需要的朋友可以參考下2018-12-12
這篇文章主要介紹了Python3自定義json逐層解析器代碼，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2020-05-05
今天小編就為大家分享一篇python解析json串與正則匹配對(duì)比方法，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2018-12-12
這篇文章主要介紹了在Python的Django框架中為代碼添加注釋的方法,需要的朋友可以參考下2015-07-07
Python MOCK SERVER moco模擬接口測(cè)試過(guò)程解析
這篇文章主要介紹了Python MOCK SERVER moco模擬過(guò)程解析,文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下2020-04-04
win7上tensorflow2.2.0安裝成功 引用DLL load failed時(shí)找不到指定模塊 tensorflo
這篇文章主要介紹了win7上tensorflow2.2.0安裝成功 引用時(shí)DLL load failed找不到指定模塊 tensorflow has no attribute xxx 解決方法,需要的朋友可以參考下2020-05-05
今天小編就為大家分享一篇基于python指定包的安裝路徑方法，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2018-10-10
Python字典常見(jiàn)操作實(shí)例小結(jié)【定義、添加、刪除、遍歷】
這篇文章主要介紹了Python字典常見(jiàn)操作,結(jié)合實(shí)例形式總結(jié)分析了Python字典的定義、添加、刪除、遍歷等常見(jiàn)使用技巧與操作注意事項(xiàng),需要的朋友可以參考下2019-10-10
python mysqldb連接數(shù)據(jù)庫(kù)
今天無(wú)事想弄下python做個(gè)gui開(kāi)發(fā)，最近發(fā)布的是python 3k，用到了數(shù)據(jù)庫(kù)，通過(guò)搜索發(fā)現(xiàn)有一個(gè)mysqldb這樣的控件，可以使用，就去官方看了下結(jié)果，沒(méi)有2.6以上的版本2009-03-03
簡(jiǎn)單的python協(xié)同過(guò)濾程序?qū)嵗a
這篇文章主要介紹了簡(jiǎn)單的python協(xié)同過(guò)濾程序，分享了相關(guān)代碼示例，小編覺(jué)得還是挺不錯(cuò)的，具有一定借鑒價(jià)值，需要的朋友可以參考下2018-01-01