Skip to main content

build_simple_cnn_classifier

Function build_simple_cnn_classifier 

Source
pub fn build_simple_cnn_classifier(
    model: &mut SequentialModel,
    graph: &mut Graph,
    input_channels: usize,
    num_classes: usize,
    stage_channels: &[usize],
    epsilon: f32,
) -> Result<(), ModelError>
Expand description

Builds a simple CNN classifier architecture for NHWC input.

Architecture: [Conv->BN->ReLU->MaxPool] x stages -> GlobalAvgPool -> Flatten -> Linear. This is a convenient builder for common CV classification tasks.