图像分割图像分割图像分割
Realize:代码实现
Teach:代码讲解
深度学习—图像分割Demo代码&＃xff08;使用PyTorch进行训练、预测、TensorRT部署&＃xff09;
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&＃43;&＃43;___2018
18. U-Net&＃43;&＃43;&＃43;___2020
19.UNet使用ResNet系列作为Encoder
U-Net-Famliy:
- U-Net___2015
- U-Net&＃43;&＃43;___2018
- U-Net&＃43;&＃43;&＃43;___2020
转TorchScript
net.eval()
trace &＃61; torch.jit.trace(net, torch.randn(1, 3, 512, 512))
torch.jit.save(trace,&＃39;UNet_model.pt&＃39;)
京公网安备 11010802041100号