/WebMarket

Analysis of images of retail shelf displays and detection the object on shelf

Primary LanguagePython

Analysis of images of retail shelf displays and detection the object on shelf

demo.mp4

Dataset

https://www.kaggle.com/datasets/manikchitralwar/webmarket-dataset

How to use this repo

  • git clone https://github.com/sayansaha934/WebMarket.git
  • change directory to WebMarket
  • conda create -n WebMarket python==3.6.9
  • conda activate WebMarket
  • pip install -r requirements.txt
  • install pytorch
  • Run python detect.py --weights weights/best.pt --source images --img 416 --conf 0.4, output images will be saved in runs/detect/exp
  • Run Training.ipynb for training and keep the model weight (.pt file) inside weights

We have used roboflow to create labels