river-segmentation / README.md
beaunix's picture
update readme
88bc0f7 verified
|
Raw
History Blame Contribute Delete
683 Bytes
metadata
license: cc-by-nc-4.0

Model loading for Aegis-River-Segmentation (SegFormer-B2, water body mask).

The underlying model was trained as a 7-class semantic segmentation model (background, water, sky, vegetation, building, vehicle, person), but only the 'water' clas s (id=1) is reliable in practice. This Space collapses the output to binary: water vs everything else.

IMPORTANT: transformers must be pinned to ==4.45.2. A later transformers version renamed an internal decode_head submodule (decode_head.linear_projections -> decode_head.linear_c), which silently breaks the correspondence between this checkpoint's keys and the model architecture on a mismatched version.