2021
07-22
07-22
解决pytorch读取自制数据集出现过的问题
问题1问题描述:TypeError:default_collate:batchmustcontaintensors,numpyarrays,numbers,dictsorlists;found<class'PIL.Image.Image'>解决方式数据格式不对,把image转成tensor,参数transform进行如下设置就可以了:transform=transform.ToTensor()。注意检测一下transform问题2问题描述:TypeError:append()takesexactlyoneargument(2given)出现问题的地方imgs.append(words[0],int(words[1]))解决方式...
继续阅读 >