Image Recognition

This is an image recognition project with API:

• Inception_v3 model pre-trained with images on ImageNet
• Django REST Framework

Quick video to see how easy you can run image recognition on your local machine:

Installation

• Python v3.6 and above
• Django
• Django REST Framework
• PyTorch

Usage

1. Simply run

   . start-api.sh

2. Open localhost:8000, click the link "api": "http://localhost:8000/api/" in as marked in following screenshot
3. Enter the image path into following Imagepath and click POST button

The image is recognized by AI model and return the top 1 result to you

Support

• Contact engineer: [email protected]

Roadmap

• display top 3 result with confidence score
• deploy to DigitalOcean, host it on server
• merge other pre-trained model for fun!

Authors

Chloe Wang

• GitHub
• LinkedIn
• Twitter

Contributing

pull requests are welcome. for major changes, please open an issue first to discuss what you would like to change.

Please make sure to update tests as appropriate.

License

MIT

Acknowledgement

Thanks to readme for this template and tools.