项目作者: 1044197988
项目描述 :
基于Tensorflow的常用模型,包括分类分割、新型激活、卷积模块,可在Tensorflow2.0下运行。
高级语言: Python
项目地址: git://github.com/1044197988/TF.Keras-Commonly-used-models.git
TF.Keras-常用型号
自己整理的一些tensorflow下ķeras实现的模型,可在Tensorflow2.X下运行
提示:以下模型均不包含预训练权重的载入,只是模型的实现;不同的卷积模块大部分在分类分割模型中已包含。
分类模型:
- AlexNet
- Darknet53
- DenseNet
- Dual_path_network
- GoogleNet
- MNasNet
- Resnet34
- Resnet50
- SEResNeXt
- VGG16
- Squeeze_Excite-Network
- MobileNetV3
- Efficientnet
- SE_HRNet
- ResNest
分割模型:
分割损失函数:
- Focal_Tversky_loss
- C_Focal_loss
- B_Focal_loss
- LovaszSoftmax
- WeightedCCE
- jaccard_loss
- bce_jaccard_loss
- cce_jaccard_loss
- dice_loss
- bce_dice_loss
- cce_dice_loss
分割指标:
- iou_score
- jaccard_score
- f1_score
- f2_score
- dice_score
新型激活函数:
卷积模块:
Layer:
- FRN
- attention(PAM空间注意力和CAM通道注意力)
- BiFPN
Others: