首先我想說我是 PyTorch 的新手，所以如果我的問題水平不是那么高，我很抱歉。我想知道您是否可以幫我解決一些問題（我實(shí)際上有兩個(gè)問題）。它們背后的故事：我正在研究圖像分類。我的測試數(shù)據(jù)根據(jù)標(biāo)簽分為子文件夾，我通過 DataLoader 加載它們。第一個(gè)問題：1）如果你用特定的batch size訓(xùn)練你的模型，用其他sizes測試它會(huì)影響準(zhǔn)確率嗎？2) 有沒有辦法加載和使用帶有位于單個(gè)文件夾（沒有子文件夾）中的測試數(shù)據(jù)的模型。據(jù)我所知，DataLoader 需要子文件夾。
1 回答
不負(fù)相思意
TA貢獻(xiàn)1777條經(jīng)驗(yàn) 獲得超10個(gè)贊
這取決于您是否使用依賴于批次中其他項(xiàng)目的操作。如果您正在使用諸如批量標(biāo)準(zhǔn)化之類的東西，它可能會(huì)，但一般來說，如果您的網(wǎng)絡(luò)單獨(dú)處理批量項(xiàng)目，則不會(huì)。
如果您查看torch.utils.data.Dataset的文檔，您會(huì)發(fā)現(xiàn)數(shù)據(jù)集本質(zhì)上只需要
__len__和__getitem__方法，其中前者表示數(shù)據(jù)集包含多少個(gè)項(xiàng)目，后者獲取第ith 個(gè)項(xiàng)目 - 無論是圖像還是標(biāo)簽、圖像及其分割掩碼或其他東西。沒有什么可以阻止您編寫自定義數(shù)據(jù)集。我建議您查看DatasetFolder 的源代碼并根據(jù)您的需要進(jìn)行修改。
添加回答
舉報(bào)
0/150
提交
取消