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的解決方式就是小編分享給大家的全部?jī)?nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。
相關(guān)文章
Pandas 解決dataframe的一列進(jìn)行向下順移問題
今天小編就為大家分享一篇Pandas 解決dataframe的一列進(jìn)行向下順移問題，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過來看看吧2019-12-12
Python 有很多黑魔法，為了不分你的心，今天只講 metaclass。對(duì)于 metaclass 這種特性，有兩種極端的觀點(diǎn)：下面小編將為大家詳細(xì)的介紹，剛興趣的小伙伴可以參考一下2021-09-09
Tensorflow訓(xùn)練模型越來越慢的2種解決方案
今天小編就為大家分享一篇Tensorflow訓(xùn)練模型越來越慢的2種解決方案，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過來看看吧2020-02-02
Python Flask 請(qǐng)求數(shù)據(jù)獲取響應(yīng)詳解
這篇文章主要介紹了Python Flask請(qǐng)求數(shù)據(jù)獲取響應(yīng)的實(shí)現(xiàn)方法，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過來看看吧2021-10-10
python 實(shí)現(xiàn)簡(jiǎn)易的記事本
這篇文章主要介紹了python 實(shí)現(xiàn)簡(jiǎn)易的記事本的示例代碼，幫助大家更好的理解和學(xué)習(xí)python，感興趣的朋友可以了解下2020-11-11
Python命令啟動(dòng)Web服務(wù)器實(shí)例詳解
這篇文章主要介紹了Python命令啟動(dòng)Web服務(wù)器實(shí)例詳解的相關(guān)資料,需要的朋友可以參考下2017-02-02
這篇文章主要介紹了Keras—embedding嵌入層的用法詳解，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過來看看吧2020-06-06
SpringBoot實(shí)現(xiàn)登錄注冊(cè)常見問題解決方案
這篇文章主要介紹了SpringBoot實(shí)現(xiàn)登錄注冊(cè)常見問題解決方案,文中通過示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下2020-03-03
今天小編就為大家分享一篇Anaconda 離線安裝 python 包的操作方法，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過來看看吧2018-06-06
Python基于Tkinter開發(fā)一個(gè)爬取B站直播彈幕的工具
這篇文章主要介紹了Python Tkinter如何開發(fā)一個(gè)爬取B站直播彈幕的工具，幫助大家更好的利用python進(jìn)行圖形界面的開發(fā)學(xué)習(xí)，感興趣的朋友可以了解下2021-05-05