pytorch dataloader 取batch_size時(shí)候出現(xiàn)bug的解決方式
1、
RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0. Got 342 and 281 in dimension 3 at /pytorch/aten/src/TH/generic/THTensorMoreMath.cpp:1333
2、
RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0. Got 3 and 1 in dimension 1
bug1 修改方法是將圖片resize 成一個(gè)尺度：
解決1：
如果用transforms來resize ，關(guān)于尺寸的transform就是transforms.Resize(input_size, interpolation=3)了。
查看Resize的文檔:
size (sequence or int) – Desired output size. If size is a sequence like (h, w), output size will be matched to this.
If size is an int, smaller edge of the image will be matched to this number. i.e, if height > width, then image will be rescaled to (size * height / width, size)
size這個(gè)參數(shù)既可以是一個(gè)數(shù)字，又可以是一個(gè)tuple，如果是數(shù)字，圖片處理后的輸出尺寸每次都會(huì)被重新計(jì)算為(input_size*height/width, size)，這樣如果輸入圖片尺寸不一致，輸出圖片尺寸會(huì)不一致， 將導(dǎo)致bug1 問題
解決2：
也可以直接用opencv resize
以上這篇pytorch dataloader 取batch_size時(shí)候出現(xiàn)bug的解決方式就是小編分享給大家的全部內(nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。
相關(guān)文章
這篇文章主要介紹了Django框架使用mysql視圖操作,結(jié)合實(shí)例形式分析了mysql視圖的創(chuàng)建、查看以及Django框架使用mysql視圖相關(guān)操作技巧,需要的朋友可以參考下2019-05-05
Python連接SQLite數(shù)據(jù)庫并進(jìn)行增冊(cè)改查操作方法詳解
這篇文章主要介紹了Python對(duì)SQLite數(shù)據(jù)庫進(jìn)行增冊(cè)改查操作方法詳解,需要的朋友可以參考下2020-02-02
這篇文章主要為大家介紹了Pandas字符串操作的各種方法及速度測(cè)試,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進(jìn)步,早日升職加薪2023-08-08
使用python爬取連續(xù)降水?dāng)?shù)據(jù)信息實(shí)例
這篇文章主要為大家介紹了使用python提取連續(xù)降水?dāng)?shù)據(jù)信息實(shí)現(xiàn)示例,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進(jìn)步,早日升職加薪2024-01-01
python程序中的線程操作 concurrent模塊使用詳解
這篇文章主要介紹了python程序中的線程操作 concurrent模塊使用詳解,文中通過示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下2019-09-09
python中集合set是一個(gè)無序不重復(fù)元素的集，基本功能包括關(guān)系測(cè)試和消除重復(fù)元素，還可以計(jì)算交集、差集、并集等，它與列表（list）的行為類似，這篇文章主要介紹了python?set()去重的底層原理,需要的朋友可以參考下2022-01-01
在本篇文章里小編給大家整理的是關(guān)于python在不同條件下的輸入與輸出，有需要的朋友們可以學(xué)習(xí)下。2020-02-02
使用opencv對(duì)圖像進(jìn)行編碼，一方面是圖像二進(jìn)制傳輸?shù)男枰?，另一方面?duì)圖像壓縮。本文主要介紹了python cv2圖像質(zhì)量壓縮的算法示例，感興趣的可以了解一下2021-06-06
下面小編就為大家分享一篇numpy.delete刪除一列或多列的方法，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過來看看吧2018-04-04
linux之父進(jìn)程使用kill函數(shù)殺死子進(jìn)程方式
這篇文章主要介紹了linux之父進(jìn)程使用kill函數(shù)殺死子進(jìn)程方式，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。如有錯(cuò)誤或未考慮完全的地方，望不吝賜教2023-06-06