摘要:
n = 5 #类别数 indices = torch.randint(0, n, size=(15,15)) #生成数组元素0~5的二维数组(15*15) one_hot = torch.nn.functional.one_hot(indices, n) #size=(15, 15, n) 1. O 阅读全文
摘要:
最近一直在学pytorch,copy了几个经典的入门问题。现在作一下总结。 首先,做的小项目主要有 分类问题:Mnist手写体识别、FashionMnist识别、猫狗大战 语义分割:Unet分割肝脏图像、遥感图像 先把语义分割的心得总结一下,目前只是一部分,以后还会随着学习的深入慢慢往里面加新的感悟 阅读全文