Skip to content

vastai modelzoo 是瀚博半导体维护的AI模型库,提供了人工智能多个领域(CV、AUDIO、NLP、LLM、MLLM等)的开源模型在瀚博训推芯片上的部署、训练示例。

License

Notifications You must be signed in to change notification settings

Vastai/VastModelZOO

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

logo License company


VastModelZOO是瀚博半导体维护的AI模型平台,提供了人工智能多个领域(CV、AUDIO、NLP、LLM、MLLM等)的开源模型在瀚博训推芯片上的部署、训练示例。

VastModelZOO旨在基于瀚博半导体的硬件产品和软件SDK,展示最佳的编程实践,以达成模型的快速移植和最优性能。

为方便大家使用VastModelZOO,我们将持续增加典型模型和基础插件。

依赖软件

基于瀚博半导体的硬件产品使用VastModelZOO前,需联系销售代表获取部署软件包。

模型列表

CV Models
  • classification
model codebase model list model type runtime
CSPNet timm
model name
  • cspresnet50
  • cspresnext50
  • cspdarknet53
classification Build_In
CSPNet mmcls
model name
  • cspresnet50
  • cspresnext50
  • cspdarknet53
classification Build_In
CSPNet ppcls
model name
  • cspdarknet53
classification Build_In
DenseNet timm
model name
  • densenet121
  • densenet161
  • densenet169
  • densenet201
  • densenetblur121d
classification Build_In
DenseNet torchvision
model name
  • densenet121
  • densenet161
  • densenet169
  • densenet201
classification Build_In
DenseNet mmcls
model name
  • densenet121
  • densenet161
  • densenet169
  • densenet201
classification Build_In
DenseNet keras
model name
  • densenet121
  • densenet169
  • densenet201
classification Build_In
DenseNet ppcls
model name
  • densenet121
  • densenet161
  • densenet169
  • densenet201
  • densenet264
classification Build_In
DLANet timm
model name
  • dla34
  • dla46_c
  • dla46x_c
  • dla60
  • dla60x
  • dla60x_c
  • dla102
  • dla102x
  • dla102x2
  • dla169
  • dla60_res2net
  • dla60_res2next
classification Build_In
DLANet ppcls
model name
  • dla34
  • dla46_c
  • dla60
  • dla60x
  • dla60x_c
  • dla102
  • dla102x
  • dla102x2
  • dla169
classification Build_In
DLANet ucbdrive
model name
  • dla34
  • dla46_c
  • dla46x_c
  • dla60
  • dla60x
  • dla60x_c
  • dla102
  • dla102x
  • dla102x2
  • dla169
classification Build_In
EfficientNet timm
model name
  • efficientnet_b0
  • efficientnet_b1
  • efficientnet_b2
  • efficientnet_b3
  • efficientnet_b4
  • tf_efficientnet_b5
  • tf_efficientnet_b6
  • tf_efficientnet_b7
  • tf_efficientnet_b8
classification Build_In
EfficientNet_v2 timm
model name
  • efficientnetv2_rw_t
  • efficientnetv2_rw_s
  • efficientnetv2_rw_m
classification Build_In
ESNet ppcls
model name
  • ESNet_x0_25
  • ESNet_x0_5
  • ESNet_x0_75
  • ESNet_x1_0
classification Build_In
GhostNet timm
model name
  • ghostnet_100
classification Build_In
GhostNet ppcls
model name
  • ghostnet_x0_5
  • ghostnet_x1_0
  • ghostnet_x1_3
  • ghostnet_x1_3_ssld
classification Build_In
HRNet official
model name
  • HRNet_w18_small_model_v1
  • HRNet_w18_small_model_v2
  • HRNet_w18
  • HRNet_w30
  • HRNet_w32
  • HRNet_w40
  • HRNet_w44
  • HRNet_w48
  • HRNet_w64
classification Build_In
HRNet mmcls
model name
  • HRNet_w18
  • HRNet_w30
  • HRNet_w32
  • HRNet_w40
  • HRNet_w44
  • HRNet_w48
  • HRNet_w64
classification Build_In
HRNet timm
model name
  • HRNet_w18_small_model_v1
  • HRNet_w18_small_model_v2
  • HRNet_w18
  • HRNet_w30
  • HRNet_w32
  • HRNet_w40
  • HRNet_w44
  • HRNet_w48
  • HRNet_w64
classification Build_In
HRNet ppcls
model name
  • HRNet_W18_C
  • HRNet_W18_C_ssld
  • HRNet_W30_C
  • HRNet_W32_C
  • HRNet_W40_C
  • HRNet_W44_C
  • HRNet_W48_C
  • HRNet_W48_C_ssld
  • HRNet_W64_C
  • SE_HRNet_W64_C_ssld
classification Build_In
Inception_v3 timm
model name
  • inception_v3
  • tf_inception_v3
  • adv_inception_v3
  • gluon_inception_v3
classification Build_In
Inception_v3 torchvision
model name
  • inception_v3
classification Build_In
Inception_v3 ppcls
model name
  • inception_v3
classification Build_In
Inception_v4 timm
model name
  • inception v4
classification Build_In
Inception_v4 ppcls
model name
  • inception v4
classification Build_In
IResNet official
model name
  • iresnet50
  • iresnet101
  • iresnet152
  • iresnet200
classification Build_In
MNasNet torchvision
model name
  • mnasnet0_5
  • mnasnet1_0
classification Build_In
MobileNet_v1 keras
model name
  • mobilenetv1
classification Build_In
MobileNet_v1 ppcls
model name
  • mobilenet_v1_x0.25
  • mobilenet_v1_x0.5
  • mobilenet_v1_x0.75
  • mobilenet_v1
  • mobilenet_v1_ssld
classification Build_In
MobileNet_v2 torchvision
model name
  • mobilenetv2
classification Build_In
MobileNet_v2 timm
model name
  • mobilenetv2
classification Build_In
MobileNet_v2 mmcls
model name
  • mobilenetv2
classification Build_In
MobileNet_v2 keras
model name
  • mobilenetv2
classification Build_In
MobileNet_v2 ppcls
model name
  • mobilenet_v2_x0.25
  • mobilenet_v2_x0.5
  • mobilenet_v2_x0.75
  • mobilenet_v2
  • mobilenet_v2_x1.5
  • mobilenet_v2_x2.0
  • mobilenet_v2_ssld
classification Build_In
MobileNet_v3 torchvision
model name
  • mobilenet_v3_small
  • mobilenet_v3_large
classification Build_In
MobileNet_v3 timm
model name
  • mobilenet_v3_x1.0
  • mobilenet_v3_small_x0.5
  • mobilenet_v3_small_x0.75
  • mobilenet_v3_small_x1.0
  • mobilenet_v3_large_x1.0
classification Build_In
MobileNet_v3 ppcls
model name
  • mobilenet_v3_small_x0.35
  • mobilenet_v3_small_x0.35_ssld
  • mobilenet_v3_small_x0.5
  • mobilenet_v3_small_x0.75
  • mobilenet_v3_small_x1.0
  • mobilenet_v3_small_x1.0_ssld
  • mobilenet_v3_small_x1.25
  • mobilenet_v3_large_x0.35
  • mobilenet_v3_large_x0.5
  • mobilenet_v3_large_x0.75
  • mobilenet_v3_large_x1.0
  • mobilenet_v3_large_x1.0_ssld
  • mobilenet_v3_large_x1.25
classification Build_In
MobileNet_v3 showlo
model name
  • mobilenet_v3_small
classification Build_In
MobileNet_v3 sqlai
model name
  • mobilenet_v3_small
  • mobilenet_v3_large
classification Build_In
MobileOne official
model name
  • mobileone_s0
  • mobileone_s1
  • mobileone_s2
  • mobileone_s3
  • mobileone_s4
classification Build_In
MobileViT apple
model name
  • mobilevit-s
classification Build_In
PPLCNet ppcls
model name
  • PPLCNet_x0_25
  • PPLCNet_x0_35
  • PPLCNet_x0_5
  • PPLCNet_x0_75
  • PPLCNet_x1_0
  • PPLCNet_x1_5
  • PPLCNet_x2_0
  • PPLCNet_x2_5
  • PPLCNet_x0_5_ssld
  • PPLCNet_x1_0_ssld
  • PPLCNet_x2_5_ssld
classification Build_In
PPLCNet_v2 ppcls
model name
  • PPLCNetV2_base
  • PPLCNetV2_base_ssld
classification Build_In
RepOPT official
model name
  • RepOpt-VGG-B1
  • RepOpt-VGG-B2
  • RepOpt-VGG-L1
  • RepOpt-VGG-L2
classification Build_In
RepVGG official
model name
  • RepVGG-A0
  • RepVGG-A1
  • RepVGG-A2
  • RepVGG-B0
  • RepVGG-B1
  • RepVGG-B2
  • RepVGG-B1g2
  • RepVGG-B1g4
  • RepVGG-B2g4
  • RepVGG-B3
  • RepVGG-B3g4
classification Build_In
Res2Net ppcls
model name
  • res2net50_14w_8s
  • res2net50_26w_4s
  • res2net50_vd_26w_4s
  • res2net50_vd_26w_4s_ssld
  • res2net101_vd_26w_4s
  • res2net101_vd_26w_4s_ssld
  • res2net200_vd_26w_4s
  • res2net200_vd_26w_4s_ssld
classification Build_In
ResNeSt official
model name
  • resnest50
  • resnest101
  • resnest200
  • resnest269
classification Build_In
ResNet timm
model name
  • resnet18
  • resnet26
  • resnet34
  • resnet50
  • resnet101
  • resnet152
  • gluon_resnet18_v1b
  • gluon_resnet34_v1b
  • gluon_resnet50_v1b
  • gluon_resnet50_v1c
  • gluon_resnet50_v1d
  • gluon_resnet50_v1s
  • gluon_resnet101_v1b
  • gluon_resnet101_v1c
  • gluon_resnet101_v1d
  • gluon_resnet101_v1s
  • gluon_resnet152_v1b
  • gluon_resnet152_v1c
  • gluon_resnet152_v1d
  • gluon_resnet152_v1s
classification Build_In
ResNet torchvision
model name
  • resnet18
  • resnet34
  • resnet50
  • resnet101
  • resnet152
classification Build_In
ResNet mmcls
model name
  • resnet18
  • resnet34
  • resnet50
  • resnet101
  • resnet152
classification Build_In
ResNet ppcls
model name
  • resnet18
  • resnet18_vd
  • resnet34
  • resnet34_vd
  • resnet34_vd_ssld
  • resnet50
  • resnet50_vc
  • resnet50_vd
  • resnet50_vd_ssld
  • resnet101
  • resnet101_vd
  • resnet101_vd_ssld
  • resnet152
  • resnet152_vd
  • resnet200_vd
classification Build_In
ResNet keras
model name
  • resnet50
  • resnet50v2
  • resnet101
  • resnet101v2
  • resnet152
  • resnet152v2
classification Build_In
ResNet oneflow
model name
  • resnet18
  • resnet34
  • resnet50
  • resnet101
  • resnet152
classification Build_In
ResNeXt timm
model name
  • resnext50_32x4d
  • resnext50d_32x4d
  • resnext101_32x8d
  • resnext101_64x4d
  • tv_resnext50_32x4d
classification Build_In
ResNeXt torchvision
model name
  • resnext50_32x4d
  • resnext101_32x8d
classification Build_In
ResNeXt mmcls
model name
  • resnext50_32x4d
  • resnext101_32x4d
  • resnext101_32x8d
  • resnext152_32x4d
classification Build_In
ResNeXt ppcls
model name
  • resnext50_32x4d
  • resnext50_64x4d
  • resnext50_vd_32x4d
  • resnext50_vd_64x4d
  • resnext101_32x4d
  • resnext101_64x4d
  • resnext101_vd_32x4d
  • resnext101_vd_64x4d
  • resnext152_32x4d
  • resnext152_64x4d
  • resnext152_vd_32x4d
  • resnext152_vd_64x4d
  • resnext101_32x8d_wsl
  • resnext101_32x16d_wsl
  • resnext101_32x32d_wsl
classification Build_In
SENet timm
model name
  • seresnet50
  • seresnet152d
  • seresnext26d_32x4d
  • seresnext26t_32x4d
  • seresnext50_32x4d
  • seresnext101_32x8d
  • seresnext101d_32x8d
  • legacy_seresnet18
  • legacy_seresnet34
  • legacy_seresnet50
  • legacy_senet154
  • legacy_seresnext26_32x4d
  • legacy_seresnext50_32x4d
  • legacy_seresnext101_32x4d
  • legacy_seresnet101
  • legacy_seresnet152
classification Build_In
ShuffleNet_v1 mmcls
model name
  • shufflenet_v1
classification Build_In
ShuffleNet_v1 megvii
model name
  • Group3_0.5x
  • Group3_1.0x
  • Group3_1.5x
  • Group3_2.0x
  • Group8_0.5x
  • Group8_1.0x
  • Group8_1.5x
  • Group8_2.0x
classification Build_In
ShuffleNet_v2 torchvision
model name
  • shufflenet_v2_x0_5
  • shufflenet_v2_x1_0
classification Build_In
ShuffleNet_v2 mmcls
model name
  • shufflenet_v2_x1_0
classification Build_In
ShuffleNet_v2 ppcls
model name
  • shufflenet_v2_x0_25
  • shufflenet_v2_x0_33
  • shufflenet_v2_x0_5
  • shufflenet_v2_x1_0
  • shufflenet_v2_x1_5
  • shufflenet_v2_x2_0
  • shufflenet_v2_swish
classification Build_In
ShuffleNet_v2 megvii
model name
  • shufflenet_v2_x0.5
  • shufflenet_v2_x1.0
  • shufflenet_v2_x1.5
  • shufflenet_v2_x2.0
classification Build_In
Swin microsoft
model name
  • swin-b
classification Build_In
VGG timm
model name
  • vgg11
  • vgg11_bn
  • vgg13
  • vgg13_bn
  • vgg16
  • vgg16_bn
  • vgg19
  • vgg19_bn
classification Build_In
VGG torchvision
model name
  • vgg11
  • vgg11_bn
  • vgg13
  • vgg13_bn
  • vgg16
  • vgg16_bn
  • vgg19
  • vgg19_bn
classification Build_In
VGG mmcls
model name
  • vgg11
  • vgg11_bn
  • vgg13
  • vgg13_bn
  • vgg16
  • vgg16_bn
  • vgg19
  • vgg19_bn
classification Build_In
VGG ppcls
model name
  • vgg11
  • vgg13
  • vgg16
  • vgg19
classification Build_In
ViT huggingface
model name
  • vit-base-patch16-224
classification Build_In
VOVNet timm
model name
  • ese_vovnet19b_dw
  • ese_vovnet39b
classification Build_In
Wide ResNet timm
model name
  • wide_resnet50_2
  • wide_resnet101_2
classification Build_In
Wide ResNet torchvision
model name
  • wide_resnet50_2
  • wide_resnet101_2
classification Build_In
Xception ppcls
model name
  • xception41
  • xception41_deeplab
  • xception65
  • xception65_deeplab
  • xception71
classification Build_In
  • object detection
model codebase model list model type runtime
CenterNet official
model name
  • centernet_res18
object detection Build_In
DETR facebook
model name
  • detr_res50
object detection Build_In
FastestDet official
model name
  • fastestdet
object detection Build_In
Grounding-DINO official
model name
  • groundingdino_swint_ogc
  • groundingdino_swinb_cogcoor
object detection Build_In
NanoDet official
model name
  • nanodet_m
  • nanodet_plus_m
  • nanodet_plus_m_1.5x
object detection Build_In
RetinaNet pytorch
model name
  • retinanet_resnet50
object detection Build_In
RetinaNet mlcommons
model name
  • retinanet_resnext50_32x4d_fpn
object detection Build_In
RetinaNet mmdet
model name
  • retinanet_r18_fpn_1x_coco
  • retinanet_r18_fpn_1x8_1x_coco
  • retinanet_r50_fpn_1x_coco
  • retinanet_r50_fpn_2x_coco
  • retinanet_r50_fpn_fp16_1x_coco
  • retinanet_r50_caffe_fpn_1x_coco
  • retinanet_r101_fpn_1x_coco
  • retinanet_r101_fpn_2x_coco
  • retinanet_r101_caffe_fpn_1x_coco
  • retinanet_x101_32x4d_fpn_1x_coco
  • retinanet_x101_32x4d_fpn_2x_coco
  • retinanet_x101_64x4d_fpn_1x_coco
  • retinanet_x101_64x4d_fpn_2x_coco
  • retinanet_r50_fpn_mstrain_3x_coco
  • retinanet_r101_caffe_fpn_mstrain_3x_coco
  • retinanet_r101_fpn_mstrain_3x_coco
  • retinanet_x101_64x4d_fpn_mstrain_3x_coco
object detection Build_In
RT-DETR official
model name
  • rtdetr_r18vd
object detection Build_In
YOLO-World official
model name
  • yolo_world_v2_l
object detection Build_In
YOLOF mmdet
model name
  • yolof
object detection Build_In
YoloFastV2 official
model name
  • yolofastv2
object detection Build_In
YOLOR official
model name
  • yolor_d6
  • yolor_e6
  • yolor_w6
  • yolor_p6
object detection Build_In
Yolov3 pytorch(u)
model name
  • yolov3
  • yolov3-spp
  • yolov3-tiny
object detection Buid_In
Yolov4 darknet
model name
  • yolov4
  • yolov4_tiny
  • yolov4_csp
  • yolov4_csp_swish
  • yolov4_csp_x_swish
  • yolov4x_mish
object detection Buid_In
Yolov4 bubbliiiing
model name
  • yolov4
  • yolov4_tiny
object detection Buid_In
Yolov4 tianxiaomo
model name
  • yolov4
object detection Buid_In
Yolov5 pytorch(u)
model name
  • yolov5n
  • yolov5s
  • yolov5m
  • yolov5l
  • yolov5x
  • yolov5n6
  • yolov5s6
  • yolov5m6
  • yolov5l6
  • yolov5x6
object detection Build_In
Yolov5 mmyolo
model name
  • yolov5n
  • yolov5s
  • yolov5m
  • yolov5l
  • yolov5n6
  • yolov5s6
  • yolov5m6
  • yolov5l6
object detection Build_In
yolov6 yolov6
model name
  • YOLOv6-n
  • YOLOv6-tiny
  • YOLOv6-s
object detection Build_In
yolov7 yolov7
model name
  • YOLOv7
  • YOLOv7x
  • YOLOv7-w6
  • YOLOv7-e6
  • YOLOv7-d6
  • YOLOv7-e6e
object detection Build_In
yolov8 yolov8
model name
  • YOLOv8n
  • YOLOv8s
  • YOLOv8m
  • YOLOv8b
  • YOLOv8l
  • YOLOv8x
object detection Build_In
yolov10 yolov10
model name
  • YOLOv10-N
  • YOLOv10-S
  • YOLOv10-M
  • YOLOv10-B
  • YOLOv10-L
  • YOLOv10-X
object detection Build_In
Yolov11 official
model name
  • yolo11n
  • yolo11s
  • yolo11m
  • yolo11l
  • yolo11x
object detection Build_In
Yolov12 official
model name
  • yolov12n
  • yolov12s
  • yolov12m
  • yolov12l
  • yolov12x
object detection Build_In
Yolox official
model name
  • yolox_s
  • yolox_m
  • yolox_l
  • yolox_x
  • yolox_darknet
  • yolox_tiny
  • yolox_nano
object detection Build_In
Yolox mmyolo
model name
  • yolox_s
  • yolox_tiny
object detection Build_In
  • segmentation
model codebase model list model type runtime
BiSeNet pytorch
model name
  • bisenetv1
  • bisenetv2
segmentation Build_In
BiSeNet pytorch
model name
  • bisenet
  • bisenet_2class
face segmentation Build_In
Deeplab_v3 pytorch
model name
  • deeplabv3_resnet50
  • deeplabv3_resnet101
segmentation Build_In
Deeplab_v3_plus pytorch
model name
  • deeplabv3plus_resnet50
  • deeplabv3plus_resnet101
segmentation Build_In
FCN pytorch
model name
  • fcn8s
  • fcn16s
  • fcn32s
segmentation Build_In
FCN mmseg
model name
  • fcn_r50_d8_20k
  • fcn_r50_d8_40k
  • fcn_r101_d8_20k
  • fcn_r101_d8_40k
segmentation Build_In
HRNet_Seg pytorch
model name
  • HRNetV2-W48
  • HRNetV2-W18-Small-v1
  • HRNetV2-W18-Small-v2
segmentation Build_In
Human_Seg pytorch
model name
  • unet_resnet18
  • deeplabv3plus_resnet18
human segmentation Build_In
Mask2Former official
model name
  • maskformer2_R50
instance segmentation Build_In
MODNet official
model name
  • modnet
matting Build_In
ParseNet pytorch
model name
  • parsenet
face segmentation Build_In
R2U_Net pytorch
model name
  • r2u_net
segmentation Build_In
RefineNet pytorch
model name
  • refinenet_resnet101
  • refinenet_lw_resnet50
  • refinenet_lw_resnet101
  • refinenet_lw_resnet152
  • refinenet_lw_mobilenetv2
segmentation Build_In
STDC official
model name
  • stdc1
segmentation Build_In
Unet bubbliiiing
model name
  • unet_vgg16
  • unet_resnet50
segmentation Build_In
Unet milesial
model name
  • unet_scale0.5
  • unet_scale1.0
segmentation Build_In
Unet keras
model name
  • unet
segmentation Build_In
Unet3P pytorch
model name
  • unet3p
  • unet3p_deepsupervision
segmentation Build_In
UnetPP pytorch
model name
  • unetpp
segmentation Build_In
Yolov8-seg ultralytics
model name
  • yolov8n-seg
  • yolov8s-seg
  • yolov8m-seg
  • yolov8l-seg
  • yolov8x-seg
instance segmentation Build_In
  • face alignment
model codebase model list model type runtime
HIH pytorch
model name
  • hih_wflw_s
  • hih_wflw_b
face alignment Build_In
HRNet_Face_Alignment pytorch
model name
  • HR18_300W
  • HR18_AFLW
  • HR18_COFW
  • HR18_WFLW
face alignment Build_In
PFLD pytorch
model name
  • pfld
face alignment Build_In
PIPNet official
model name
  • pip_resnet18
  • pip_resnet101
  • pipnet_mobilenet_v2
  • pipnet_mobilenet_v3
face alignment Build_In
  • face detection
model codebase model list model type runtime
CenterFace onnx
model name
  • centerface
  • centerface_bnmerged
face detection Build_In
RetinaFace pytorch
model name
  • retinaface-resnet50
  • retinaface-mobilenet0.25
face detection Build_In
SCRFD insightface
model name
  • scrfd_500m
  • scrfd_500m_bnkps
  • scrfd_1g
  • scrfd_2.5g
  • scrfd_2.5g_bnkps
  • scrfd_10g
  • scrfd_10g_bnkps
  • scrfd_34g
face detection Build_In
YOLO5Face pytorch
model name
  • yolov5n0.5_face
  • yolov5n_face
  • yolov5s_face
  • yolov5m_face
  • yolov5l_face
face detection Build_In
  • facial quality
model codebase model list model type runtime
SDD-FIQA sdd_fiqa
model name
  • sdd_fiqa
face quality Build_In
  • face recognize
model codebase model list model type runtime
FaceNet tensorflow
model name
  • facenet_vggface2
  • facenet_casia_webface
face recognize Build_In
FaceNet pytorch
model name
  • facenet_vggface2
  • facenet_casia_webface
face recognize Build_In
  • facial attribute
model codebase model list model type runtime
FairFace official
model name
  • fairface_res34
face attribute Build_In
  • image colorization
model codebase model list model type runtime
GPEN official
model name
  • gpen
image colorization Build_In
  • image compress
model codebase model list model type runtime
Elic official
model name
  • elic
image compress Build_In
Mlicpp official
model name
  • mlicpp
image compress Build_In
  • image retrieval
model codebase model list model type runtime
DINOv2 official
model name
  • dinov2_vitl14_reg4
image_retrieval Build_In
  • low light image enhancement
model codebase model list model type runtime
PairLIE official
model name
  • pairlie
low light image enhancement Build_In
USUIR official
model name
  • uieb
  • ruie
low light image enhancement Build_In
  • mot
model codebase model list model type runtime
ByteTrack official
model name
  • ByteTrack_ablation
  • bytetrack_x_mot17
  • bytetrack_l_mot17
  • bytetrack_m_mot17
  • bytetrack_s_mot17
  • bytetrack_nano_mot17
  • bytetrack_tiny_mot17
  • bytetrack_x_mot20
mot Build_In
DeepSort pytorch
model name
  • fast reid
mot Build_In
FairMOT official
model name
  • fairmot_yolov5s
mot Build_In
JDE pytorch
model name
  • jde_1088x608_uncertainty
  • jde_864x480_uncertainty
  • jde_576x320_uncertainty
mot Build_In
OCSORT pytorch
model name
  • ocsort_mot17_ablation
  • ocsort_x_mot17
  • ocsort_x_mot20
  • ocsort_dance_model
mot Build_In
  • pose
model codebase model list model type runtime
OpenPose pytorch
model name
  • body model
  • hand model
pose Build_In
HRNet_Pose mmpose
model name
  • hrnet_pose
pose Build_In
Yolov8_Pose ultralytics
model name
  • yolov8n_pose
  • yolov8s_pose
  • yolov8m_pose
  • yolov8l_pose
  • yolov8x_pose
  • yolov8x_pose_p6
pose Build_In
  • reid
model codebase model list model type runtime
Fast_Reid official
model name
  • market_bot_R50
  • market_bot_S50
  • market_bot_R50_ibn
  • market_bot_R101_ibn
reid Build_In
  • salient object detection
model codebase model list model type runtime
BSANet official
model name
  • bsanet
salient object detection Build_In
BSANet salod
model name
  • bsanet
salient object detection Build_In
DHSNet salod
model name
  • dhsnet
salient object detection Build_In
EGNet salod
model name
  • egnet
salient object detection Build_In
F3Net official
model name
  • f3net
salient object detection Build_In
GCPANet salod
model name
  • gcpanet
salient object detection Build_In
ISNet official
model name
  • isnet
salient object detection Build_In
SCRN salod
model name
  • scrn
salient object detection Build_In
U2Net official
model name
  • u2net
  • u2net_human_seg
  • u2netp
salient object detection Build_In
  • super_resolution
model codebase model list model type runtime
Bicubic_PlusPlus official
model name
  • bicubic_plusplus_2x
  • bicubic_plusplus_3x
super resolution Build_In
BSRGAN official
model name
  • bsrnet_x4
  • bsrgan_x4
  • bsrgan_x2
super resolution Build_In
ECBSR official
model name
  • ecbsr_plain_2x
super resolution Build_In
ECBSR basicsr
model name
  • ecbsr_rgb_2x
super resolution Build_In
EDSR basicsr
model name
  • edsr_1x
  • edsr_m2x
super resolution Build_In
EDSR official
model name
  • edsr_x2
  • edsr_baseline_x2
  • edsr_baseline_x4
super resolution Build_In
ESRGAN official
model name
  • esrgan_x4
super resolution Build_In
FSRNet* pytorch
model name
  • fsrnet
super resolution Build_In
GPEN official
model name
  • gpen
super resolution Build_In
LESRCNN official
model name
  • lesrcnn_x2
  • lesrcnn_x4
super resolution Build_In
MSRN official
model name
  • msrn
super resolution Build_In
NCNet official
model name
  • ncnet
super resolution Build_In
PFSRNet official
model name
  • pfsrnet
face super resolution Build_In
RCAN official
model name
  • rcan
  • rcan2
super resolution Build_In
RCAN basicsr
model name
  • rcan
super resolution Build_In
Real_ESRGAN official
model name
  • real_esrgan_x4plus
  • real_esrnet_x4plus
  • real_esrgan_x4plus_anime_6b
super resolution Build_In
RealSR official
model name
  • DF2K
  • DF2K-JPEG
  • DPED
super resolution Build_In
SPARNet official
model name
  • sparnet_light_attn3d
super resolution Build_In
VDSR pytorch
model name
  • vdsr
super resolution Build_In
  • text detection
model codebase model list model type runtime
CT ppocr
model name
  • det_r18_vd_ct
text detection Build_In
DBNet ppocr
model name
  • dbnet_mobilenet_v3
  • dbnet_resnet50_vd
  • ch_PP_OCRv3_det
  • ch_PP_OCRv4_det
  • en_PP_OCRv3_det
text detection Build_In
DBNet official
model name
  • dbnet_resnet18
  • dbnet_resnet50
text detection Build_In
DBNet mmocr
model name
  • dbnet_resnet18_fpnc_1200e_icdar2015
  • dbnet_resnet18_fpnc_1200e_totaltext
  • dbnet_resnet18_fpnc_100k_synthtext
  • dbnet_resnet50_1200e_icdar2015
  • dbnet_resnet50_oclip_1200e_icdar2015
text detection Build_In
EAST ppocr
model name
  • det_r50_vd_east
  • det_mv3_east
text detection Build_In
FAST pytorch
model name
  • fast_base_ic17mlt_640
  • fast_small_ic17mlt_640
  • fast_tiny_ic17mlt_640
  • fast_tiny_tt_448_finetune_ic17mlt
  • fast_tiny_tt_512_finetune_ic17mlt
  • fast_tiny_tt_640_finetune_ic17mlt
  • fast_small_tt_512_finetune_ic17mlt
  • fast_small_tt_640_finetune_ic17mlt
  • fast_base_tt_512_finetune_ic17mlt
  • fast_base_tt_640_finetune_ic17mlt
  • fast_base_tt_800_finetune_ic17mlt
  • fast_tiny_ctw_512_finetune_ic17mlt
  • fast_small_ctw_512_finetune_ic17mlt
  • fast_base_ctw_512_finetune_ic17mlt
  • fast_base_ctw_640_finetune_ic17mlt
  • fast_tiny_ic15_736_finetune_ic17mlt
  • fast_small_ic15_736_finetune_ic17mlt
  • fast_base_ic15_736_finetune_ic17mlt
  • fast_base_ic15_896_finetune_ic17mlt
  • fast_base_ic15_1280_finetune_ic17mlt
  • fast_tiny_msra_512_finetune_ic17mlt
  • fast_tiny_msra_736_finetune_ic17mlt
  • fast_small_msra_736_finetune_ic17mlt
  • fast_base_msra_736_finetune_ic17mlt
text detection Build_In
FCENet mmocr
model name
  • fcenet_resnet50_fpn_1500e_icdar2015
  • fcenet_resnet50_oclip_fpn_1500e_icdar2015
  • fcenet_resnet50_fpn_1500e_totaltext
  • fcenet_resnet50_oclip_fpn_1500e_ctw1500
text detection Build_In
PAN pan_pp.pytorch
model name
  • pan_r18_ctw
  • pan_r18_ctw_finetune
  • pan_r18_ic15
  • pan_r18_ic15_finetune
  • pan_r18_tt
  • pan_r18_tt_finetune
  • pan_r18_msra
  • pan_r18_msra_finetune
text detection Build_In
PAN++ detection only pan_pp.pytorch
model name
  • panpp_r18_ic15_det
text detection Build_In
PSENet pytorch
model name
  • psenet_r50_ic15_736
  • psenet_r50_ic15_1024
  • psenet_r50_ic15_736_finetune
  • psenet_r50_ic15_1024_finetune
  • psenet_r50_ctw
  • psenet_r50_ctw_finetune
  • psenet_r50_tt
  • psenet_r50_tt_finetune
text detection Build_In
Text Snake mmocr
model name
  • textsnake_resnet50_fpn
  • textsnake_resnet50_oclip_fpn
text detection Build_In
  • text recognition
model codebase model list model type runtime
CNN_CTC pytorch
model name
  • resnet_fc
ocr recognize Build_In
CRNN ppocr
model name
  • resnet34_vd
text recognition Build_In
PPOCR_V4_REC ppocr
model name
  • ppocr_v4_rec
ocr recognize Build_In
NLP Models
  • information extraction
model codebase model list model type runtime
uie uie_pytorch
model name
  • uie-base
Information extraction Build_In
  • named entity recognition
model codebase model list model type runtime
BERT huggingface
model name
  • bert_base_zh_ner-256
named entity recognition Build_In
RoBERTa huggingface
model name
  • roberta_wwm_ext_base_zh-256
named entity recognition Build_In
ERNIE huggingface
model name
  • ernie1.0_base_zh
  • ernie2.0_base_en
  • ernie2.0_large_en
  • ernie3.0_base_zh
  • ernie3.0_medium_zh
  • ernie3.0_xbase_zh
named entity recognition Build_In
RoBERTa huggingface
model name
  • roberta_wwm_ext_base_zh
named entity recognition Build_In
XLMRoBERTa huggingface
model name
  • xlmroberta_base_zh_ner
named entity recognition Build_In
  • question answering
model codebase model list model type runtime
BERT tensorflow
model name
  • bert_base_en_qa-384
  • bert_large_en_qa-384
question answering Build_In
BERT huggingface
model name
  • bert_base_en_qa-384
  • bert_large_en_qa-384
question answering Build_In
ERNIE huggingface
model name
  • ernie1.0_base_zh
  • ernie2.0_base_en
  • ernie2.0_large_en
  • ernie3.0_base_zh
  • ernie3.0_medium_zh
  • ernie3.0_xbase_zh
question answering Build_In
  • sentence classification
model codebase model list model type runtime
BERT tensorflow
model name
  • bert_base_mrpc_cls-128
  • bert_large_mrpc_cls-128
sentence classification Build_In
BERT huggingface
model name
  • bert_base_mrpc_cls-128
  • bert_base_mrpc_cls-512
  • bert_large_mrpc_cls-128
  • bert_large_mrpc_cls-512
  • bert_base_imdb_cls-128
  • bert_base_sst2_cls-128
sentence classification Build_In
BERT modelscope
model name
  • nlp_structbert_sentiment-classification_chinese-base
sentence classification Build_In
Electra CIB
model name
  • electra_small-512
sentence classification Build_In
Electra huggingface
model name
  • electra_small_dc_mrpc-128
  • electra_small_gen_mrpc-128
  • electra_base_dc_mrpc-128
  • electra_base_gen_mrpc-128
  • electra_large_dc_mrpc-128
sentence classification Build_In
ERNIE huggingface
model name
  • ernie1.0_base_zh
  • ernie2.0_base_en
  • ernie2.0_large_en
  • ernie3.0_base_zh
  • ernie3.0_medium_zh
  • ernie3.0_xbase_zh
sentence classification Build_In
RoBERTa huggingface
model name
  • roberta_base_en_cls-128
sentence classification Build_In
XLMRoBERTa huggingface
model name
  • xlmroberta_base_en_cls
sentence classification Build_In
  • Text2Vec

  • 以下是Embedding模型

model codebase model list model type runtime
acge huggingface
model name
  • acge_text_embedding
Embedding model Build_In
bge huggingface
model name
  • bge-m3
  • bge-small-en-v1.5
  • bge-base-en-v1.5
  • bge-large-en-v1.5
Embedding model Build_In
bce huggingface
model name
  • bce-embedding-base_v1
Embedding model Build_In
e5 huggingface
model name
  • multilingual-e5-small
  • multilingual-e5-base
  • multilingual-e5-large
  • multilingual-e5-large-instruct
Embedding model Build_In
gte huggingface
model name
  • gte-small
  • gte-base
  • gte-large
Embedding model Build_In
m3e huggingface
model name
  • m3e-small
  • m3e-base
  • m3e-large
Embedding model Build_In
text2vec huggingface
model name
  • text2vec-base-chinese
  • text2vec-base-multilingual
  • text2vec-large-chinese
Embedding model Build_In
  • 以下是Reranker模型
model codebase model list model type runtime
bge huggingface
model name
  • bge-reranker-base
  • bge-reranker-large
  • bge-reranker-v2-m3
Reranker model Build_In
bce huggingface
model name
  • bce-reranker-base_v1
Reranker model Build_In
LLM Models
model codebase model list model type runtime
Aquila2 huggingface
model name
  • Aquila2-7B
  • Aquila2-34B
large language model Build_In
BaiChuan2 huggingface
model name
  • Baichuan2-7B-Base
  • Baichuan2-7B-Chat
  • Baichuan2-13B-Base
  • Baichuan2-13B-Chat
large language model Build_In
ChatGLM2 huggingface
model name
  • chatglm2-6b
large language model Build_In
ChatGLM3 huggingface
model name
  • chatglm3-6b-base
  • chatglm3-6b
large language model Build_In
ChatGLM4 huggingface
model name
  • glm-4-9b-hf
  • glm-4-9b-chat-hf
large language model Build_In
DeepSeek-R1-Distill huggingface
model name
  • DeepSeek-R1-Distill-Qwen-1.5B
  • DeepSeek-R1-Distill-Qwen-7B
  • DeepSeek-R1-Distill-Qwen-14B
  • DeepSeek-R1-Distill-Qwen-32B
  • DeepSeek-R1-Distill-Llama-8B
  • DeepSeek-R1-Distill-Llama-70B
large language model Build_In
Falcon huggingface
model name
  • falcon-7b
large language model Build_In
Gemma huggingface
model name
  • gemma-2b
  • gemma-2b-it
  • gemma-7b
  • gemma-7b-it
large language model Build_In
Gemma-1.1 huggingface
model name
  • gemma-1.1-2b-it
  • gemma-1.1-7b-it
large language model Build_In
Internlm2 huggingface
model name
  • internlm2-1_8b
  • internlm2-chat-1_8b
  • internlm2-7b
  • internlm2-chat-7b
  • internlm2-20b
  • internlm2-chat-20b
large language model Build_In
Internlm2_5 huggingface
model name
  • internlm2_5-1_8b
  • internlm2_5-1_8b-chat
  • internlm2_5-7b
  • internlm2_5-7b-chat
  • internlm2_5-20b
  • internlm2_5-20b-chat
large language model Build_In
Internlm3 huggingface
model name
  • internlm3-8b-instruct
large language model Build_In
LLaMA huggingface
model name
  • meta-llama-33b
large language model Build_In
LLaMA-2 huggingface
model name
  • Llama-2-7b-hf
  • Llama-2-7b-chat-hf
  • Llama-2-13b-hf
  • Llama-2-13b-chat-hf
  • Llama-2-70b-hf
  • Llama-2-70b-chat-hf
large language model Build_In
LLaMA-3 huggingface
model name
  • Meta-Llama-3-8B
  • Meta-Llama-3-8B-Instruct
  • Meta-Llama-3-70B
  • Meta-Llama-3-70B-Instruct
large language model Build_In
LLaMA-3.1 huggingface
model name
  • Meta-Llama-3.1-8B
  • Meta-Llama-3.1-8B-Instruct
  • Meta-Llama-3.1-70B
  • Meta-Llama-3.1-70B-Instruct
large language model Build_In
LLaMA-3.2 huggingface
model name
  • Llama-3.2-1B
  • Llama-3.2-1B-Instruct
  • Llama-3.2-3B
  • Llama-3.2-3B-Instruct
large language model Build_In
LLaMA-3.3 huggingface
model name
  • Llama-3.3-70B-Instruct
large language model Build_In
Mistral huggingface
model name
  • Mistral-7B-Instruct-v0.2
  • Ministral-8B-Instruct-2410-HF
large language model Build_In
Mobillama huggingface
model name
  • MobiLlama-05B
  • MobiLlama-05B-Chat
  • MobiLlama-1B
  • MobiLlama-1B-Chat
large language model Build_In
OpenBuddy huggingface
model name
  • openbuddy-mistral2-7b-v20.3-32k
  • openbuddy-qwen1.5-14b-v21.1-32k
  • openbuddy-deepseek-67b-v18.1-4k
large language model Build_In
Phi-3 huggingface
model name
  • Phi-3-mini-4k-instruct
large language model Build_In
Qwen1.5 huggingface
model name
  • Qwen1.5-0.5B
  • Qwen1.5-0.5B-Chat
  • Qwen1.5-1.8B
  • Qwen1.5-1.8B-Chat
  • Qwen1.5-4B
  • Qwen1.5-4B-Chat
  • Qwen1.5-7B
  • Qwen1.5-7B-Chat
  • Qwen1.5-14B
  • Qwen1.5-14B-Chat
  • Qwen1.5-32B
  • Qwen1.5-32B-Chat
  • Qwen1.5-72B
  • Qwen1.5-72B-Chat
  • Qwen1.5-110B-Chat
large language model Build_In
Qwen2 huggingface
model name
  • Qwen2-0.5B
  • Qwen2-0.5B-Instruct
  • Qwen2-1.5B
  • Qwen2-1.5B-Instruct
  • Qwen2-7B
  • Qwen2-7B-Instruct
  • Qwen2-72B
  • Qwen2-72B-Instruct
large language model Build_In
Qwen2.5 huggingface
model name
  • Qwen2.5-0.5B
  • Qwen2.5-0.5B-Instruct
  • Qwen2.5-1.5B
  • Qwen2.5-1.5B-Instruct
  • Qwen2.5-3B
  • Qwen2.5-3B-Instruct
  • Qwen2.5-7B
  • Qwen2.5-7B-Instruct
  • Qwen2.5-14B
  • Qwen2.5-14B-Instruct
  • Qwen2.5-32B
  • Qwen2.5-32B-Instruct
  • Qwen2.5-72B
  • Qwen2.5-72B-Instruct
large language model Build_In
Qwen3 huggingface
model name
  • Qwen3-0.6B
  • Qwen3-1.7B
  • Qwen3-4B
  • Qwen3-8B
  • Qwen3-4B-Instruct-2507
  • Qwen3-4B-Thinking-2507
large language model Build_In
QWQ huggingface
model name
  • QwQ-32B-Preview
  • QwQ-32B
large language model Build_In
Skywork-OR1 huggingface
model name
  • Skywork-OR1-Math-7B
  • Skywork-OR1-7B-Preview
  • Skywork-OR1-32B-Preview
large language model Build_In
Stablelm huggingface
model name
  • 12b
large language model Build_In
Tigerbot huggingface
model name
  • tigerbot-7b
  • tigerbot-13b
large language model Build_In
wizardlm huggingface
model name
  • WizardLM-2-7B
large language model Build_In
Yi huggingface
model name
  • yi-6b
  • yi-9b
  • yi-34b
large language model Build_In
model codebase model list model type runtime
Qwen2.5 huggingface
model name
  • Qwen2.5-7B-Instruct-GPTQ-Int4
  • Qwen2.5-14B-Instruct-GPTQ-Int4
  • Qwen2.5-32B-Instruct-GPTQ-Int4
large language model vLLM
Qwen3 huggingface
model name
  • Qwen3-30B-A3B-FP8
  • Qwen3-30B-A3B-Instruct-2507-FP8
  • Qwen3-30B-A3B-Thinking-2507-FP8
  • Qwen3-Coder-30B-A3B-Instruct-FP8
large language model vLLM
DeepSeek-V3 huggingface
model name
  • DeepSeek-V3-Base
  • DeepSeek-V3
  • DeepSeek-V3-0324
  • DeepSeek-V3.1
  • DeepSeek-V3.1-Base
  • DeepSeek-V3.1-Terminus
large language model vLLM
DeepSeek-R1 huggingface
model name
  • DeepSeek-R1
  • DeepSeek-R1-0528
large language model vLLM

以下模型为Domain-LLM专业领域模型(医疗、金融、DeepResearch等),一般基于General-LLM模型为基座+专业领域数据集微调+强化学习而来

model codebase model list model type runtime priority
Tongyi-DeepResearch huggingface
model name
  • Tongyi-DeepResearch-30B-A3B-PF8
LLM vLLM p1
VLM Models
model codebase model list model type runtime
CLIP openai
model name
  • clip
vision-language model Build_In
Logics_Parsing huggingface
model name
  • Logics-Parsing
vision-language model Build_In
OCRFlux huggingface
model name
  • OCRFlux-3B
vision-language model Build_In
olmOCR2 huggingface
model name
  • olmOCR-2-7B-1025
vision-language model Build_In
Qwen2-VL huggingface
model name
  • Qwen2-VL-2B-Instruct
  • Qwen2-VL-7B-Instruct
vision-language model Build_In
Qwen2.5-VL huggingface
model name
  • Qwen2.5-VL-3B-Instruct
  • Qwen2.5-VL-7B-Instruct
  • Qwen2.5-VL-32B-Instruct
vision-language model Build_In
RolmOCR huggingface
model name
  • RolmOCR
vision-language model Build_In
SigLip google
model name
  • siglip
vision-language model Build_In

免责声明

  • VastModelZOO提供的模型仅供您用于非商业目的,请参考原始模型来源许可证进行使用
  • VastModelZOO描述的数据集均为开源数据集,如您使用这些数据集,请参考原始数据集来源许可证进行使用
  • 如您不希望您的数据集或模型公布在VastModelZOO上,请您提交issue,我们将尽快处理

使用许可

  • VastModelZOO提供的模型,如原始模型有许可证描述,请以该许可证为准
  • VastModelZOO遵循Apache 2.0许可证许可

About

vastai modelzoo 是瀚博半导体维护的AI模型库,提供了人工智能多个领域(CV、AUDIO、NLP、LLM、MLLM等)的开源模型在瀚博训推芯片上的部署、训练示例。

Topics

Resources

License

Stars

Watchers

Forks

Packages

No packages published

Languages