Chess Piece Detection Using SSD

End to end object detection project using SSD(Single Shot Detection).
The training is done using TFOD2.0(Tensorflow object detection) 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

After setting up the virtual environment download the trained weights from here.
After downloading the trained weights place it under the directory services/chess_piece_detector/application/ai/weights/
After performing the above steps go to services/chess_piece_detector/api and run app.py

python app.py

After running the app.py the web app can be accessed at http://127.0.0.1:8000/ 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.