模型支持
该版本的 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 | 支持 | 支持 | 不适用 | 不适用 | 支持 | 支持 | 不适用 | 不适用 | |
|