Skip to content

Latest commit

 

History

History

image-classifier

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
 
 
 
 
 
 
 
 

To run this:

  • pip3 install -r requirements.txt
  • First, you need to train the model using python train.py
  • Edit the code in test.py for you optimal model weights in results folder ( currently does not because you need to train first ) and run:
    python test.py
    
    Output:
    10000/10000 [==============================] - 3s 331us/step
    Test accuracy: 81.17999999999999 %
    frog