Annotations are present for 10 Indian Celebrities:
- Virat Kohli
- Katrina Kaif
- Alia Bhatt
- Shahrukh Khan
- Akshay Kumar
- Salman Khan
- Deepika Padukone
- Sachin Tendulkar
- Amitabh Bachchan
- Sonam Kapoor
Note: Make sure you have python and pip installed on your system if you don't visit Python pip
git clone https://github.com/NanoNets/celebrity-detection.git
cd celebrity-detection
sudo pip install requests
Get your free API Key from http://app.nanonets.com/user/api_key
export NANONETS_API_KEY=YOUR_API_KEY_GOES_HERE
python ./code/create-model.py
_Note: This generates a MODEL_ID that you need for the next step
export NANONETS_MODEL_ID=YOUR_MODEL_ID
_Note: you will get YOUR_MODEL_ID from the previous step
The training data is found in images
(image files) and annotations
(annotations for the image files)
python ./code/upload-training.py
Once the Images have been uploaded, begin training the Model
python ./code/train-model.py
The model takes ~2 hours to train. You will get an email once the model is trained. In the meanwhile you check the state of the model
python ./code/model-state.py
Once the model is trained. You can make predictions using the model
python ./code/prediction.py PATH_TO_YOUR_IMAGE.jpg
Sample Usage:
python ./code/prediction.py ./images/7972a1733e354ab5191874322080b50e.jpg
Note the python sample uses the comverted json instead of the xml payload for convenience purposes, hence it has no dependencies.