【Python solution】TypeError: can‘t convert np.ndarray of type numpy.bool_.解决方案

问题:

在运行代码masks = torch.as_tensor(masks, dtype=torch.uint8)
出现了TypeError: can’t convert np.ndarray of type numpy.bool_.,而这个是因为不能将bool类型的转化,表明了np.bool和tensor之间是不能直接转化的。

解决方案:

而实际上这个解决方案已经包含在了后面的信息中,
TypeError: can’t convert np.ndarray of type numpy.bool_. The only supported types are: double, float, float16, int64, int32, and uint8.
这个即表示了仅仅只有后面这些类型的可以转换。

因此我上面写的代码中,masks是一个bool值,因此为了转化为tensor的uint8类型,我选择在array中先转化为uint8 ,于是写下了如下的代码

masks = masks.astype(np.uint8)
masks = torch.as_tensor(masks,dytpe=torch.uint8)

而第一行代码即先在np.array中先转换为uint8,满足要求后再转化为tensor

总结

虽然这个问题解决了,但是还是不清楚哪个版本可以运行,哪些版本不能运行,我的版本应该是torch1.0.1版本的,出现了该错误。等找到原因后再来更新。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值