RuntimeError: Could not infer dtype of NoneType

RuntimeError: Could not infer dtype of NoneType

出错原因

dataloader在dataset中一个一个读取数据的时候遇到了脏数据(空值之类的),导致在网络推理的时候出现了错误。

解决方法

方法一:跳过该数据

在数据放入dataset中之前判断image 和 label是否为空,如果为空则跳过该数据,不把他存入dataset

方法二:使用collate_fn函数

https://blog.csdn.net/guyuealian/article/details/91129367

参考链接

别人的解决方案:
https://www.yht7.com/news/135031

判断数据是否为空用pandas.isnull(obj),记得import pandas
https://book.itheima.net/course/221/1270308811000782849/1271373716495802370

关于PyTorch读取数据的dataloader、dataset具体情况:
https://pytorch.org/tutorials/beginner/basics/data_tutorial.html

如何判断数组array是否为空:
https://blog.csdn.net/lcmssd/article/details/45013451?utm_medium=distribute.pc_aggpage_search_result.none-task-blog-2aggregatepagefirst_rank_ecpm_v1~rank_v31_ecpm-1-45013451.pc_agg_new_rank&utm_term=pycharm+array+%E5%88%A4%E6%96%AD%E6%98%AF%E7%A9%BA%E7%9A%84&spm=1000.2123.3001.4430

你可能感兴趣的:(编程,机器学习,pytorch,深度学习,python)