Vitis AI提供两种应用例程,VART和Vitis AI Library,下文详细描述了VART中自动驾驶应用的实现过程。
编译AI模型
对于DNNDK API和VART API编译AI库的方式相同,
编译yolov3模型
yolov3模型用于自动驾驶。
$~/Vitis-AI/AI-Model-Zoo/all_models_1.2/caffe
$cp -r dk_yolov3_cityscapes_256_512_0.9_5.46G_1.2/ ../../../Tool-Example/
$cd ~/Vitis-AI
$./docker_run.sh xilinx/vitis-ai:1.2.82
$conda activate vitis-ai-caffe
生成.dcf文件,修改.json文件,这两步已经完成。
在readme中:
remove parameter "yolo_height" and "yolo_width" in "transform_param" in data layer, like the following 2 lines:yolo_height: 256,yolo_width: 512,删除这两行进行编译。
- 运行编译指令
vai_c_