Skip to main content

postprocess

Function postprocess 

Source
fn postprocess(logits: &[f32], pre: &Preprocessed) -> Vec<u8> 
Expand description

Resample logits back to the cropped grid, argmax, un-crop. Returns the column-major mask.