MCPcopy Create free account
hub / github.com/bitbionic/keras-to-tensorflow / buildClassifier

Function buildClassifier

k2tf_trainer.py:49–108  ·  view source on GitHub ↗

Builds a very simple CNN outputing num_categories. Args: img_shape (int): The shape of the image to feed the CNN - defaults to 128 num_categories (int): The number of categories to feed the CNN Returns: keras.models.Model: a simple CNN

( img_shape=128, num_categories=5 )

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 1

k2tf_trainer.pyFile · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected