Vitis AI Library v2.5 支持的模型 - 2.5 简体中文

Vitis AI Library 用户指南 (UG1354)

Document ID
UG1354
Release Date
2022-06-15
Version
2.5 简体中文

模型支持

该版本的 Vitis™ AI Library 支持下列模型。

表 1. Vitis AI Library 支持的模型
编号 神经网络 ZCU102/ ZCU104/ KV260 VCK190 U50LV-DPUCAHX8H U50LV/ U55C-DPUCAHX8H-DWC VCK5000-DPUCVDX8H-4PE VCK5000- DPUCVDX8H- 6PE- aieMISC VCK5000- DPUCVDX8H- 6PE- aieDWC VCK5000- DPUCVDX8H- 8PE 应用
1 inception_resnet_v2_tf 支持 支持 支持 支持 支持 支持 支持 不适用 图像分类
2 inception_v1_tf 支持 支持 支持 支持 支持 支持 支持 支持
3 inception_v3_tf 支持 支持 支持 支持 支持 支持 支持 不适用
4 inception_v4_2016_09_09_tf 支持 支持 支持 支持 支持 支持 支持 不适用
5 mobilenet_v1_0_25_128_tf 支持 支持 不适用 支持 支持 不适用 支持 不适用
6 mobilenet_v1_0_5_160_tf 支持 支持 不适用 支持 支持 不适用 支持 不适用
7 mobilenet_v1_1_0_224_tf 支持 支持 不适用 支持 支持 不适用 支持 不适用
8 mobilenet_v2_1_0_224_tf 支持 支持 不适用 支持 支持 不适用 支持 不适用
9 mobilenet_v2_1_4_224_tf 支持 支持 不适用 支持 支持 不适用 支持 不适用
10 resnet_v1_101_tf 支持 支持 支持 支持 支持 支持 支持 支持
11 resnet_v1_152_tf 支持 支持 支持 支持 支持 支持 支持 支持
12 resnet_v1_50_tf 支持 支持 支持 支持 支持 支持 支持 支持
13 vgg_16_tf 支持 支持 支持 支持 支持 支持 支持 不适用
14 vgg_19_tf 支持 支持 支持 支持 支持 支持 支持 不适用
15 ssd_mobilenet_v1_coco_tf 支持 支持 不适用 支持 支持 不适用 支持 不适用 对象检测
16 ssd_mobilenet_v2_coco_tf 支持 支持 不适用 支持 支持 不适用 支持 不适用
17 ssd_resnet_50_fpn_coco_tf 支持 支持 支持 支持 支持 支持 支持 支持
18 yolov3_voc_tf 支持 支持 支持 支持 支持 支持 支持 支持
19 mlperf_ssd_resnet34_tf 支持 支持 支持 支持 支持 支持 支持 支持
20 resnet50 支持 支持 支持 支持 支持 支持 支持 支持 图像分类
21 resnet18 支持 支持 支持 支持 支持 支持 支持 支持
22 inception_v1 支持 支持 支持 支持 支持 支持 支持 支持
23 inception_v2 支持 支持 支持 支持 支持 支持 支持 不适用
24 inception_v3 支持 支持 支持 支持 支持 支持 支持 不适用
25 inception_v4 支持 支持 支持 支持 支持 支持 支持 不适用
26 mobilenet_v2 支持 支持 不适用 支持 支持 不适用 支持 不适用
27 squeezenet 支持 支持 支持 支持 支持 支持 支持 支持
28 ssd_pedestrian_pruned_0_97 支持 支持 支持 支持 支持 支持 支持 支持 ADAS 行人检测
29 ssd_traffic_pruned_0_9 支持 支持 支持 支持 支持 支持 支持 支持 交通流量检测
30 ssd_adas_pruned_0_95 支持 支持 支持 支持 支持 支持 支持 支持 ADAS 车辆检测
31 ssd_mobilenet_v2 支持 支持 不适用 支持 支持 不适用 支持 不适用 对象检测
32 refinedet_pruned_0_8 支持 支持 支持 支持 支持 支持 支持 支持
33 refinedet_pruned_0_92 支持 支持 支持 支持 支持 支持 支持 支持
34 refinedet_pruned_0_96 支持 支持 支持 支持 支持 支持 支持 支持
35 vpgnet_pruned_0_99 支持 支持 支持 支持 支持 支持 支持 支持 ADAS 车道检测
36 fpn 支持 支持 支持 支持 支持 支持 支持 支持 ADAS 分割
37 sp_net 支持 支持 支持 支持 支持 支持 支持 支持 姿势估算
38 openpose_pruned_0_3 支持 支持 支持 支持 支持 支持 支持 支持
39 densebox_320_320 支持 支持 支持 支持 支持 支持 支持 支持 面部检测
40 densebox_640_360 支持 支持 支持 支持 支持 支持 支持 支持
41 face_landmark 支持 支持 支持 支持 支持 支持 支持 支持 面部检测和识别
42 reid 支持 支持 支持 不适用 不适用 不适用 不适用 不适用 对象跟踪
43 multi_task 支持 支持 支持 支持 支持 支持 支持 支持 ADAS
44 yolov3_adas_pruned_0_9 支持 支持 支持 支持 支持 支持 支持 支持 对象检测
45 yolov3_voc 支持 支持 支持 支持 支持 支持 支持 支持
46 yolov3_bdd 支持 支持 支持 支持 支持 支持 支持 支持
47 yolov2_voc 支持 支持 支持 支持 支持 支持 支持 支持
48 yolov2_voc_pruned_0_66 支持 支持 支持 支持 支持 支持 支持 支持
49 yolov2_voc_pruned_0_71 支持 支持 支持 支持 支持 支持 支持 支持
50 yolov2_voc_pruned_0_77 支持 支持 支持 支持 支持 支持 支持 支持
51 facerec_resnet20 支持 支持 支持 支持 支持 支持 支持 不适用 面部识别
52 facerec_resnet64 支持 支持 支持 支持 支持 支持 支持 不适用
53 plate_detection 支持 支持 支持 支持 支持 支持 支持 支持 车牌识别
54 plate_recognition 支持 支持 支持 支持 支持 支持 支持 支持
55 FPN_Res18_Medical_segmentation 支持 支持 支持 支持 支持 支持 支持 支持 医学分割
56 refinedet_baseline 支持 支持 支持 支持 支持 支持 支持 支持 对象检测
57 resnet50_pt 支持 支持 支持 支持 支持 支持 支持 支持 图像分类
58 squeezenet_pt 支持 支持 支持 支持 支持 支持 支持 支持
59 inception_v3_pt 支持 支持 支持 支持 支持 支持 支持 不适用
60

personreid-res50_pt

支持 支持 支持 支持 支持 支持 支持 不适用 对象跟踪
61

facereid-large_pt

支持 支持 支持 支持 支持 支持 支持 不适用
62

facereid-small_pt

支持 支持 支持 支持 支持 支持 支持 不适用
63

SemanticFPN_cityscapes_pt

支持 支持 支持 支持 支持 支持 支持 支持 分割
64

facerec-resnet20_mixed_pt

支持 支持 支持 支持 支持 支持 支持 不适用 面部识别
65 face-quality_pt 支持 支持 支持 支持 支持 支持 支持 支持
66 MT-resnet18_mixed_pt 支持 支持 不适用 支持 支持 不适用 支持 不适用 ADAS
67 salsanext_pt 支持 支持 支持 支持 支持 支持 支持 支持 点云
68 pointpillars_kitti_12000_0_pt

pointpillars_kitti_12000_1_pt

支持 支持 不适用 不适用 支持 支持 不适用 不适用
69 unet_chaos-CT_pt 支持 支持 支持 支持 支持 支持 支持 支持 CT 分割
70 FPN-resnet18_covid19-seg_pt 支持 支持 支持 支持 支持 支持 支持 支持 Covid-19 分割
71 ENet_cityscapes_pt 支持 支持 支持 支持 支持 支持 支持 支持 分割
72 personreid-res18_pt 支持 支持 支持 支持 支持 支持 支持 不适用 对象跟踪
73 yolov4_leaky_spp_m 支持 支持 支持 支持 支持 支持 支持 不适用 对象检测
74 hourglass-pe_mpii 支持 支持 不适用 不适用 支持 不适用 不适用 不适用 姿势估算
75 retinaface 支持 支持 不适用 支持 支持 不适用 支持 不适用 面部检测
76 FPN-resnet18_Endov 支持 支持 不适用 不适用 支持 不适用 不适用 不适用 机器人仪器分割
77 tiny_yolov3_vmss 支持 支持 支持 支持 支持 支持 支持 支持 对象检测
78 face-quality 支持 支持 支持 支持 支持 支持 支持 支持 面部识别
79 ssdlite_mobilenet_v2_coco_tf 支持 支持 不适用 支持 支持 不适用 支持 不适用 对象检测
80 ssd_inception_v2_coco_tf 支持 支持 不适用 支持 支持 不适用 支持 不适用
81 MLPerf_resnet50_v1.5_tf 支持 支持 支持 支持 支持 支持 支持 支持 图像分类
82 mobilenet_edge_1_0_tf 支持 支持 不适用 支持 支持 不适用 支持 不适用
83 mobilenet_edge_0_75_tf 支持 支持 不适用 支持 支持 不适用 支持 不适用
84 refinedet_VOC_tf 支持 支持 支持 支持 支持 支持 支持 支持 对象检测
85 RefineDet-Medical_EDD_tf 支持 支持 支持 支持 支持 支持 支持 支持 医疗检测
86 resnet_v2_50_tf 支持 支持 不适用 不适用 支持 不适用 不适用 不适用 图像分类
87 resnet_v2_101_tf 支持 支持 不适用 不适用 支持 不适用 不适用 不适用
88 resnet_v2_152_tf 支持 支持 不适用 不适用 支持 不适用 不适用 不适用
89 mobilenet_v2_cityscapes_tf 支持 支持 不适用 不适用 支持 不适用 支持 不适用 分割
90 inception_v2_tf 支持 支持 不适用 支持 支持 不适用 支持 不适用 图像分类
91 resnet50_tf2 支持 支持 支持 支持 支持 支持 支持 支持
92 mobilenet_1_0_224_tf2 支持 支持 不适用 支持 支持 不适用 支持 不适用
93 inception_v3_tf2 支持 支持 支持 支持 支持 支持 支持 不适用
94 medical_seg_cell_tf2 支持 支持 支持 支持 支持 支持 支持 支持 医学分割
95 semantic_seg_citys_tf2 支持 支持 支持 支持 支持 支持 支持 支持 分割
96 efficientNet-edgetpu-S_tf 支持 支持 不适用 支持 支持 不适用 支持 不适用 图像分类
97 efficientNet-edgetpu-M_tf 支持 支持 不适用 支持 支持 不适用 支持 不适用
98 efficientNet-edgetpu-L_tf 支持 支持 不适用 不适用 支持 不适用 不适用 不适用
99 SemanticFPN_Mobilenetv2_pt 支持 支持 不适用 支持 支持 不适用 支持 不适用 分割
100 pointpillars_nuscenes_40000_64_0_pt

pointpillars_nuscenes_40000_64_1_pt

支持 支持 支持 支持 支持 支持 支持 不适用 三维对象检测
101 pointpainting_nuscenes_40000_64_0_pt

pointpainting_nuscenes_40000_64_1_pt

支持 支持 支持 支持 支持 支持 支持 不适用 二维语义分割和三维对象检测
102 salsanext_v2_pt 支持 支持 支持 支持 支持 支持 支持 支持 三维分割
103 centerpoint_0_pt

centerpoint_1_pt

支持 支持 不适用 不适用 支持 支持 不适用 不适用 4D 雷达检测
104 multi_task_v3_pt 支持 支持 不适用 支持 支持 不适用 支持 不适用 ADAS
105 FADNet_0_pt

FADNet_1_pt

FADNet_2_pt

支持 支持 支持 支持 支持 支持 支持 支持 深度估算
106 rcan_pruned_tf 支持 支持 支持 支持 支持 支持 支持 支持 超分辨率
107 efficientnet-b0_tf2 支持 支持 不适用 不适用 支持 不适用 不适用 不适用 分类
108 yolov4_leaky_spp_m_pruned_0_36 支持 支持 支持 支持 支持 支持 支持 不适用 对象检测
109 pmg_pt 支持 支持 支持 支持 支持 支持 支持 不适用 品牌识别
110 bcc_pt 支持 支持 不适用 支持 支持 不适用 支持 不适用 贝叶斯人群计数
111 SA_gate_base_pt 支持 支持 不适用 不适用 支持 不适用 不适用 不适用 室内分割
112 ultrafast_pt 支持 支持 支持 支持 支持 支持 支持 支持 道路标线检测
113 HardNet_MSeg_pt 支持 支持 不适用 不适用 支持 不适用 不适用 不适用 息肉分割
114 drunet_pt 支持 支持 支持 支持 支持 支持 支持 支持 图像去噪
115 person-orientation_pruned_558m_pt 支持 支持 支持 支持 支持 支持 支持 不适用 人体朝向估计
116 ofa_resnet50_0_9B_pt 支持 支持 支持 支持 支持 支持 支持 不适用 分类
117 SESR_S_pt 支持 支持 支持 支持 支持 支持 支持 支持 图像超分辨率
118 C2D2_Lite_0_pt

C2D2_Lite_1_pt

支持 支持 不适用 支持 支持 不适用 支持 不适用 覆盖范围预测
119 ofa_depthwise_res50_pt 支持 支持 不适用 支持 支持 不适用 支持 不适用 分类
120 FADNet_pruned_0_pt

FADNet_pruned_1_pt

FADNet_pruned_2_pt

支持 支持 不适用 不适用 支持 不适用 支持 不适用 深度估算
121 PSMNet_pruned_0_pt

PSMNet_pruned_1_pt

PSMNet_pruned_2_pt

不适用 支持 不适用 不适用 不适用 不适用 不适用 不适用
122 FairMot_pt 支持 支持 支持 支持 支持 支持 支持 支持 联合检测与跟踪
123 mobilenet_v3_small_1_0_tf2 支持 支持 不适用 不适用 支持 不适用 不适用 不适用 分类
124 ssr_pt 支持 支持 不适用 支持 支持 不适用 支持 不适用 光谱移除
125 clocs_pointpillars_kitti_0_pt

clocs_pointpillars_kitti_1_pt

clocs_fusion_cnn_pt

clocs_yolox_pt

支持 支持 不适用 不适用 支持 支持 不适用 不适用 图像与激光雷达融合三维检测
126 tsd_yolox_pt 支持 支持 支持 支持 支持 支持 支持 不适用 交通标志检测
127 solo_pt 支持 支持 不适用 不适用 支持 支持 不适用 不适用 实例分割
128 chen_color_resnet18_pt 支持 支持 支持 支持 支持 支持 支持 支持 分类
129 face_mask_detection_pt 支持 支持 不适用 支持 支持 不适用 支持 不适用 面罩检测
130 ofa_rcan_latency_pt 支持 支持 支持 支持 支持 支持 支持 支持 超分辨率
131 ocr_pt 支持 支持 支持 支持 支持 支持 支持 支持 OCR
132 textmountain_pt 支持 支持 支持 支持 支持 支持 支持 支持 OCR
133 vehicle_make_resnet18_pt 支持 支持 支持 支持 支持 支持 支持 支持 分类
134 vehicle_type_resnet18_pt 支持 支持 支持 支持 支持 支持 支持 支持 分类
135 ofa_yolo_pt 支持 支持 支持 支持 支持 支持 支持 不适用 对象检测
136 ofa_yolo_pruned_0_30_pt 支持 支持 支持 支持 支持 支持 支持 不适用
137 ofa_yolo_pruned_0_50_pt 支持 支持 支持 支持 支持 支持 支持 不适用
138 efficientdet_d2_tf 支持 支持 不适用 不适用 不适用 不适用 不适用 不适用
139 superpoint_tf 支持 支持 不适用 不适用 不适用 不适用 不适用 不适用 SLAM
140 hfnet_tf 支持 支持 不适用 不适用 不适用 不适用 不适用 不适用 SLAM
141 movenet_ntd_pt 支持 支持 不适用 支持 支持 不适用 支持 不适用 姿势估算
142 yolov3_coco_416_tf2 支持 支持 支持 支持 支持 支持 支持 支持 对象检测
143 yolov4_leaky_416_tf 支持 支持 不适用 不适用 支持 支持 不适用 不适用
144 yolov4_leaky_512_tf 支持 支持 不适用 不适用 支持 支持 不适用 不适用
  1. 含有“_tf”或“_tf2”后缀的网络是在 TensorFlow 上训练的。
  2. 含有“_pt”的网络是在 PyTorch 上训练的。
  3. 不含后缀的网络是在 Caffe 上训练的。