快捷導(dǎo)航

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)文章

Nginx+Uwsgi+Django 項(xiàng)目部署到服務(wù)器的思路詳解
這篇文章主要介紹了Nginx+Uwsgi+Django 項(xiàng)目部署到服務(wù)器的思路，本文通過(guò)實(shí)例代碼給大家介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值,需要的朋友可以參考下
2020-05-05
淺談django2.0 ForeignKey參數(shù)的變化
今天小編就為大家分享一篇淺談django2.0 ForeignKey參數(shù)的變化，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧
2019-08-08
利用python調(diào)用攝像頭的實(shí)例分析
在本篇文章里小編給大家整理了一篇關(guān)于利用python調(diào)用攝像頭的實(shí)例分析內(nèi)容，有需要的朋友們跟著參考下。
2021-06-06
python下載圖片實(shí)現(xiàn)方法(超簡(jiǎn)單)
下面小編就為大家?guī)?lái)一篇python下載圖片實(shí)現(xiàn)方法(超簡(jiǎn)單)。小編覺(jué)得挺不錯(cuò)的，現(xiàn)在就分享給大家，也給大家做個(gè)參考。一起跟隨小編過(guò)來(lái)看看吧
2017-07-07
通過(guò)實(shí)例解析python創(chuàng)建進(jìn)程常用方法
這篇文章主要介紹了通過(guò)實(shí)例解析python創(chuàng)建進(jìn)程常用方法,文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下
2020-06-06
Python 安裝第三方庫(kù) pip install 安裝慢安裝不上的解決辦法
很多朋友反映在使用pip install安裝python 第三方庫(kù)的過(guò)程中會(huì)出現(xiàn)網(wǎng)速很慢，或者是安裝下載到中途，停止，卡主，或者是下載報(bào)錯(cuò)等問(wèn)題，下面小編給大家?guī)?lái)了解決方法，一起看看吧
2019-06-06
python+pytest接口自動(dòng)化之日志管理模塊loguru簡(jiǎn)介
python中有一個(gè)用起來(lái)非常簡(jiǎn)便的第三方日志管理模塊--loguru，不僅可以避免logging的繁瑣配置，而且可以很簡(jiǎn)單地避免在logging中多進(jìn)程多線程記錄日志時(shí)出現(xiàn)的問(wèn)題，甚至還可以自定義控制臺(tái)輸出的日志顏色，接下來(lái)我們來(lái)學(xué)習(xí)怎么使用loguru模塊進(jìn)行日志管理
2022-05-05
對(duì)python列表里的字典元素去重方法詳解
今天小編就為大家分享一篇對(duì)python列表里的字典元素去重方法詳解，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧
2019-01-01
Python3爬蟲(chóng)之自動(dòng)查詢(xún)天氣并實(shí)現(xiàn)語(yǔ)音播報(bào)
這篇文章主要介紹了Python3爬蟲(chóng)之自動(dòng)查詢(xún)天氣并實(shí)現(xiàn)語(yǔ)音播報(bào)，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧
2019-02-02
使用Python進(jìn)行Ping測(cè)試的操作指南
在網(wǎng)絡(luò)工程中,Ping測(cè)試是一種常用的網(wǎng)絡(luò)診斷工具,用于檢查網(wǎng)絡(luò)連接的可達(dá)性和響應(yīng)時(shí)間,隨著Python編程語(yǔ)言的廣泛應(yīng)用,越來(lái)越多的網(wǎng)絡(luò)工程師開(kāi)始使用Python進(jìn)行自動(dòng)化網(wǎng)絡(luò)測(cè)試和管理任務(wù),本篇文章將詳細(xì)介紹如何使用Python進(jìn)行Ping測(cè)試,需要的朋友可以參考下
2024-06-06