Pilldex: A pill recognition REST API using Convolutional Neural Networks.
These instructions will get you a copy of the project up and running on your local machine for development and testing purposes. See deployment for notes on how to deploy the project on a live system.
- Clone this repo
- Install requirements
- Train the model
- Run the script
- Check http://localhost:5000
/home/ruifgmonteiro/Desktop/example.png
- OpenCV
- Keras
- Python 3.6
- Sqlite3
- CUDA
- Flask
$ git clone https://github.com/ruifgmonteiro/pilldex.git
$ pip install -r requirements.txt
Make sure you have the following installed:
- tensorflow
- keras
- flask
- pillow
- h5py
- gevent
- sqlite3
Python 2.7 or 3.5+ are supported and tested.
$ python train.py --dataset dataset --model pilldex.model --labelbin lb.pickle
$ tensorboard --logdir=logs/ --port=8101
$ python app.py
Soon: Deploying webapp in Heroku.
- OpenCV - Computer Vision Library
- Flask - The web framework used
- Keras - Python high level deep learning library
- Tensorflow - An open source machine learning library for research and production.
- Sqlite3 - Database engine
- Rui Monteiro - Initial work - rfgm6
See also the list of contributors who participated in this project.
This project is licensed under the MIT License - see the LICENSE.md file for details
- Adrian Rosebrock from PyImageSearch