Automated AI self checkout system using CNN and Yolo. This project was created in collaboration with Rishi Bhattasali for Tech AI hackathon organized by IIC, TMSL.
demo.mp4
This projects aims to enhance the efficiency of grocery retail outlets with our AI-driven self-checkout system, leveraging computer vision technology to enable quick and seamless transactions, ensuring a streamlined shopping experience for customers. Our solution aims to replace traditional cashiers, thereby improving profit margins and enabling automated AI-driven sales tracking and inventory replenishment.
The datataset used in this project is created by our team and is available in the data folder which includes images for 7 Indian daily used grocery items.
This is a sample dataset, our future goal is to create a large dataset for various items to increase the efficiency as well coverage of our model.
-
After downloading the dataset, unzip it and place in the root directory.
-
Clone the repository
git clone https://github.com/priyanshudutta04/AI-Self-Checkout.git
- Install dependencies
pip install -r requirements.txt
- Run the Python File
python yolo_products.py
Note: If GPU is available install cuda toolkit and cuDNN for faster execution during Model-Training
Contributions are welcome! If you have ideas for improving the model or adding new features, please feel free to fork the repository and submit a pull request or open a issue.
If you like this project, do give it a ⭐and share it with your friends.