Skip to main content

Module model

Module model 

Source
Expand description

Native RLX graphs for ocrs detection (U-Net) and recognition (CRNN + GRU).

Modules§

weights

Structs§

DetectionGraphConfig
RecognitionGraphConfig

Constants§

NUM_CLASSES

Functions§

build_detection_graph
build_recognition_after_g1_graph
Recognition graph ending after the first bidirectional GRU ([seq, batch, 512]).
build_recognition_after_g2_graph
Recognition graph ending after the second GRU ([seq, batch, 512]).
build_recognition_after_logits_graph
Recognition graph ending after the linear head ([seq, batch, classes] logits).
build_recognition_conv_graph
Conv stack only; output [seq, batch, 128] (GRU input layout).
build_recognition_graph
log_softmax_last_axis
RTen-compatible log-softmax on the last axis of a row-major [outer, classes] buffer.