def mask2onehot(mask, num_classes):
# mask为灰度图,(H,W)→(K,H,W)
_mask = [mask == i for i in range(num_classes)]
return np.array(_mask).astype(np.uint8)
Pytorch:One-hot编码
最新推荐文章于 2024-01-11 01:55:37 发布
def mask2onehot(mask, num_classes):
# mask为灰度图,(H,W)→(K,H,W)
_mask = [mask == i for i in range(num_classes)]
return np.array(_mask).astype(np.uint8)