1、Unet网络推理:用Vgg网络作为主干网络
电脑路径:
libtorch Gpu,图片大小需要改为640 *640,推理时间为:训练时间结果为:
2、Unet网络推理:用Unet本身网络
电脑路径:
图片大小需要改为640 *640,推理时间为:0.2s
3、Swin_Unet网络推理:利用注意力机制
电脑路径:
图片大小需要改为224 *224,推理时间为:0.2s左右
4、deeplabv3网络推理:利用mobilenet或者xception作为主干网络
电脑路径:
图片大小需要改为640 *640,推理时间为:0.6s左右
5、TransUNet网络推理:
电脑路径:
图片大小需要改为320 *320,推理时间为:0.27s左右