The Vitis™ AI Library contains the following types of neural network libraries based on Caffe framework:
- Classification
- Face detection
- SSD detection
- Pose detection
- Semantic segmentation
- Road line detection
- YOLOV3 detection
- YOLOV2 detection
- Openpose detection
- RefineDet detection
- ReID detection
- Multitask
- Face recognition
- Plate detection
- Plate recognition
- Medical segmentation
Also, the Vitis™ AI contains the following types of neural network libraries based on TensorFlow framework:
- Classification
- SSD detection
- YOLOv3 detection
- Medical detection
And, the Vitis™ AI supports the following type of neural network libraries based on PyTorch framework.
- Classification
- ReID detection
- Face recognition
- Semantic segmentation
- Point cloud
- Medical segmentation
- 3D segmentation
The related libraries are open source and can be modified as needed. The open source codes are available on Github.
The Vitis™ AI Library provides image test samples and video test samples for all the above networks. In addition, the kit provides the corresponding performance test program. For video based testing, we recommend to use raw video for evaluation. Decoding by software libraries on Arm® CPU may have inconsistent decoding time, which may affect the accuracy of evaluation.