Product Recognition - 3.5 English

Vitis AI Library User Guide (UG1354)

Document ID
Release Date
3.5 English

PMG model can be used for fine-grained goods product recognition, for example, RP2K dataset. The model is ResNet18-based and the detailed model structure is shown in the picture below. On the rp2k dataset, this model can achieve 96.4% top-1 float accuracy with 13.82M parameters and 2.28G Flops. Model final deployment and quantized top-1 accuracies are 96.19% and 96.18%, respectively.

Figure 1. Product Recognition Example

The following table lists the PMG models supported by the Vitis AI Library.

Table 1. PMG Models
No Model Name Framework
1 pmg_pt PyTorch