/prc-seg

Price segmentation module for CVPR'21 Retail Vision workshop solution

Primary LanguagePythonMIT LicenseMIT

Price segmentation model

input output

Locate bounding box with a price on a pricelabel with semantic segmentation model.

Train

  1. Clone this repo
  2. Install requirements pip install -r requirements.txt
  3. Run training python train.py

Inference

See example.py