Skip to content

Latest commit

 

History

History
193 lines (153 loc) · 42.2 KB

model_zoo.md

File metadata and controls

193 lines (153 loc) · 42.2 KB
Raw

🛠️ X-AnyLabeling Model Zoo

Classification

Name Description Configuration Size Link
pulc_person_attribute.onnx PersonAttribute-PULC pulc_person_attribute.yaml 6.59MB baidu | github
pulc_vehicle_attribute.onnx VehicleAttribute-PULC pulc_vehicle_attribute.yaml 6.55MB baidu | github
internimage_l_22kto1k_384.onnx InternImage-Large internimage_l_22kto1k_384.yaml 853.16MB baidu | github
yolov5s-cls.onnx YOLOv5-Cls-ImageNet yolov5s_cls.yaml 20.81MB baidu | github
yolov8s-cls.onnx YOLOv8-Cls-ImageNet yolov8s_cls.yaml 24.28MB baidu | github

Keypoint Detection

  • Facial Landmark Detection
Name Description Configuration Size Link
yolov6lite_l_face.onnx Facial Landmark Detection yolov6lite_l_face.yaml 4.16MB baidu | github
yolov6lite_m_face.onnx Facial Landmark Detection yolov6lite_m_face.yaml 3.00MB baidu | github
yolov6lite_s_face.onnx Facial Landmark Detection yolov6lite_s_face.yaml 2.10MB baidu | github
  • Pose Estimation
Name Description Configuration Size Link
yolov8n-pose.onnx YOLOv8-COCO yolov8n_pose.yaml 12.75MB baidu | github
yolov8x-pose-p6.onnx YOLOv8-COCO yolov8x_pose_p6.yaml 378.92MB baidu | github
dw-ll_ucoco_384.onnx DWPose(2D human whole-body pose estimation) yolox_l_dwpose_ucoco.yaml 128.17MB baidu | github
yolox_l.onnx YOLOX(2D human whole-body pose estimation) yolox_l_dwpose_ucoco.yaml 206.71MB baidu | github

Lane Detection

Name Description Configuration Size Link
clrnet_tusimple_r18.onnx CLRNet-Tusimple (CVPR2022) clrnet_tusimple_r18.yaml 59.04MB baidu | github

Multi-Object-Tracking

Name Description Configuration Size Link
yolov5m.onnx YOLOv5m-ByteTrack yolov5m_bytetrack.yaml 81.19MB baidu | github
yolov5m.onnx YOLOv5m-OCSort yolov5m_ocsort.yaml 81.19MB baidu | github

Object Detection

  • Horizontal Bounding Box
Name Description Configuration Size Link
rtdetr_r50vd_6x_coco.onnx RT-DETR-COCO rtdetr_r50.yaml 160.96MB baidu | github
yolo_nas_l.onnx YOLO-NAS-COCO yolo_nas_l.yaml 160.38MB baidu | github
yolo_nas_m.onnx YOLO-NAS-COCO yolo_nas_m.yaml 121.87MB baidu | github
yolo_nas_s.onnx YOLO-NAS-COCO yolo_nas_s.yaml 46.62MB baidu | github
yolov5x.onnx YOLOv5-COCO yolov5x.yaml 331.19MB baidu | github
yolov5l.onnx YOLOv5-COCO yolov5l.yaml 177.94MB baidu | github
yolov5m.onnx YOLOv5-COCO yolov5m.yaml 81.19MB baidu | github
yolov5s.onnx YOLOv5-COCO yolov5s.yaml 27.98MB baidu | github
yolov5n.onnx YOLOv5-COCO yolov5n.yaml 7.54MB baidu | github
yolov6x_mbla.onnx YOLOv6-COCO yolov6x_mbla.yaml 300.95MB baidu | github
yolov6l_mbla.onnx YOLOv6-COCO yolov6l_mbla.yaml 176.72MB baidu | github
yolov6m_mbla.onnx YOLOv6-COCO yolov6m_mbla.yaml 99.61MB baidu | github
yolov6s_mbla.onnx YOLOv6-COCO yolov6s_mbla.yaml 44.49MB baidu | github
yolov6s.onnx YOLOv6-COCO yolov6s.yaml 70.88MB baidu | github
yolov6s6.onnx YOLOv6-COCO yolov6s6.yaml 158.47MB baidu | github
yolov7.onnx YOLOv7-COCO yolov7.yaml 140.90MB baidu | github
yolov8x.onnx YOLOv8-COCO yolov8x.yaml 260.37MB baidu | github
yolov8l.onnx YOLOv8-COCO yolov8l.yaml 166.79MB baidu | github
yolov8m.onnx YOLOv8-COCO yolov8m.yaml 98.94MB baidu | github
yolov8s.onnx YOLOv8-COCO yolov8s.yaml 42.75MB baidu | github
yolov8n.onnx YOLOv8-COCO yolov8n.yaml 12.21MB baidu | github
yolov8s.onnx YOLOv8 with SAHI-COCO yolov8s_sahi.yaml 42.75MB baidu | github
damoyolo_tinynasL20_T_420.onnx DAMO-YOLO-COCO damo_yolo_t.yaml 32.45MB baidu | github
damoyolo_tinynasL25_S_460.onnx DAMO-YOLO-COCO damo_yolo_s.yaml 62.09MB baidu | github
damoyolo_tinynasL35_M_492.onnx DAMO-YOLO-COCO damo_yolo_m.yaml 107.58MB baidu | github
damoyolo_tinynasL45_L_508.onnx DAMO-YOLO-COCO damo_yolo_l.yaml 160.52MB baidu | github
Gold_n_dist.onnx Gold-YOLO-COCO gold_yolo_n.yaml 23.58MB baidu | github
Gold_s_pre_dist.onnx Gold-YOLO-COCO gold_yolo_s.yaml 89.06MB baidu | github
Gold_m_pre_dist.onnx Gold-YOLO-COCO gold_yolo_m.yaml 169.88MB baidu | github
Gold_l_pre_dist.onnx Gold-YOLO-COCO gold_yolo_l.yaml 286.79MB baidu | github
  • Oriented Bounding Box
Name Description Configuration Size Link
yolov5n_obb_drone_vehicle.onnx YOLOv5-OBB-DroneVehicle yolov5n_obb_drone_vehicle.yaml 8.39MB baidu | github
yolov5s_obb_csl_dotav10.onnx YOLOv5-OBB-DOTA-v1.0 yolov5s_obb_csl_dotav10.yaml 29.77MB baidu | github
yolov5m_obb_csl_dotav15.onnx YOLOv5-OBB-DOTA-v1.5 yolov5m_obb_csl_dotav15.yaml 83.59MB baidu | github
yolov5m_obb_csl_dotav20.onnx YOLOv5-OBB-DOTA-v2.0 yolov5m_obb_csl_dotav20.yaml 83.62MB baidu | github

OCR

Name Description Configuration Size Link
ch_PP-OCRv4_det_infer.onnx Ultra-lightweight model supporting Chinese, English, and multi-language text detection model ch_ppocr_v4.yaml 4.53MB baidu | github
ch_PP-OCRv4_rec_infer.onnx Ultra-lightweight model supporting Chinese, English, and digits recognition model ch_ppocr_v4.yaml 10.33MB baidu | github
ch_ppocr_mobile_v2.0_cls_infer.onnx Original classifier model, classifying the detected text line angles ch_ppocr_v4.yaml 569KB baidu | github

Segment-Anything-Model

  • General Scenario
Name Description Configuration Size Link
edge_sam_encoder.onnx EdgeSAM edge_sam.yaml 21.02MB baidu | github
edge_sam_decoder.onnx EdgeSAM edge_sam.yaml 17.78MB baidu | github
sam_vit_b_01ec64.encoder.onnx SAM ViT-base encoder segment_anything_vit_b.yaml 342.58MB baidu | github
sam_vit_b_01ec64.decoder.onnx SAM ViT-base decoder segment_anything_vit_b.yaml 15.74MB baidu | github
sam_vit_b_01ec64.encoder.quant.onnx SAM ViT-base encoder(Quantized) segment_anything_vit_b_quant.yaml 103.78MB baidu | github
sam_vit_b_01ec64.decoder.quant.onnx SAM ViT-base decoder(Quantized) segment_anything_vit_b_quant.yaml 8.34MB baidu | github
sam_vit_l_0b3195.encoder.onnx SAM ViT-large encoder segment_anything_vit_l.yaml 1.15GB baidu | github
sam_vit_l_0b3195.decoder.onnx SAM ViT-large decoder segment_anything_vit_l.yaml 15.74MB baidu | github
sam_vit_l_0b3195.encoder.quant.onnx SAM ViT-large encoder(Quantized) segment_anything_vit_l_quant.yaml 317.18MB baidu | github
sam_vit_l_0b3195.decoder.quant.onnx SAM ViT-large decoder(Quantized) segment_anything_vit_l_quant.yaml 8.34MB baidu | github
mobile_sam.encoder.onnx MobileSAM encoder mobile_sam_vit_h.yaml 26.85MB baidu | github
sam_vit_h_4b8939.decoder.quant.onnx MobileSAM decoder mobile_sam_vit_h.yaml 8.34MB baidu | github
sam_vit_h_4b8939.encoder.quant.onnx SAM ViT-huge encoder(Quantized) segment_anything_vit_h_quant.yaml 626.40MB baidu | github
sam_vit_h_4b8939.decoder.quant.onnx SAM ViT-huge decoder(Quantized) segment_anything_vit_h_quant.yaml 8.34MB baidu | github
efficientvit_sam_l0_vit_h.encoder.onnx EfficientViT-SAM ViT-huge encoder efficientvit_sam_l0_vit_h.yaml 117.25MB baidu | github
efficientvit_sam_l0_vit_h.decoder.onnx EfficientViT-SAM ViT-huge decoder efficientvit_sam_l0_vit_h.yaml 15.63MB baidu | github
efficientvit_sam_l1_vit_h.encoder.onnx EfficientViT-SAM ViT-huge encoder efficientvit_sam_l1_vit_h.yaml 166.31MB baidu | github
efficientvit_sam_l1_vit_h.decoder.onnx EfficientViT-SAM ViT-huge decoder efficientvit_sam_l1_vit_h.yaml 15.63MB baidu | github
sam_hq_vit_b_encoder.onnx HQ-SAM ViT-base encoder sam_hq_vit_b.yaml 342.37MB baidu | github
sam_hq_vit_b_decoder.onnx HQ-SAM ViT-base decoder sam_hq_vit_b.yaml 19.74MB baidu | github
sam_hq_vit_l_encoder.onnx HQ-SAM ViT-large encoder sam_hq_vit_l.yaml 1.15GB baidu | github
sam_hq_vit_l_decoder.onnx HQ-SAM ViT-large decoder sam_hq_vit_l.yaml 20.74MB baidu | github
sam_hq_vit_l_encoder_quant.onnx HQ-SAM ViT-large encoder(Quantized) sam_hq_vit_l_quant.yaml 307.96MB baidu | github
sam_hq_vit_l_decoder HQ-SAM ViT-large decoder sam_hq_vit_l_quant.yaml 20.74MB baidu | github
sam_hq_vit_h_encoder_quant.onnx HQ-SAM ViT-huge encoder(Quantized) sam_hq_vit_h_quant.yaml 625.55MB baidu | github
sam_hq_vit_h_decoder HQ-SAM ViT-huge decoder sam_hq_vit_h_quant.yaml 21.74MB baidu | github
  • Medical Scenario
Name Description Configuration Size Link
sam-med2d_b.encoder.onnx SAM-Med2D ViT-base encoder sam_med2d_vit_b.yaml 1019.49MB baidu | github
sam-med2d_b.decoder.onnx SAM-Med2D ViT-base decoder sam_med2d_vit_b.yaml 15.60MB baidu | github
medsam_vit_b.encoder.onnx MedSAM ViT-base encoder medsam_vit_b.yaml 342.58MB baidu | github
medsam_vit_b.decoder.onnx MedSAM ViT-base decoder medsam_vit_b.yaml 15.74MB baidu | github
sam_model_best_large_ssl_buidnewprocess.encoder.onnx LVMSAM-Ultrasound Breast Cancer Segmentation Model with ViT-base encoder lvm_sam_ssk_buid_vit_b.yaml 342.58MB baidu | github
sam_model_best_large_ssl_buidnewprocess.decoder.onnx LVMSAM-Ultrasound Breast Cancer Segmentation Model ViT-base decoder lvm_sam_ssk_buid_vit_b.yaml 15.74MB baidu | github
sam_model_best_large_ssl_isiconlytrain.encoder.onnx LVMSAM-Skin Lesion Segmentation Model ViT-base encoder lvm_sam_ssk_isic_vit_b.yaml 342.58MB baidu | github
sam_model_best_large_ssl_isiconlytrain.decoder.onnx LVMSAM-Skin Lesion Segmentation Model ViT-base decoder lvm_sam_ssk_isic_vit_b.yaml 15.74MB baidu | github
sam_model_best_large_ssl_kvasir.encoder.onnx LVMSAM-Colorectal Polyp Segmentation Model ViT-base encoder lvm_sam_ssk_kvasir_vit_b.yaml 342.58MB baidu | github
sam_model_best_large_ssl_kvasir.decoder.onnx LVMSAM-Colorectal Polyp Segmentation Model ViT-base decoder lvm_sam_ssk_kvasir_vit_b.yaml 15.74MB baidu | github

Segmentation

Name Description Configuration Size Link
yolov5s-seg.onnx YOLOv5-COCO yolov5s_seg.yaml 29.45MB baidu | github
yolov8x-seg.onnx YOLOv8-COCO yolov8x_seg.yaml 274.10MB baidu | github
yolov8l-seg.onnx YOLOv8-COCO yolov8l_seg.yaml 175.59MB baidu | github
yolov8m-seg.onnx YOLOv8-COCO yolov8m_seg.yaml 104.23MB baidu | github
yolov8s-seg.onnx YOLOv8-COCO yolov8s_seg.yaml 45.25MB baidu | github
yolov8n-seg.onnx YOLOv8-COCO yolov8n_seg.yaml 13.18MB baidu | github

Union Task

Name Description Configuration Size Link
resnet50.onnx ResNet50-ImageNet yolov5s_resnet50.yaml 97.42MB baidu | github
yolov5s.onnx YOLOv5-COCO yolov5s_resnet50.yaml 27.98MB baidu | github
mobile_sam.encoder.onnx MobileSAM encoder(YOLOv5-SAM) yolov5s_mobile_sam_vit_h.yaml 26.85MB baidu | github
sam_vit_h_4b8939.decoder.quant.onnx MobileSAM decoder(YOLOv5-SAM) yolov5s_mobile_sam_vit_h.yaml 8.34MB baidu | github
yolov5s.onnx YOLOv5-COCO(YOLOv5-SAM) yolov5s_mobile_sam_vit_h.yaml 27.98MB baidu | github
efficientvit_sam_l0_vit_h.encoder.onnx YOLOv8-EfficientViT-SAM yolov8n_efficientvit_sam_l0_vit_h.yaml 117.25MB baidu | github
efficientvit_sam_l0_vit_h.decoder.onnx YOLOv8-EfficientViT-SAM yolov8n_efficientvit_sam_l0_vit_h.yaml 15.63MB baidu | github
yolov8n.onnx YOLOv8-EfficientViT-SAM yolov8n_efficientvit_sam_l0_vit_h.yaml 12.21MB baidu | github
yolov5m.onnx YOLOv5-RAM yolov5m_ram.yaml 81.19MB baidu | github
ram_swin_large_14m.onnx YOLOv5-RAM yolov5m_ram.yaml 865.66MB baidu | github

Visual-Language LLM

Name Description Configuration Size Link
groundingdino_swint_ogc_quant.onnx GroundingDINO groundingdino_swint_ogc_quant.yaml 171.28MB baidu | github
groundingdino_swinb_cogcoor_quant.onnx GroundingDINO groundingdino_swinb_cogcoor_quant.yaml 258.90MB baidu | github
ram_swin_large_14m.onnx Recognize Anything ram_swin_large_14m.yaml 865.66MB baidu | github
groundingdino_swint_ogc_quant.onnx GroundingSAM-SwinB with HQ-SAM-VitL-QInt8 groundingdino_swinb_attn_fuse_sam_hq_vit_l_quant.yaml 964.04MB baidu | github
sam_hq_vit_l_encoder_quant.onnx GroundingSAM-SwinB with HQ-SAM-VitL-QInt8 groundingdino_swinb_attn_fuse_sam_hq_vit_l_quant.yaml 307.96MB baidu | github
sam_hq_vit_l_decoder GroundingSAM-SwinB with HQ-SAM-VitL-QInt8 groundingdino_swinb_attn_fuse_sam_hq_vit_l_quant.yaml 20.74MB baidu | github