Tensorflow、Keras 实现的 U-net、V-net、U-net++、UNET 3+、Attention U-net、R2U-net、ResUnet-a、U^2-Net、TransUNET 和 Swin-UNET，其中可选的 Ima
Tensorflow、Keras 实现的 U-net、V-net、U-net++、UNET 3+、Attention U-net、R2U-net、ResUnet-a、U^2-Net、TransUNET 和 Swin-UNET，其中可选的 Ima
tensorflow.keras 实现了 U-net、V-net、U-net++、UNET 3+、Attention U-net、R2U-net、ResUnet-a、U^2-Net、TransUNET 和 Swin-UNET，其中可选的 ImageNet 训练的骨架。keras_unet_collection.models 包含用于为 keras 模型配置超参数选项的函数。
keras_unet_collection.models |
名称 | 参考 |
|---|---|---|
unet_2d |
U-net | Ronneberger 等人 (2015) |
vnet_2d |
V-net (针对 2D 输入进行修改) | Milletari 等人 (2016) |
unet_plus_2d |
U-net++ | Zhou 等人 (2018) |
r2_unet_2d |
R2U-Net | Alom 等人 (2018) |
att_unet_2d |
注意力 U-net | Oktay 等人 (2018) |
resunet_a_2d |
ResUnet-a | Diakogiannis 等人 (2020) |
u2net_2d |
U^2-Net | Qin 等人 (2020) |
unet_3plus_2d |
UNET 3+ | Huang 等人 (2020) |
transunet_2d |
TransUNET | Chen 等人 (2021) |
swin_unet_2d |
Swin-UNET | Hu 等人 (2021) |
**注意：**两个 Transformer 模型与 NumPy 1.20 不兼容；建议使用 NumPy 1.19.5。
暂无开放 Issues，或尚未同步最近议题。