产品识别 - 2.5 简体中文

Vitis AI Library 用户指南 (UG1354)

Document ID
UG1354
Release Date
2022-06-15
Version
2.5 简体中文

PMG 模型可用于高精度物品的产品识别,例如,RP2K 数据集。该模型基于 ResNet18,详细的模型结构如下图所示。在 rp2k 数据集上,此模型可达成 96.4% 的最高浮动准确度,含 13.82M 个参数和 2.28G 每秒浮点运算次数。模型最终部署准确度和量化后的最高准确度分别为 96.19% 和 96.18%。

图 1. 量产识别示例

下表列出了 Vitis AI Library 支持的 PMG 模型。

表 1. PMG 模型
编号 模型名称 框架
1 pmg_pt PyTorch