Skip to content

Instantly share code, notes, and snippets.

@obeshor

obeshor/transfer_learning.py

Created June 25, 2019 22:34
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save obeshor/83675924348aeea819c66429d832e791 to your computer and use it in GitHub Desktop.
Save obeshor/83675924348aeea819c66429d832e791 to your computer and use it in GitHub Desktop.
Download ZIP
Tranfer Learning with TFHub
Raw
transfer_learning.py
module_selection = ("inception_v3", 299, 2048) #@param ["(\"mobilenet_v2\", 224, 1280)", "(\"inception_v3\", 299, 2048)"] {type:"raw", allow-input: true}
handle_base, pixels, FV_SIZE = module_selection
MODULE_HANDLE ="https://tfhub.dev/google/tf2- preview/{}/feature_vector/2".format(handle_base)
IMAGE_SIZE = (pixels, pixels)
BATCH_SIZE = 64 #@param {type:"integer"}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment