Skip to content

Latest commit

 

History

History

official

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

What Is New

  • We've done code refactoring for classic SOTA models,modularized data processing, model definition&creation, training process and other common components with new lanched MindSpore CV/NLP/Audio/Yolo/OCR Series toolbox

  • Old models were implemented by original MindSpore API with some tricks for training process speedup

  • More information for model performance, please check benchmark.

Standard Models

Computer Vision

Image Classification(backbone)

model acc@1 mindcv recipe vanilla mindspore
vgg11 71.86 config
vgg13 72.87 config
vgg16 74.61 config link
vgg19 75.21 config link
resnet18 70.21 config link
resnet34 74.15 config link
resnet50 76.69 config link
resnet101 78.24 config link
resnet152 78.72 config link
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 link
inceptionv4 80.88 config link
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 link
shufflenet_v1_g3_x0_5 57.05 config
shufflenet_v1_g3_x1_5 67.77 config link
shufflenet_v2_x0_5 57.05 config
shufflenet_v2_x1_0 67.77 config link
shufflenet_v2_x1_5 57.05 config
shufflenet_v2_x2_0 67.77 config
xception 79.01 config link
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 link
efficientnet_b1 78.95 config link
efficientnet_b2 79.80 link
efficientnet_b3 80.50 link
efficientnet_v2 83.77 link
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 link
vit_l_16_224 76.34 config
vit_l_32_224 73.71 config
swintransformer_tiny 80.82 config link
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

Object Detection

yolo

model map mindyolo recipe vanilla mindspore
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 link
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 link
yolov3_darknet53 45.5 config link
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

Classic

model map mind_series recipe vanilla mindspore
ssd_vgg16 23.2 link
ssd_mobilenetv1 22.0 link
ssd_mobilenetv2 29.1 link
ssd_resnet50 34.3 link
fastrcnn 58 link
maskrcnn_mobilenetv1 coming soon link
maskrcnn_resnet50 coming soon link

Semantic Segmentation

model mind_series recipe vanilla mindspore
ocrnet link
deeplab v3 link
deeplab v3 plus link
unet link

OCR

model dataset fscore mindocr recipe vanilla mindspore
dbnet_mobilenetv3 icdar2015 77.28 config link
dbnet_resnet18 icdar2015 83.71 config link
dbnet_resnet50 icdar2015 84.99 config link
dbnet_resnet50 msra-td500 85.03 config
dbnet++_resnet50 icdar2015 86.60 config
psenet_resnet152 icdar2015 82.06 config link
east_resnet50 icdar2015 84.87 config link
svtr_tiny IC03,13,15,IIT,etc 89.02 config
crnn_vgg7 IC03,13,15,IIT,etc 82.03 config link
crnn_resnet34_vd IC03,13,15,IIT,etc 84.45 config
rare_resnet34_vd IC03,13,15,IIT,etc 85.19 config

Face

model dataset acc mindface recipe vanilla mindspore
arcface_mobilefacenet-0.45g MS1MV2 98.70 config
arcface_r50 MS1MV2 99.76 config
arcface_r100 MS1MV2 99.38 config link
arcface_vit_t MS1MV2 99.71 config
arcface_vit_s MS1MV2 99.76 config
arcface_vit_b MS1MV2 99.81 config
arcface_vit_l MS1MV2 99.75 config
retinaface_mobilenet_0.25 WiderFace 90.77/88.2/74.76 config link
retinaface_r50 WiderFace 95.07/93.61/84.84 config link

NLP

nlp

model mindformer recipe vanilla mindspore
bert_base config link
t5_small config
gpt2_small config
gpt2_13b config
gpt2_52b config
pangu_alpha config
glm_6b config
glm_6b_lora config
llama_7b config
llama_13b config
llama_65b config
llama_7b_lora config
bloom_560m config
bloom_7.1b config
bloom_65b config
bloom_176b config

Recommendation

model mind_series recipe vanilla mindspore
deepfm link
wide&deep link

GNN

model mind_series recipe vanilla mindspore
GCN link

Audio

model mind_series recipe vanilla mindspore
deepspeech2 link
ecapa_tdnn link
lpcnet link
melgan link
tacotron2 link

More models are coming

Disclaimers

Mindspore only provides scripts that downloads and preprocesses public datasets. We do not own these datasets and are not responsible for their quality or maintenance. Please make sure you have permission to use the dataset under the dataset’s license. The models trained on these dataset are for non-commercial research and educational purpose only.

To dataset owners: we will remove or update all public content upon request if you don’t want your dataset included on Mindspore, or wish to update it in any way. Please contact us through a Github/Gitee issue. Your understanding and contribution to this community is greatly appreciated.

MindSpore is Apache 2.0 licensed. Please see the LICENSE file.

License

Apache License 2.0