Pytorch使用技巧之Dataloader中的collate_fn參數(shù)詳析

更新時(shí)間：2022年03月17日 10:17:29 作者：政在學(xué)習(xí)

collate_fn 參數(shù)的目的主要是為了隨心所欲的轉(zhuǎn)變數(shù)據(jù)的類型,這個(gè)數(shù)據(jù)是用DataLoader加載的,比如img，target,下面這篇文章主要給大家介紹了關(guān)于Pytorch使用技巧之Dataloader中的collate_fn參數(shù)的相關(guān)資料,需要的朋友可以參考下

以MNIST為例

from torchvision import datasets
mnist = datasets.MNIST(root='./data/', train=True, download=True)
print(mnist[0])

結(jié)果

(<PIL.Image.Image image mode=L size=28x28 at 0x196E3F1D898>, 5)

MINIST數(shù)據(jù)集的dataset是由一張圖片和一個(gè)label組成的元組

dataloader = torch.utils.data.DataLoader(dataset=mnist, batch_size=2, shuffle=True,collate_fn=lambda x:x)
for each in dataloader:
    print(each)
    break

結(jié)果

[(<PIL.Image.Image image mode=L size=28x28 at 0x2CB3B105630>, 0), (<PIL.Image.Image image mode=L size=28x28 at 0x2CB3B105668>, 2)]

collate_fn為lamda x:x時(shí)表示對(duì)傳入進(jìn)來的數(shù)據(jù)不做處理

下面自定義collate_fn看看什么效果

def collate(data):
    img = []
    label = []
    for each in data:
        img.append(each[0])
        label.append(each[1])
    return img,label
dataloader = torch.utils.data.DataLoader(dataset=mnist, batch_size=2, shuffle=True,collate_fn=lambda x:collate(x))
for each in dataloader:
    print(each)
    break

結(jié)果

([<PIL.Image.Image image mode=L size=28x28 at 0x241433A36D8>, <PIL.Image.Image image mode=L size=28x28 at 0x241433A3710>], [9, 3])

說明：若不設(shè)置collate_fn參數(shù)則會(huì)使用默認(rèn)處理函數(shù)

但必須保證傳進(jìn)來的數(shù)據(jù)都是tensor格式否則會(huì)報(bào)錯(cuò)

附：DataLoader完整的參數(shù)表如下：

class torch.utils.data.DataLoader(
    dataset,
    batch_size=1,
    shuffle=False,
    sampler=None,
    batch_sampler=None,
    num_workers=0,
    collate_fn=<function default_collate>,
    pin_memory=False,
    drop_last=False,
    timeout=0,
    worker_init_fn=None)

DataLoader在數(shù)據(jù)集上提供單進(jìn)程或多進(jìn)程的迭代器

幾個(gè)關(guān)鍵的參數(shù)意思：

- shuffle：設(shè)置為True的時(shí)候，每個(gè)世代都會(huì)打亂數(shù)據(jù)集

- collate_fn：如何取樣本的，我們可以定義自己的函數(shù)來準(zhǔn)確地實(shí)現(xiàn)想要的功能

- drop_last：告訴如何處理數(shù)據(jù)集長度除于batch_size余下的數(shù)據(jù)。True就拋棄，否則保留

總結(jié)

到此這篇關(guān)于Pytorch使用技巧之Dataloader中的collate_fn參數(shù)的文章就介紹到這了,更多相關(guān)Dataloader中的collate_fn參數(shù)內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章:

相關(guān)文章

淺析NumPy 切片和索引
這篇文章主要介紹了NumPy 切片和索引的相關(guān)資料，幫助大家更好的理解和學(xué)習(xí)NumPy的相關(guān)知識(shí)，感興趣的朋友可以了解下。
2020-09-09
Python編程之順序執(zhí)行與程序的主入口詳解
程序從程序入口進(jìn)入,到程序執(zhí)行結(jié)束,大體是按照順序結(jié)構(gòu)執(zhí)行語句、函數(shù)或代碼塊,掌握程序的結(jié)構(gòu),有利于把握程序的主體框架,下面這篇文章主要給大家介紹了關(guān)于Python編程之順序執(zhí)行與程序的主入口的相關(guān)資料,需要的朋友可以參考下
2022-12-12
Python模塊Typing.overload的使用場(chǎng)景分析
在 Python 中,typing.overload 是一個(gè)用于定義函數(shù)重載的裝飾器,函數(shù)重載是指在一個(gè)類中可以定義多個(gè)相同名字但參數(shù)不同的函數(shù),使得在調(diào)用函數(shù)時(shí)可以根據(jù)參數(shù)的不同選擇不同的函數(shù)執(zhí)行,這篇文章主要介紹了Python模塊Typing.overload的使用,需要的朋友可以參考下
2024-02-02
Python實(shí)現(xiàn)生命游戲的示例代碼(tkinter版)
生命游戲是由劍橋大學(xué)約翰·何頓·康威設(shè)計(jì)的計(jì)算機(jī)程序，一時(shí)吸引了各行各業(yè)一大批人的興趣。本文將用Python實(shí)現(xiàn)這一游戲，感興趣的可以嘗試一下
2022-08-08
Pytorch-mlu?實(shí)現(xiàn)添加逐層算子方法詳解
本文主要分享了在寒武紀(jì)設(shè)備上?pytorch-mlu?中添加逐層算子的方法教程，代碼具有一定學(xué)習(xí)價(jià)值，有需要的朋友可以借鑒參考下，希望能夠有所幫助
2021-11-11
python批量修改文件名的示例
這篇文章主要介紹了python批量修改文件名的示例，幫助大家更好的使用python處理文件，感興趣的朋友可以了解下
2020-09-09
Python編程應(yīng)用設(shè)計(jì)原則詳解
什么是好用的代碼呢？其實(shí)就是代碼質(zhì)量比較高，如何評(píng)價(jià)代碼質(zhì)量的高低呢？最常用的、最重要的評(píng)價(jià)標(biāo)準(zhǔn)，就是代碼的可維護(hù)性、可讀性、可擴(kuò)展性、靈活性、簡潔性、可復(fù)用性、可測(cè)試性
2021-09-09
python實(shí)現(xiàn)將兩個(gè)文件夾合并至另一個(gè)文件夾(制作數(shù)據(jù)集)
這篇文章主要介紹了python實(shí)現(xiàn)將兩個(gè)文件夾合并至另一個(gè)文件夾(制作數(shù)據(jù)集)，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過來看看吧
2020-04-04
Python實(shí)現(xiàn)的根據(jù)IP地址計(jì)算子網(wǎng)掩碼位數(shù)功能示例
這篇文章主要介紹了Python實(shí)現(xiàn)的根據(jù)IP地址計(jì)算子網(wǎng)掩碼位數(shù)功能,涉及Python數(shù)值運(yùn)算相關(guān)操作技巧,需要的朋友可以參考下
2018-05-05
Django實(shí)現(xiàn)組合搜索的方法示例
本篇文章主要介紹了Django實(shí)現(xiàn)組合搜索的方法示例，小編覺得挺不錯(cuò)的，現(xiàn)在分享給大家，也給大家做個(gè)參考。一起跟隨小編過來看看吧
2018-01-01