Skip to content

Region recognition support#81

Merged
ankandrew merged 80 commits intomasterfrom
ankandrew/country-recognition
Mar 14, 2026
Merged

Region recognition support#81
ankandrew merged 80 commits intomasterfrom
ankandrew/country-recognition

Conversation

@ankandrew
Copy link
Copy Markdown
Owner

@ankandrew ankandrew commented Mar 14, 2026

New CCT v2 models with region recognition (supporting +65 countries). There are also bunch of other enhancements and fixes in this PR.

Model changes

  • Added empty plates (from backgrounds, noise, textures, etc.) to recognize all padded plates (empty plate).
  • Region has Unknown region trained mainly with synth data.
  • Changed gelu to silu due to export issues with some lib versions.
  • The v2 pre-trained models have seen almost x3 more data than previous v1 models.
  • Added new (and more correct) attention_layout, (that way each head doesn't use the full projection_dim)
  • In both pre-trained v2 models, the xs and s versions, have over 0.99 for val_region_macro_f1 on validation held split (containing more than 114_000 samples).

@ankandrew ankandrew merged commit 9ce7a5b into master Mar 14, 2026
6 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant