pytorch 實現(xiàn)將自己的圖片數(shù)據(jù)處理成可以訓(xùn)練的圖片類型
更新時間：2020年01月08日 15:11:29 作者：冰菓(笑)
今天小編就為大家分享一篇pytorch 實現(xiàn)將自己的圖片數(shù)據(jù)處理成可以訓(xùn)練的圖片類型，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
為了使用自己的圖像數(shù)據(jù)，需要仿照pytorch數(shù)據(jù)輸入創(chuàng)建新的類，其中數(shù)據(jù)格式為numpy.ndarray。
將自己的圖片保存到numpy.ndarray中，然后創(chuàng)建類
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 實現(xiàn)將自己的圖片數(shù)據(jù)處理成可以訓(xùn)練的圖片類型就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關(guān)文章
Nginx+Uwsgi+Django 項目部署到服務(wù)器的思路詳解
這篇文章主要介紹了Nginx+Uwsgi+Django 項目部署到服務(wù)器的思路，本文通過實例代碼給大家介紹的非常詳細，對大家的學習或工作具有一定的參考借鑒價值,需要的朋友可以參考下2020-05-05
淺談django2.0 ForeignKey參數(shù)的變化
今天小編就為大家分享一篇淺談django2.0 ForeignKey參數(shù)的變化，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2019-08-08
在本篇文章里小編給大家整理了一篇關(guān)于利用python調(diào)用攝像頭的實例分析內(nèi)容，有需要的朋友們跟著參考下。2021-06-06
下面小編就為大家?guī)硪黄猵ython下載圖片實現(xiàn)方法(超簡單)。小編覺得挺不錯的，現(xiàn)在就分享給大家，也給大家做個參考。一起跟隨小編過來看看吧2017-07-07
這篇文章主要介紹了通過實例解析python創(chuàng)建進程常用方法,文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值,需要的朋友可以參考下2020-06-06
Python 安裝第三方庫 pip install 安裝慢安裝不上的解決辦法
很多朋友反映在使用pip install安裝python 第三方庫的過程中會出現(xiàn)網(wǎng)速很慢，或者是安裝下載到中途，停止，卡主，或者是下載報錯等問題，下面小編給大家?guī)砹私鉀Q方法，一起看看吧2019-06-06
python+pytest接口自動化之日志管理模塊loguru簡介
python中有一個用起來非常簡便的第三方日志管理模塊--loguru，不僅可以避免logging的繁瑣配置，而且可以很簡單地避免在logging中多進程多線程記錄日志時出現(xiàn)的問題，甚至還可以自定義控制臺輸出的日志顏色，接下來我們來學習怎么使用loguru模塊進行日志管理2022-05-05
今天小編就為大家分享一篇對python列表里的字典元素去重方法詳解，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2019-01-01
這篇文章主要介紹了Python3爬蟲之自動查詢天氣并實現(xiàn)語音播報，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧2019-02-02
在網(wǎng)絡(luò)工程中,Ping測試是一種常用的網(wǎng)絡(luò)診斷工具,用于檢查網(wǎng)絡(luò)連接的可達性和響應(yīng)時間,隨著Python編程語言的廣泛應(yīng)用,越來越多的網(wǎng)絡(luò)工程師開始使用Python進行自動化網(wǎng)絡(luò)測試和管理任務(wù),本篇文章將詳細介紹如何使用Python進行Ping測試,需要的朋友可以參考下2024-06-06