Skip to content

Latest commit

 

History

History
22 lines (22 loc) · 1.09 KB

File metadata and controls

22 lines (22 loc) · 1.09 KB

To run this:

  • pip3 install -r requirements.txt
  • To train the model (already trained and the optimal weights are in results folder):
    python train.py
    
    This will load the flower dataset, construct the MobileNetV2 model with its weights and starts training.
  • 86% accuracy was achieved on 5 classes of flowers which are daisy, dandelion, roses, sunflowers and tulips.
  • To evaluate the model as well as visualizing different flowers and its corresponding predictions:
    python test.py
    
    This will Outputs:
    23/23 [==============================] - 6s 264ms/step
    Val loss: 0.5659930361524
    Val Accuracy: 0.8166894659134987
    
    and plots: Predicted Flowers Check the tutorial for more information.