1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42
|
/*
* Copyright 2019 Xilinx Inc.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
Before running the program, please download the corresponding model and install it.
The models required by this sample are: ssd_pedestrian_pruned_0_97 and sp_net
You can find the detailed informantion of these models under
Vitis-AI/models/AI-Model-Zoo/model-list/cf_ssdpedestrian_coco_360_640_0.97_5.9G_1.3/model.yaml
Vitis-AI/models/AI-Model-Zoo/model-list/cf_SPnet_aichallenger_224_128_0.54G_1.3/model.yaml
In the model.yaml, you will find the model's download links for different platforms.
Please choose the corresponding model and download it.
Take ZCU102/ZCU104 as an example, execute the following commands to download and install the models.
wget https://www.xilinx.com/bin/public/openDownload?filename=ssd_pedestrian_pruned_0_97-zcu102_zcu104_kv260-r1.4.0.tar.gz -O ssd_pedestrian_pruned_0_97-zcu102_zcu104_kv260-r1.4.0.tar.gz
wget https://www.xilinx.com/bin/public/openDownload?filename=sp_net-zcu102_zcu104_kv260-r1.4.0.tar.gz -O sp_net-zcu102_zcu104_kv260-r1.4.0.tar.gz
mkdir -p /usr/share/vitis_ai_library/models
tar -xzvf ssd_pedestrian_pruned_0_97-zcu102_zcu104_kv260-r1.4.0.tar.gz
cp ssd_pedestrian_pruned_0_97 /usr/share/vitis_ai_library/models -r
tar -xzvf sp_net-zcu102_zcu104_kv260-r1.4.0.tar.gz
cp sp_net /usr/share/vitis_ai_library/models -r
For U50, execute the following commands.
wget https://www.xilinx.com/bin/public/openDownload?filename=ssd_pedestrian_pruned_0_97-u50-DPUCAHX8H-r1.4.0.tar.gz -O ssd_pedestrian_pruned_0_97-u50-DPUCAHX8H-r1.4.0.tar.gz
wget https://www.xilinx.com/bin/public/openDownload?filename=sp_net-u50-DPUCAHX8H-r1.4.0.tar.gz -O sp_net-u50-DPUCAHX8H-r1.4.0.tar.gz
sudo mkdir -p /usr/share/vitis_ai_library/models
tar -xzvf ssd_pedestrian_pruned_0_97-u50-DPUCAHX8H-r1.4.0.tar.gz
sudo cp ssd_pedestrian_pruned_0_97 /usr/share/vitis_ai_library/models -r
tar -xzvf sp_net-u50-DPUCAHX8H-r1.4.0.tar.gz
sudo cp sp_net /usr/share/vitis_ai_library/models -r
|