This is the repository developed for 'Image Classifier in TensorFlow in 5 Min on YouTube using this CodeLab by Google as a guide.
Scientists can use this classifier to automatically label whether a bird image is of a Malayan Night Heron or of a Night Heron.
So the bird on the top is a malayan night heron and the bottom is night heron.
-
Start the docker image
docker run -it -v <absolute_path_to_this_directory>:/tf_files/ gcr.io/tensorflow/tensorflow:latest-devel
-
Run the label_image script to label the image in the docker container.
python /tf_files/label_image.py <path_to_file>
Final test accuracy = 76.4%
- malayan night heron (score = 0.78996)
- night heron (score = 0.21004)
- malayan night heron (score = 0.99932)
- night heron (score = 0.00068)
- night heron (score = 0.94634)
- malayan night heron (score = 0.05366)