YOLOv5 is an upgraded version of YOLOv4 and does the same thing as YOLOv4. YOLOv5 uses the PyTorch framework.
The following table lists the YOLOv5 detection models supported by the Vitis AI Library.
No | Model Name | Framework |
---|---|---|
1 | yolov5_large_pt | PyTorch |
2 | yolov5s6_pt | |
3 | yolov5_nano_pt | |
4 | yolov5_m_pt | |
5 | yolov5l_pt |