File size: 683 Bytes
38ca0b7 88bc0f7 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 | ---
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.
|