demo.mp4
https://www.kaggle.com/datasets/manikchitralwar/webmarket-dataset
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 inruns/detect/exp
- Run
Training.ipynb
for training and keep the model weight (.pt
file) insideweights
We have used roboflow to create labels