Skip to content

CS 5542 BigData Lab Report #10

Amy Lin edited this page Apr 6, 2017 · 17 revisions

TensorFlow PROGRAMMING - Retrain Inception Model Final Layer

[ QUESTION ]

Retrain Inception Model Final Layer for Image Dataset not covered in class. Report accuracy + Visualizations(Tensor Board) + Report Confusion Matrix for training and validation/testing.

[ IMPLEMENTATION ]

  • Use pre-trained Inception Model v3 to retrain different filter size of kernel in the pooling layer.
  • Remove the final layer and train a new one based on our own datasets ( Kansas City Fountains ).
  • Calculate bottleneck values for each image. --> To perform classification before the final output layer.
  • Training ( % of images used in current training ) + Validation Accuracy ( % of correct label images ) + Cross-entropy ( how well the learning process is ) are calculated after each 10 iterations.
  • The accuracy greatly enhanced compared to using only CNN model!

< DATASET >

3 Kansas City Fountains

[ RESULTS ]

Tensorflow Main Graph

Final Accuracy for 200 Steps : 92.3% !!!



Legends for the following graph



  • Accuracy


  • Retrain Accuracy


  • Activation




  • Cross-Entropy


  • Final Training Ops





Web-based Application ( Tensorflow API )

Connect Web application using Python and Tensorflow API

  • I used the gunicorn to activate label_image.py to connect with Tensorflow API. When clicking the predict button through the web page, predictions of the image content is run in python tensorflow. Images are the 3 fountains in Kansas City that are used in the group project. The web is giving me predictions of my fountains but somehow the images aren't displaying!.. I would need to clear what's going on with that.
Clone this wiki locally