Official Models

View Source On Gitee

Domain Suite and Extension Packages

See MindSpore Transformers documentation for details on large language model libraries.

Image Classification (backbone)

model

acc@1

supported by mindcv

vgg11

71.86

config

vgg13

72.87

config

vgg16

74.61

config

vgg19

75.21

config

resnet18

70.21

config

resnet34

74.15

config

resnet50

76.69

config

resnet101

78.24

config

resnet152

78.72

config

resnetv2_50

76.90

config

resnetv2_101

78.48

config

dpn92

79.46

config

dpn98

79.94

config

dpn107

80.05

config

dpn131

80.07

config

densenet121

75.64

config

densenet161

79.09

config

densenet169

77.26

config

densenet201

78.14

config

seresnet18

71.81

config

seresnet34

75.36

config

seresnet50

78.31

config

seresnext26

77.18

config

seresnext50

78.71

config

skresnet18

73.09

config

skresnet34

76.71

config

skresnet50_32x4d

79.08

config

resnext50_32x4d

78.53

config

resnext101_32x4d

79.83

config

resnext101_64x4d

80.30

config

resnext152_64x4d

80.52

config

rexnet_x09

77.07

config

rexnet_x10

77.38

config

rexnet_x13

79.06

config

rexnet_x15

79.94

config

rexnet_x20

80.64

config

resnest50

80.81

config

resnest101

82.50

config

res2net50

79.35

config

res2net101

79.56

config

res2net50_v1b

80.32

config

res2net101_v1b

95.41

config

googlenet

72.68

config

inceptionv3

79.11

config

inceptionv4

80.88

config

mobilenet_v1_025

53.87

config

mobilenet_v1_050

65.94

config

mobilenet_v1_075

70.44

config

mobilenet_v1_100

72.95

config

mobilenet_v2_075

69.98

config

mobilenet_v2_100

72.27

config

mobilenet_v2_140

75.56

config

mobilenet_v3_small

68.10

config

mobilenet_v3_large

75.23

config

shufflenet_v1_g3_x0_5

57.05

config

shufflenet_v1_g3_x1_5

67.77

config

shufflenet_v2_x0_5

57.05

config

shufflenet_v2_x1_0

67.77

config

shufflenet_v2_x1_5

57.05

config

shufflenet_v2_x2_0

67.77

config

xception

79.01

config

ghostnet_50

66.03

config

ghostnet_100

73.78

config

ghostnet_130

75.50

config

nasnet_a_4x1056

73.65

config

mnasnet_0.5

68.07

config

mnasnet_0.75

71.81

config

mnasnet_1.0

74.28

config

mnasnet_1.4

76.01

config

efficientnet_b0

76.89

config

efficientnet_b1

78.95

config

regnet_x_200mf

68.74

config

regnet_x_400mf

73.16

config

regnet_x_600mf

73.34

config

regnet_x_800mf

76.04

config

regnet_y_200mf

70.30

config

regnet_y_400mf

73.91

config

regnet_y_600mf

75.69

config

regnet_y_800mf

76.52

config

mixnet_s

75.52

config

mixnet_m

76.64

config

mixnet_l

78.73

config

hrnet_w32

80.64

config

hrnet_w48

81.19

config

bit_resnet50

76.81

config

bit_resnet50x3

80.63

config

bit_resnet101

77.93

config

repvgg_a0

72.19

config

repvgg_a1

74.19

config

repvgg_a2

76.63

config

repvgg_b0

74.99

config

repvgg_b1

78.81

config

repvgg_b2

79.29

config

repvgg_b3

80.46

config

repvgg_b1g2

78.03

config

repvgg_b1g4

77.64

config

repvgg_b2g4

78.80

config

repmlp_t224

76.71

config

convnext_tiny

81.91

config

convnext_small

83.40

config

convnext_base

83.32

config

vit_b_32_224

75.86

config

vit_l_16_224

76.34

config

vit_l_32_224

73.71

config

swintransformer_tiny

80.82

config

pvt_tiny

74.81

config

pvt_small

79.66

config

pvt_medium

81.82

config

pvt_large

81.75

config

pvt_v2_b0

71.50

config

pvt_v2_b1

78.91

config

pvt_v2_b2

81.99

config

pvt_v2_b3

82.84

config

pvt_v2_b4

83.14

config

pit_ti

72.96

config

pit_xs

78.41

config

pit_s

80.56

config

pit_b

81.87

config

coat_lite_tiny

77.35

config

coat_lite_mini

78.51

config

coat_tiny

79.67

config

convit_tiny

73.66

config

convit_tiny_plus

77.00

config

convit_small

81.63

config

convit_small_plus

81.80

config

convit_base

82.10

config

convit_base_plus

81.96

config

crossvit_9

73.56

config

crossvit_15

81.08

config

crossvit_18

81.93

config

mobilevit_xx_small

68.90

config

mobilevit_x_small

74.98

config

mobilevit_small

78.48

config

visformer_tiny

78.28

config

visformer_tiny_v2

78.82

config

visformer_small

81.76

config

visformer_small_v2

82.17

config

edgenext_xx_small

71.02

config

edgenext_x_small

75.14

config

edgenext_small

79.15

config

edgenext_base

82.24

config

poolformer_s12

77.33

config

xcit_tiny_12_p16

77.67

config

OCR

Text Detection

model

dataset

fscore

supported by mindocr

dbnet_mobilenetv3

icdar2015

77.23

config

dbnet_resnet18

icdar2015

81.73

config

dbnet_resnet50

icdar2015

85.05

config

dbnet++_resnet50

icdar2015

86.74

config

psenet_resnet152

icdar2015

82.06

config

east_resnet50

icdar2015

84.87

config

fcenet_resnet50

icdar2015

84.12

config

Text Recognition

model

dataset

acc

supported by mindocr

svtr_tiny

IC03,13,15,IIIT,etc

89.02

config

crnn_vgg7

IC03,13,15,IIIT,etc

82.03

config

crnn_resnet34_vd

IC03,13,15,IIIT,etc

84.45

config

rare_resnet34_vd

IC03,13,15,IIIT,etc

85.19

config

Direction Classification

model

dataset

acc

supported by mindocr

mobilenetv3

RCTW17,MTWI,LSVT

94.59

config

Object Detection

YOLO

model

map

supported by mindyolo

yolov8_n

37.2

config

yolov8_s

44.6

config

yolov8_m

50.5

config

yolov8_l

52.8

config

yolov8_x

53.7

config

yolov7_t

37.5

config

yolov7_l

50.8

config

yolov7_x

52.4

config

yolov5_n

27.3

config

yolov5_s

37.6

config

yolov5_m

44.9

config

yolov5_l

48.5

config

yolov5_x

50.5

config

yolov4_csp

45.4

config

yolov4_csp(silu)

45.8

config

yolov3_darknet53

45.5

config

yolox_n

24.1

config

yolox_t

33.3

config

yolox_s

40.7

config

yolox_m

46.7

config

yolox_l

49.2

config

yolox_x

51.6

config

yolox_darknet53

47.7

config

Reinforcement Learning

Algorithm

Discrete Action Space

Continuous Action Space

CPU

GPU

Ascend

Environment

Reward

DQN

/

CartPole-v0

195

PPO

/

HalfCheetah-v2

4800

AC

/

CartPole-v0

195

A2C

/

CartPole-v0

195

DDPG

/

HalfCheetah-v2

4800

QMIX

/

SMAC/Simple Spread

90%/-145

SAC

/

HalfCheetah-v2

4800

TD3

/

HalfCheetah-v2

4800

C51

/

CartPole-v0

195

A3C

/

/

CartPole-v0

195

CQL

/

Hopper-v0

3500

MAPPO

/

Simple Spread

-145

GAIL

/

HalfCheetah-v2

4800

AWAC

/

Ant-v2

5000

Dreamer

/

/

Walker-walk

900

IQL

/

Walker2d-v2

3000

MADDPG

/

simple_spread

-140

Double DQN

/

CartPole-v0

195

Policy Gradient

/

CartPole-v0

195

Dueling DQN

/

CartPole-v0

195

Scientific Computing

Fields

Network

MindSpore Implementation

Ascend

GPU

General Physics

auq_pinns

Link

General Physics

cpinns

Link

General Physics

deep_hpms

Link

General Physics

deep_ritz

Link

General Physics

deepbsde

Link

General Physics

deeponet

Link

General Physics

dgm

Link

General Physics

fbsnns

Link

General Physics

fpinns

Link

General Physics

gradient_pathologies_pinns

Link

General Physics

hp_vpinns

Link

General Physics

laaf

Link

General Physics

mgnet

Link

General Physics

multiscale_pinns

Link

General Physics

pfnn

Link

General Physics

phygeonet

Link

General Physics

pi_deeponet

Link

General Physics

pinns

Link

General Physics

pinns_ntk

Link

General Physics

ppinns

Link

General Physics

xpinns

Link

Hamiltonian Systems

sympnets

Link

Elastodynamics

pinn_elastodynamics

Link

Thermodynamics

pinn_heattransfer

Link

Meteorology

enso

Link

Geology

inversion_net

Link

Geology

pinn_helmholtz

Link

Marine Physics

ocean_model

Link

Marine Physics

pinns_swe

Link

Electromagnetic

meta_auto_decoder

Link

Electromagnetic

pinn_fwi

Link

Electromagnetic

time_domain_maxwell

Link

Electromagnetic

frequency_domain_maxwell

Link

Electromagnetic

AD_FDTD

Link

Electromagnetic

SED_ANN

Link

Electromagnetic

Metasurface_holograms

Link

Electromagnetic

maxwell_net

Link

Computational Biology

MEGA-Fold

Link

Computational Biology

MEGA-EvoGen

Link

Computational Biology

MEGA-Assessment

Link

Computational Biology

ColabDesign

Link

Computational Biology

DeepFRI

Link

Computational Biology

FAAST

Link

Computational Biology

JT-VAE

Link

Computational Biology

MG-BERT

Link

Computational Biology

Multimer

Link

Computational Biology

ProteinMPNN

Link

Computational Biology

UFold

Link

Computational Biology

ESM-IF1

Link

Computational Biology

ESM2

Link

Computational Biology

Grover

Link

Computational Biology

Pafnucy

Link

Computational Fluid Dynamics

FNO1D

Link

Computational Fluid Dynamics

KNO1D

Link

Computational Fluid Dynamics

FNO2D

Link

Computational Fluid Dynamics

KNO2D

Link

Computational Fluid Dynamics

FNO3D

Link

Computational Fluid Dynamics

CAE-LSTM

Link

Computational Fluid Dynamics

eHDNN

Link

Computational Fluid Dynamics

HDNN

Link

Computational Fluid Dynamics

ViT

Link

Computational Fluid Dynamics

PeRCNN

Link

Computational Fluid Dynamics

Burgers1D

Link

Computational Fluid Dynamics

Cylinder Flow

Link

Computational Fluid Dynamics

PDE-Net

Link

Computational Fluid Dynamics

hfm

Link

Computational Fluid Dynamics

label_free_dnn_surrogate

Link

Computational Fluid Dynamics

nsf_nets

Link