Skip to content
Branch: master
Find file History
Latest commit b5f44c3 Apr 4, 2019
Type Name Latest commit message Commit time
Failed to load latest commit information.
network Fixing small issues in old tutorials Apr 4, 2019 Keras Finetuning tutorial complete code Dec 12, 2017 Adding ReadMe Dec 14, 2017 Adding ReadMe Dec 14, 2017

#1. Start pretraining by running this command: python -train ../tutorial-2-image-classifier/training_data/ -val ../tutorial-2-image-classifier/testing_data/ -num_class 2

#Now, a weights file with the name cv-tricks_pretrained_model.h5 should be saved. Now, in order to start fine-tuning process, run this command python -train ../tutorial-2-image-classifier/training_data/ -val ../tutorial-2-image-classifier/testing_data/ -num_class 2

#This will save the finetuned weights(cv-tricks_fine_tuned_model.h5) with 98%+ accuracy

#Now, you can use the predict script to run this trained model on a new image of a dog or cat python --image test.jpg

This prints a list which contains the probabilities for image being cat or dog.

#[ prob of cat , prob of dog]

You can’t perform that action at this time.