ImageNet classification with Python and Keras using ResNet and VGG

• This module classify and tests the accuracy of ResNet and VGG models trained on ImageNet dataset.

• Loads the ResNet(winner) and VGG16 (runner up) of ImageNet Large Scale Recognition Challenge to predict and classify.

• Predicts and Classify to save the output in ./output/ directory.

• By default ResNet is selected for prediction which can be changed in the code by selecting the function.

Execution

1. pip3 install -r requirements.txt

2. python3 test_imagenet.py --image image.jpg

Results