|
--- |
|
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() |
|
``` |