product-pricing

This repo contains inference code for the solution to Product Pricing Challenge.

You can find more information in the report.

Quick start

Get models

Yolov5-based detection model, price segmentation model, OCR model, GNN model.

Product embedding model might be taken from the original repo.

Build

make build
make run

Data

See the main challenge page for more information.

Step-by-step inference

1. Detect & recognize prices

python prod_pricing/predict_prices.py

2. Get product embeddings using pretrained model

python prod_pricing/prod_embeddings.py

3. Produce matching

python prod_pricing/matching/match_better.py

4. You are breathtaking!