Skip to main content

class_output_shape

Function class_output_shape 

Source
pub fn class_output_shape(
    mode: ClassifierTrainingMode,
    field: [u32; 3],
    class_count: u32,
) -> Result<[u32; 3], String>
Expand description

Class output geometry for one training mode.

Kernel mode is 1×1×class_count, the same shape as the class input. Scanner mode is field_w × field_h × 1.