2和3都有:
https://github.com/Zxl19990529/UNet3_plus
unet3 在圆分割项目中表现比较好:
可以支持动态分辨率,但是用到了
import torch.nn.functional as torch_f
可能onnx,tensorrt,mnn等不支持,
https://blog.csdn.net/jacke121/article/details/122375415
本文结尾是unet3的代码,估计可以转onnx,tensorrt等到。
这个分训练,测试,有点奇怪:
https://github.com/hellopipu/unet_plus/blob/master/model.py
32*320 140 ms
下面两个相似: