Skip to content

Instantly share code, notes, and snippets.

@zaleslaw
Created June 9, 2022 12:31
Show Gist options
  • Save zaleslaw/2c08442383372c10588c3bf4bc2ffbe6 to your computer and use it in GitHub Desktop.
Save zaleslaw/2c08442383372c10588c3bf4bc2ffbe6 to your computer and use it in GitHub Desktop.
Loads noTop model
val modelHub = TFModelHub(cacheDirectory = File("cache/pretrainedModels"))
val modelType = TFModels.CV.ResNet50(noTop = true, inputShape = intArrayOf(IMAGE_SIZE, IMAGE_SIZE, NUM_CHANNELS))
val noTopModel = modelHub.loadModel(modelType)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment