参考了部分开源代码,汇总了一些常用的网络架构;
官网下载模型,本地加载即可使用预训练模型;(代码中附下载地址)
目前包含模型:
ResNet50;
NasNetlarge;
shuffleNetV2;
mobileNetV2;
SENet152;
BoTNet(Bottleneck Transformers for Visual Recognition-2021);
VIT(Vision Transformer-2020);
HRNet(High-resolution networks);
GhostNet(CVPR2020);
随时更新;
仅供学习参考;