Skip to content

Latest commit

 

History

History
15 lines (11 loc) · 1.08 KB

File metadata and controls

15 lines (11 loc) · 1.08 KB

This directory contains two examples of transfer learning using the "Inception V3" image classification model.

The cloudml example shows how to use Cloud Dataflow (Apache Beam) to do image preprocessing, then train and serve your model on Cloud ML. It supports distributed training on Cloud ML. It is based on the example here, with some additional modifications to make it easy to use other image sets, and a prediction web server that demos how to use the Cloud ML API for prediction once your trained model is serving.

The TF_Estimator example takes a similar approach, but is not packaged to run on Cloud ML. It also shows an example of using a custom Estimator.

The list of image sources for the images used in the "hugs/no-hugs" training is here: https://storage.googleapis.com/oscon-tf-workshop-materials/transfer_learning/hugs_photos_sources.csv