YOLOv4 是 YOLOv3 的升级版本,作用与 YOLOv3 相同。下表列出了 Vitis AI Library 支持的 YOLOv4 检测模型。
编号 | 模型名称 | 框架 |
---|---|---|
1 | yolov4_leaky_spp_m | Caffe |
2 | yolov4_leaky_spp_m_pruned_0_36 | |
3 | face_mask_detection_pt | PyTorch |
4 | yolov4_csp_pt | |
5 | yolov4_leaky_416_tf | TensorFlow |
6 | yolov4_leaky_512_tf | TensorFlow |