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.