Libraries and Samples - 2.0 English

Vitis AI Library User Guide (UG1354)

Document ID
UG1354
Release Date
2022-01-20
Version
2.0 English

TensorFlow Framework

The Vitis™ AI contains the following neural network libraries based on the TensorFlow framework:

PyTorch Framework

The Vitis™ AI supports the following type of neural network libraries based on the PyTorch framework.

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, use the raw video for evaluation. Decoding by software libraries on Arm® processors may have inconsistent decoding time, which may affect the accuracy of evaluation.

Note: For Edge devices, all the sample programs can only run on the target side but they can be cross-compiled on the host side or compiled on the target side.