PMG 模型可用于高精度物品的产品识别,例如,RP2K 数据集。该模型基于 ResNet18,详细的模型结构如下图所示。在 rp2k 数据集上,此模型可达成 96.4% 的最高浮动准确度,含 13.82M 个参数和 2.28G 每秒浮点运算次数。模型最终部署准确度和量化后的最高准确度分别为 96.19% 和 96.18%。
图 1. 量产识别示例
下表列出了 Vitis AI Library 支持的 PMG 模型。
编号 | 模型名称 | 框架 |
---|---|---|
1 | pmg_pt | PyTorch |