2023年4月14日
Pytorch one-hot编码
摘要： 1. 引言 在我们做分割任务时，通常会给一个mask，但训练时要进行onehot编码。 2. code import torch if __name__ == '__main__': label = torch.zeros(size=(1, 4, 4), dtype=torch.int) label 阅读全文
posted @ 2023-04-14 10:04 抚琴尘世客 阅读(143) 评论(0) 推荐(0) 编辑