Skip to main content

进迭时空 Model zoo

进迭时空 AI model zoo 收录了受官方加速支持的各类模型,仓库地址位于https://archive.spacemit.com/spacemit-ai/ModelZoo/

Model zoo列表

|--dataset
|--gguf
|--llm
|--classification.tar.gz
|--detection.tar.gz
|--nlp.tar.gz
|--segmentation.tar.gz

模型仓库中包括了图像分类,目标检测,自然语言处理,图像分割以及大模型等仓库,其中收录的量化模型使用进迭时空推理工具推理可以获得良好的加速性能。

模型列表

图像分类

Model NameFp32 top1/%Quant top1/%1线程推理耗时@1.6GHz/ms2线程@1.6GHz/ms4线程@1.6GHz/ms
resnet1869.6469.6438.631.1220.18
resnet5075.4676.0097.323774.3449.15
seresnet5079.2679.00113.707102.6369.31
resnet50-v1.575.974.9697.816674.3249.28
resnext5080.6479.28149.08123.0995.5
mobilenet_v171.6471.4232.5121.7913.47
mobilenet_v271.3271.6636.6124.9115.81
mobilenet_v3_large73.2673.2036.81540.91631.624
efficientnet_v1_b076.8874.781.561.638.85
efficientnet_v1_b176.776.1114.0187.0256.66
efficientnet_v2_s80.379.68149.06116.0677.92
mnasnet0_567.6867.538.8728.0923.21
mnasnet1_073.3872.9457.7640.81527.36
v100_gpu64_6ms74.5874.8259.339847.603131.5937
shufflenet_v2_x1_068.7268.639.499832.527327.4406
inception_v166.0465.1122.19786.118163.6081
inception_resnet_v280.1880.04309.361227.131152.768
inception_v376.7476.32125.45787.046756.5531

目标检测

Model NameFp32 mAP/%Quant mAP/%1线程推理耗时@1.6GHz/ms2线程推理耗时@1.6GHz/ms4线程推理耗时@1.6GHz/ms
yolov6_n_dyn(320x320)39.538.5325.55317.767309.926
yolov6_n(640x640)50.747.2420.043365.711335.315
yolov6_s_dyn(320x320)45.545.3344.415313.461286.881
yolov5_n_dyn(320x320)35.134.1130.924100.53276.286
yolov5_n(640x640)44.042.9490.434359.839259.232
yolov5_s_dyn(320x320)42.040.4180.013138.45999.3787
yolov3_mobilenetv2(320x320)33.732.4451.454430.331390.081
yolov3_darknet53_dyn(320x320)39.837.8595.954500.68369.697
yolox_s(640x640)50.749.9452.011344.289222.526

姿态估计

Model Name1线程推理耗时@1.6GHz/ms2线程推理耗时@1.6GHz/ms4线程推理耗时@1.6GHz/ms
handpose_x_resnet50115.99378.350853.2492
rtmdet-nano_hand76.809856.021340.642
rtmdet-nano76.966256.61441.2351
rtmpose-s_simcc116.41894.811663.014
rtmpose-t39.13330.365723.3327
rtmpose-s49.70239.33230.0369