TypeError: ‘>‘ not supported between instances of ‘list‘ and ‘int‘

将标签中大于0的像素值(类别)挑选出来。

label = [0,1,2,3]
mask = label > 0
print(mask)

运行时候出现:TypeError: ‘>’ not supported between instances of ‘list’ and ‘int’
因为label是list不能和0比较,所以需要对label格式进行修改。

添加一句:

label = torch.Tensor(label)

或者

label = np.numpy(label)

取决于自己的数据类型,在训练过程中,label已经加载到cuda上了,所以他一定是一个tensor格式,训练时候不必担心。

返回一个布尔类型:

tensor([False,  True,  True,  True])

你可能感兴趣的:(pytorch踩坑,python,深度学习,人工智能)