To make it easier for developers to enjoy the benefits of Ascend ModelZoo, we are continuing to add typical networks and related pre-trained models. If you have any needs, open an issue on modelzoo/issues , and we will response to it in a timely manner.
Before you start, read the CONTRIBUTING first.
Thank you!
- 3D_ResNet_ID0421_for_PyTorch
- CRNN_for_PyTorch
- DeepMar_for_PyTorch
- Densenet121_for_PyTorch
- DenseNet161_ID0455_for_PyTorch
- DenseNet169_ID0454_for_PyTorch
- DenseNet201_ID0453_for_PyTorch
- EfficientNet-B1_ID1713_for_PyTorch
- EfficientNet-B2_ID1714_for_PyTorch
- EfficientNet-B3_ID0450_for_PyTorch
- EfficientNet-B4_ID1632_for_PyTorch
- EfficientNet-B5_ID1633_for_PyTorch
- EfficientNet_for_PyTorch
- FaceNet_for_PyTorch
- Gluon_ResNet50_v1b_for_PyTorch
- Gluon_ResNet50_v1c_for_PyTorch
- Gluon_ResNet50_v1d_for_PyTorch
- Googlenet_ID0447_for_PyTorch
- MobileNetV1_ID0094_for_PyTorch
- MobileNetV2_for_PyTorch
- MobileNetV3-Large_ID1784_for_PyTorch
- ResNest_ID0426_for_PyTorch
- ResNet50_for_PyTorch
- Shufflenetv2_for_PyTorch
- DAL_ID2732_for_PyTorch
- DB_ID0706_for_PyTorch
- Faster_Mask_RCNN_for_PyTorch
- PSENet_for_PyTorch
- RetinaNet_for_PyTorch
- RFCN_ID0418_for_PyTorch
- YoloV3_ID1790_for_PyTorch
- YOLOV4_ID0396_for_PyTorch
- Attention_R2U_Net_for_PyTorch
- AttU_Net_for_PyTorch
- DeepLabv3+_ID1695_for_PyTorch
- R2U_Net_for_PyTorch
- Bert-Squad_ID0470_for_PyTorch
- CPM_Finetune_for_PyTorch
- FOTS_ID0338_for_PyTorch
- GRU_for_PyTorch
- LSTM_ID0468_for_PyTorch
- mBART_ID2372_for_PyTorch
- Transformer_for_PyTorch
- XLM_ID0740_for_PyTorch
- DCN_for_PyTorch
- DeepFM_for_PyTorch
- WDL_for_PyTorch
- baseline-rawnet
- deepspeech
- FastPitch
- speech-transformer
- Tacotron2_for_PyTorch
- tdnn
- WaveGlow
- 3d_attention_net
- AlexNet_ID2663_for_PyTorch
- AlignedReID
- csp_resnext50-mish
- Deit_Small
- DnCNN
- DPN-131_for_PyTorch
- Efficient-3DCNNs_ID1230_for_PyTorch
- EfficientNet-B1
- EfficientNet-B3
- EfficientNet-B5_ID1621_for_PyTorch
- FixRes
- GaitSet
- GENet_for_Pytorch
- GhostNet
- GoogleNet_ID1623_for_PyTorch
- HRNet_ID1780_for_PyTorch
- InceptionResNetV2_ID1779_for_PyTorch
- InceptionV3_ID1596_for_PyTorch
- InceptionV4_ID1778_for_PyTorch
- LResNet100E-IR
- MGN
- MnasNet
- MobileNet
- MobileNetV3_large_100_for_PyTorch
- Moco-v2
- NASNet-A-Mobile
- OSNet
- PCB
- PnasNet5Large
- PointNet
- pointnetCNN
- RegNetX
- RegNetY-1.6GF
- ReidStrongBaseline
- RepVGG
- Res2Net101_v1b
- ResNeSt50_for_PyTorch
- ResNet101_ID1595_for_PyTorch
- ResNet152
- ResNet18_ID1593_for_PyTorch
- ResNet34_ID1594_for_PyTorch
- ResNeXt-50-32x4d_ID1624_for_PyTorch
- ResNeXt101_32x8d_for_PyTorch
- SE-ResNet-50
- SE-ResNext-101-32x4d
- Se-ResNext-50-32x4d
- SENet154
- ShuffleNetV1_ID1625_for_PyTorch
- ShuffleNetV2Plus_ID1626_for_PyTorch
- SkresNet50
- SPNASNet_100_for_PyTorch
- SqueezeNet1_1
- Swin-Transformer
- TNT
- TResNet
- Vehicle_Re-Identification
- Vgg16_ID1630_for_PyTorch
- Vgg19_ID1631_for_PyTorch
- vit_base_patch32_224
- Vit_small_patch16_224
- VOLO
- VoVNet39
- WideResNet101_2_for_Pytorch
- WideResNet50_2_ID1627_for_PyTorch
- Xception_ID1777_for_PyTorch
- xcit
- AdvancedEAST
- CascadedMaskRCNN
- Cascade_RCNN
- CenterFace
- CenterNet
- DSFD
- EfficientDetD0
- FaceBoxes
- FairMOT
- FCOS
- FOTS
- FSAF_for_Pytorch
- GFocalV2
- M2Det
- NasFPN
- Pointnetplus
- Pyramidbox
- RCF
- RefineDet
- Retinaface
- RetinaNet
- SimCLR_for_Pytorch
- SOLOv1
- SOLOv2
- SSD
- SSD-MobileNetV1
- SSD-MobilenetV2
- SSD-Resnet
- StyleGAN2-ADA
- TextSnake
- YOLACT
- YOLACT_plus
- YOLOR
- YOLOX
- ADNet
- BigGAN
- CGAN
- Cross-Scale-Non-Local-Attention
- CycleGAN
- DCGAN
- edsr_x2
- GAN_Pytorch
- Pix2Pix
- RDN
- Srcnn_x2_for_Pytorch
- SRFlow
- SRGAN
- stargan
- wdsr
- 3Dmppe_RootNet
- AlphaPose
- DeepPose
- HigherHRNet
- Hourglass_for_PyTorch
- Lightweight_OpenPose
- MSPN
- ST-GCN
- TransPose
- VoxelPose
- 3DUNet
- DeeplabV3_for_Pytorch
- ENet
- ErfNet
- FastSCNN
- FCN8s
- HRnet-OCR
- HRNet_SEG_for_Pytorch
- ICNet_ID1781_for_PyTorch
- IntraDA
- PointRend
- PraNet
- PSPNet
- RCAN
- RefineNet
- SETR
- Ultra-Fast-Lane-Detection
- UNet++
- UNet_for_PyTorch
- VNet
- Wseg
- BMN
- BSN
- C3D
- GloRe
- I3D
- NonLocal
- R(2+1)D
- SiamFC
- SiamRPN
- SlowFast
- TSM
- TSN
- VideoPose3D
- X3D
- albert
- Bertsum_for_PyTorch
- roberta
- TextCNN
- tinybert
- C51
- DLRM
- DQN
- RotatE
- tacotron2_ID0406_for_PyTorch
- 3D_attentionnet_ID0478_for_PyTorch
- FasterRCNN-Resnet50-FPN_ID1552_for_PyTorch
- HRNet_ID0446_for_PyTorch
- PointNet_ID0430_for_PyTorch
- SSD-ResNet34_ID0411_for_PyTorch
- 2D_Unet_ID0624_for_PyTorch
- 2S-AGCN_ID0909_for_PyTorch
- 3D_Nest_Unet_ID0476_for_PyTorch
- ADACOS_ID1082_for_PyTorch
- AdaFM-Net_ID1101_for_PyTorch
- ADLayer_ID1087_for_PyTorch
- AdvancedEast_ID0473_for_PyTorch
- AlexNet_ID0472_for_PyTorch
- ANN_ID2370_for_PyTorch
- ArcFace_ID0852_for_PyTorch
- ATS_ID2682_for_PyTorch
- AUTOAUGMENT_ID0792_for_PyTorch
- BASNET_ID1134_for_PyTorch
- BertBase_ID0490_for_PyTorch
- CDAR_ID2747_for_PyTorch
- ConvLSTM_ID1772_for_PyTorch
- coral-cnn_ID1064_for_PyTorch
- CrossFormer_ID2449_for_PyTorch
- CycleGAN_ID0521_for_PyTorch
- DBPN_ID2917_for_PyTorch
- DCAP_ID2836_for_PyTorch
- DEEP-HEAD-POSE_ID0796_for_PyTorch
- DeepLab-CRF_ID1873_for_PyTorch
- DeepSort_ID0654_for_PyTorch
- deit_ID2467_for_PyTorch
- DGMS_ID2460_for_PyTorch
- EfficientNet-B6_ID1715_for_PyTorch
- EfficientNet-B7_ID1716_for_PyTorch
- ESPCN_ID2919_for_PyTorch
- ESPCN_ID3002_for_PyTorch
- ESRGAN_ID1813_for_PyTorch
- FasterRCNN_ID0100_for_PyTorch
- FFDNet_ID0970_for_PyTorch
- FixMatch_ID0989_for_PyTorch
- GENet_ID0671_for_PyTorch
- GhostNet_ID1622_for_PyTorch
- Hourglass_ID1809_for_PyTorch
- ICT_ID1179_for_PyTorch
- InceptionV1_ID1568_for_PyTorch
- InceptionV2_ID0698_for_PyTorch
- InceptionV3_ID0445_for_PyTorch
- InceptionV4_ID0444_for_PyTorch
- InceptionV4_ID2473_for_PyTorch
- Keyword-MLP_ID2441_for_PyTorch
- LADE_ID2445_for_PyTorch
- MaskRCNN_ID0101_for_PyTorch
- mBART_ID1550_for_PyTorch
- MMAL-NET_ID1116_for_PyTorch
- MMOE_ID2865_for_PyTorch
- Mnasnet0_75_ID0439_for_PyTorch
- Mnasnet1_0_ID0438_for_PyTorch
- Mnasnet1_3_ID0437_for_PyTorch
- mobilenetv2_for_Pytorch
- MobileNetV3-Small_ID1785_for_PyTorch
- MSPN_ID0960_for_PyTorch
- MTCNN_ID0435_for_PyTorch
- Mutual-Channel-Loss_ID1113_for_PyTorch
- NeuMF_ID0351_for_PyTorch
- PASSRnet_ID0986_for_PyTorch
- pFedMe_ID1597_for_PyTorch
- PFF_ID1128_for_PyTorch
- PiT_ID2671_for_PyTorch
- Pix2Pix_ID0331_for_PyTorch
- POOLNET_ID0875_for_PyTorch
- Pysot_ID0428_for_PyTorch
- RAFT_for_PyTorch
- RANet_ID0994_for_PyTorch
- RES2NET_ID0824_for_PyTorch
- residual_adapters_ID1598_for_PyTorch
- Resnet101_ID0425_for_PyTorch
- ResNet152_ID0424_for_PyTorch
- Resnet152_ID1592_for_PyTorch
- Resnet18_ID0423_for_PyTorch
- Resnet34_ID0422_for_PyTorch
- Resnext101_32x8d_ID0420_for_PyTorch
- ResNeXt101_ID1717_for_PyTorch
- ResNeXt50_ID0419_for_PyTorch
- RRN_ID1182_for_PyTorch
- RUC_ID2470_for_PyTorch
- SEResNext_ID0415_for_PyTorch
- SG2IM_ID0786_for_PyTorch
- SimplePose_ID1038_for_PyTorch
- SINGLESHOTPOSE_ID0869_for_PyTorch
- SlowFast_ID0646_for_PyTorch
- SmartSketch_ID1046_for_PyTorch
- SqueezeNet_ID0413_for_PyTorch
- SRCNN_ID1770_for_PyTorch
- SRGAN_ID2956_for_PyTorch
- SSD-MobileNet_ID1936_for_PyTorch
- STARGAN_ID0725_for_PyTorch
- Swin-Transformer_ID2377_for_PyTorch
- TabNet_ID2862_for_PyTorch
- Token-to-Token-ViT_ID2668_for_PyTorch
- TransformerXL_ID0699_for_PyTorch
- VAE+GAN_ID0401_for_PyTorch
- VGG16_ID0467_for_PyTorch
- VGG19_ID0244_for_PyTorch
- VGGNet_for_Pytorch
- VGGNet_ID0400_for_PyTorch
- vit-base_ID0492_for_PyTorch
- VIT_ID2381_for_PyTorch
- Wide_resnet101_2_ID0398_for_PyTorch
- Wide_resnet50_2_ID0397_for_PyTorch
- Xception_ID1454_for_PyTorch
- ZERO-DCE_ID1040_for_PyTorch
- deeplabv3+_ID0326_for_PyTorch
- DeepLabV3+_ID0458_for_PyTorch
- DeepLabV3_ID0621_for_PyTorch
- SETR_ID1572_for_PyTorch
- GAN_ID1931_for_PyTorch
- DSCNN_for_PyTorch
- FFA-NET_ID1043_for_PyTorch
- BERT-ITPT-FiT_ID0340_for_PyTorch
- BERT_base_for_PyTorch
- FairSeq_Transformer_ID0496_for_PyTorch
- Retinanet_for_PyTorch
- Speech_Transformer_ID0487_for_PyTorch
- Swin-Transformer_ID2375_for_PyTorch
- Swin-Transformer_ID2379_for_PyTorch
- Textcnn_for_PyTorch
- Transformer_ID0105_for_PyTorch
- Multi-Gradient_Descent_ID0349_for_PyTorch
- Widedeep_ID2866_for_PyTorch
- EspNet_for_Pytoch
- Jasper_for_PyTorch
- LSTM
- RawNet2_for_Pytorch
- Tacotron2_for_Pytorch
- TDNN_for_Pytorch
- Wenet_for_Pytorch
- 3DUnet_for_PTorch
- CascadeRCNN-DCN
- CascadeRCNN-DCN-101_for_Pytorch
- CenterNet_for_Pytorch
- CRNN_for_Pytorch
- DB_for_PyTorch
- Deepmar_for_Pytorch
- Deepsort_for_Pytorch
- Densenet121_Pytorch_Infer
- DPN131_for_Pytorch
- EfficientNet_for_Pytorch
- FasterRCNN-DCN
- Flownet2_for_Pytorch
- GoogleNet_for_Pytorch
- I3D_for_Pytorch
- InceptionV3_for_Pytorch
- InceptionV4_for_Pytorch
- MobileNetV2_for_Pytorch
- Pelee_for_Pytorch
- PSENet_for_Pytorch
- Res2Net_v1b_101_for_PyTorch
- Resnet101_Pytorch_Infer
- Resnet18_for_PyTorch
- Resnet34_for_Pytorch
- Resnet50_Pytorch_Infer
- ResNeXt50_for_Pytorch
- SE_ResNet50_Pytorch_Infer
- Shufflenetv2_for_Pytorch
- STGCN_for_Pytorch
- U2-Net_for_PyTorch
- Vgg16_for_Pytorch
- VGG16_SSD_for_PyTorch
- Vgg19_for_Pytorch
- Wide_ResNet50_2_for_Pytorch
- YolactEdge_for_PyTorch
- Yolov3_for_Pytorch
- Yolov4_for_Pytorch
- Yolov5_for_Pytorch
- Bert_Base_Uncased
- CNN_Transformer_for_Pytorch
- textcnn
- TransformerXL_for_Pytorch
- VilBert_for_Pytorch
- FastPitch
- Jasper
- Speech-Transformer
- tdnn
- 3d_attention_net
- AlexNet
- baseline_dino_resnet50
- BMN
- C3D
- CSPResneXt50
- Deit_Small
- DPN131
- Efficient-3DCNNs
- EfficientNet-B1
- EfficientNet-B3
- EfficientNet-B5
- FixRes
- GaitSet
- GENet
- GhostNet1.0x
- GloRe
- HRNet-Image-Classification
- InceptionResnetV2
- LResNet100E-IR
- LV-Vit
- Mnasnet1_0
- MobileNet-v1
- MobileNetV3_large_100
- Moco-v2
- OSNet
- PAMTRI
- pnasnet5large
- PointNet
- PointNetCNN
- Pointnetplus
- R(2+1)D
- RegNetX-1.6GF
- RegNetY-1.6GF
- ReID_for_Pytorch
- RepVGG
- Res2Net101_v1b
- ResNeSt50
- ResNet101
- ResNet152
- ResNet18
- ResNet34
- ResNext101_32x8d
- Se-Resnext101
- SENet
- Shufflenetv1
- Shufflenetv2+
- SimCLR_inference
- Sknet50
- spnasnet_100
- Squeezenet1_1
- Swin-Transformer
- TNT
- TResNet
- vit-small
- vit_base_patch32_224
- vovnet39
- Wide_ResNet101_2
- X3D
- xception
- xcit
- 3DUnet
- AdvancedEAST
- AlphaPose
- BSN
- Cascade-RCNN-Resnet101-FPN-DCN
- Cascade_RCNN_R101
- CenterFace
- CenterNet
- CTPN
- Deepspeech
- Detr
- DSFD
- EfficientDetD0
- EfficientDetD7
- FairMOT
- FasterRCNN_FPN_DCN
- Fcos
- FOTS
- Fsaf
- GFocalV2
- M2Det
- Nasnetlarge
- NAS_FPN
- OpenPose
- pyramidbox
- RCF
- RefineDet
- Retinanet
- RFCN
- SSD
- SSD-MobileNetV1
- SSD-MobileNetV2
- SSD-Resnet34
- TextSnake
- yolor
- YOLOX
- AlignedReID
- FaceBoxes
- FaceNet
- ReId-MGN-master
- reid_PCB_baseline
- Retinaface
- BigGAN
- CGAN
- CycleGAN
- DCGAN
- GAN
- Pix2Pix
- Pix2pixHD
- StarGAN
- Cross-Scale-Non-Local-Attention
- DnCNN
- SRFlow
- wdsr
- 3DMPPE-ROOTNET
- HigherHRNet
- MSPN
- TransPose
- VideoPose3D
- ADNet
- SRGAN
- 3D_HRNet
- Cascade_Mask_RCNN
- Cascade_RCNN
- DeeplabV3
- ENet
- ErfNet
- FastSCNN
- FCN-8s
- GCNet
- ICNet
- IntraDA
- Nested_UNet
- PointRend
- PraNet
- PSPnet
- RefineNet
- SETR
- SiamMask
- SOLOV1
- SOLOV2
- Ultra-Fast-Lane-Detection
- VNet
- Wseg
- YOLACT
- YOLACT_plus
- EDSR
- RCAN
- RDN
- Real-ESRGAN
- SRCNN
- SiamFC
- SiamRPN
- I3D
- NonLocal
- SlowFast
- TSM
- TSN
- RotatE
- albert
- BertSum
- CNN_Transformer_for_Pytorch
- roberta
- TextCNN
- tinybert
- c51
- DQN
Ascend ModelZoo provides only scripts for downloading and preprocessing public datasets. We do not own these datasets and are not responsible for their quality or maintenance. Please ensure that you have the permission to use a dataset under the dataset's license. Models trained on these datasets are used only for non-commercial research and educational purposes.
To dataset owners: If you do not want your dataset to be included on ModelZoo, or wish to update it in any way, we will remove or update all public content upon your request. You can contact us with Github/Gitee issues. Your understanding and contribution to this community are greatly appreciated.
Ascend ModelZoo is Apache 2.0 licensed, see the LICENSE file.