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