junn.common.layers package

Submodules

junn.common.layers.run_model_layer module

Layer to run a model in a tiled manner.

class junn.common.layers.run_model_layer.RunModelTiled(*args: Any, **kwargs: Any)

Bases: tensorflow.python.keras.layers.

Run a model in a tiled manner.

With a fixed input size in a tiled manner over the (larger) input tensor (image).

call(raw_input_tensor, **kwargs)

Inner function called by Keras.

Parameters
  • raw_input_tensor

  • kwargs

Returns