Warning: masked_scatter_ received a mask with dtype torch.uint8, this behavior is now deprecated,ple
Warning: masked_scatter_ received a mask with dtype torch.uint8, this behavior is now deprecated,please use a mask with dtype torch.bool instead查看代码中加载训练数据集的地方,在生成mask标签的函数中将return中mask换为mask.bool()即可: heatmaps, mask = generate_label_map(Hpoint, heig
原创
2021-01-14 11:41:42 ·
2707 阅读 ·
0 评论