R-aryan/Warehouse_Apparel_Detection

End to end object detection project using YOLOV5 and serving as a REST API using Flask

Warehouse Apparel Detection

• End to end object detection project using YOLOV5.
• The training is done using YOLO V5 framework.
• The application is served as an REST API using Flask.

Steps to run the application

Setting up virtual environment.

• What is Virtual Environment in python ?
• Create virtual environment in python
• Create virtual environment Anaconda
• create a virtual environment and install requirements.txt

pip install -r requirements.txt

For Inference

• After performing the above steps go to app.py and run app.py

python app.py

• After running the app.py the web app can be accessed at http://127.0.0.1:9500/ copy this url and paste it in your browser.
• The UI will look like the following.

• The picture can be uploaded using the upload button and after uploading the image click on predict to perform inference.
• Sample Input

• Sample Output

• The application logs can also be found here.

For Training

• The training notebook can be found here
• Due to system/hardware limitations it was not possible to perform training locally.
• So google colab and google drive was used for training purpose.