OFA_YOLO 是用于对象检测的神经网络。输入是一张包含一个或多个对象的 640x640 图片,输出是由检测的信息组成的结果结构体的矢量。下图显示了 OFA_YOLO 检测的结果。
图 1. OFA_YOLO 检测示例
下表列出了 Vitis AI Library 支持的 OFA_YOLO 检测模型。
编号 | 模型名称 | 框架 |
---|---|---|
1 | ofa_yolo_pt | PyTorch |
2 | ofa_yolo_pruned_0_30_pt | |
3 | ofa_yolo_pruned_0_50_pt |