RuntimeError: Could not infer dtype of NoneType

当使用PyTorch的DataLoader加载数据时,可能会遇到脏数据,如空值,导致运行时错误。解决这个问题的方法包括在数据进入Dataset前检查并跳过空值,或者自定义collate_fn函数来处理异常。参考链接提供了具体的实现细节和示例代码,帮助开发者有效地过滤和管理数据加载过程中的问题。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

RuntimeError: Could not infer dtype of NoneType

出错原因

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

解决方法

方法一:跳过该数据

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

方法二:使用collate_fn函数

https://blog.youkuaiyun.com/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.youkuaiyun.com/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

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值