nondefaced_detector.models.model.ClassifierHead

ClassifierHead(layer, dropout)

The final block of the model

Parameters

• layer (N-D tensor with shape: (batch_size, .., input_dim)) – The flattened out feature layer output from the Submodels

• dropout (float) – Float between 0 and 1. Fraction of the input units to drop.

Returns

N-D tensor with shape: (batch_size, …, units)

Return type

tf.Tensor