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
|