图像分割 图像分割 图像分割
Realize:代码实现
Teach:代码讲解
深度学习—图像分割Demo代码(使用PyTorch进行训练、预测、TensorRT部署)
1.FCN-2015
2.U-Net-2015
3.FusionNet-2016
4.DeepLab-3.5系列-2017-2018
5.DeconvNet-2015
6.SegNet-2017
7.PSPNet-2017
8.Enet-2016
9.LinkNet-2017
10.GCN-2017
11.ExFuseNet-2018
12.DFN-2018
13.BiSeNet-2018
14.DFANet-2019
15.RedNet-2018
16.DANet-2019
17. U-Net++___2018
18. U-Net+++___2020
19.UNet使用ResNet系列作为Encoder
系列专题 系列专题 系列专题
U-Net-Famliy:
转TorchScript
net.eval()
trace = torch.jit.trace(net, torch.randn(1, 3, 512, 512))
torch.jit.save(trace,'UNet_model.pt')