| --- |
| license: mit |
| --- |
| |
| # [Visual Product Recognition Challenge](https://www.aicrowd.com/challenges/visual-product-recognition-challenge-2023) |
|
|
|
|
| The trained models for the competition. The training code for the models can be found in [HCA97/Product-Recognition](https://github.com/HCA97/Product-Recognition). |
|
|
|
|
| # How to use it? |
|
|
| You need to install `open_clip` library. |
|
|
| ```bash |
| pip install open_clip |
| ``` |
|
|
| Example of loading the model: |
|
|
| ```py |
| model = open_clip.create_model_and_transforms('ViT-H-14', None)[0].visual |
| model.load_state_dict(th.load('path to model')) |
| model.half() |
| model.eval() |
| ``` |
