SmartStock AI is an intelligent inventory management system that uses machine learning and computer vision to help warehouses predict, monitor, and optimize their stock levels.
It brings together real-time stock insights, AI-based forecasting, and object detection for a smarter and more automated inventory workflow.
- π¦ Predictive Inventory Forecasting β Anticipates stock needs based on past trends.
- π§ AI-Powered Object Detection β Uses YOLOv8 to identify items in uploaded shelf images.
- π Interactive Dashboard β Built with Streamlit for seamless visualization and control.
- π· Image-Based Shelf Scanning β Upload shelf images to automatically detect missing items.
- π Modular Design β Combines dynamic stock prediction and visual analysis in one app.
- β‘ Real-Time Insights β Detect anomalies, track stock levels, and visualize performance instantly.
SMARTSTOCKAI/
βββ data/
β βββ anomalies.csv
β βββ forecast_pre.csv
β βββ inventory_data.csv
β βββ product_details.csv
βββ forecasts/
βββ images/
βββ models/
β βββ anomaly_detector.pkl
β βββ forecast_model.pkl
βββ scripts/
β βββ __pycache__/
β βββ generate_sample_data.py
β βββ predict_api.py
β βββ preprocess.py
β βββ train_anomaly.py
β βββ train_prophet.py
βββ app.py
βββ README.md
βββ shelf_image.py
βββ shelf_monitor.py
βββ smartstock_vision_app.py
βββ SmartStockAI.ipynb
βββ yolov8n.pt
ββ requirements.txt
ββ README.md- Language: Python 3.10+
- Frontend: Streamlit (interactive web app)
- ML & Forecasting: Scikit-learn, NumPy, Pandas
- Computer Vision: OpenCV, Ultralytics YOLOv8
- Visualization: Matplotlib, Seaborn, Plotly
- Environment: Virtualenv / Conda
1οΈβ£ Clone the Repository
git clone https://github.com/harishy0406/SmartStockAI.git
cd SmartStockAI2οΈβ£ Create Virtual Environment
python -m venv venv
# Activate it
venv\Scripts\activate # (Windows)
source venv/bin/activate # (Linux/Mac)3οΈβ£ Install Required Libraries
pip install -r requirements.txt4οΈβ£ Run the Streamlit App
- For SmartStockAI Dashboard
streamlit run app.py- For Shelf Image Detection
streamlit run app_upload.py
- π Automated Low-Stock Alerts via Email or SMS
- βοΈ Cloud Dashboard for multi-warehouse integration
- πΉ Live Shelf Monitoring using IP cameras
- π€ Vision Transformer Integration (ViT) for improved recognition
- π§Ύ ERP API Integration for seamless business workflows
Want to make SmartStock AI even smarter? Fork the repo, improve features, and create a pull request! Letβs redefine warehouse intelligence β together π