Add Language Model Decoder (!274) · Merge requests · Automatic Text Recognition / DAN

Merged Solene Tarride requested to merge lm-decoder into main 1 year ago

Oct 13, 2023
- Fix broken images · fb6dee79
  Solene Tarride authored 1 year ago
  
  fb6dee79
Oct 12, 2023
- Fix documentation · d6d5d342
  Solene Tarride authored 1 year ago
  
  d6d5d342
- Fix lint · deef2057
  Solene Tarride authored 1 year ago
  
  deef2057
- Fix docstring again · 7279f1cc
  Solene Tarride authored 1 year ago
  
  7279f1cc
- Add details for the CTC decoder · d69453c8
  Solene Tarride authored 1 year ago
  
  d69453c8
- Fix docstring · 23ac0f44
  Solene Tarride authored 1 year ago
  
  23ac0f44
- Remove old documentation · f371551b
  Solene Tarride authored 1 year ago
  
  f371551b
- Improve documentation · 8270f98f
  Solene Tarride authored 1 year ago
  
  8270f98f
- Improve LMdecoder docstring · 50324298
  Mélodie Boillet authored 1 year ago and Solene Tarride committed 1 year ago
  
  50324298
- Apply suggestions · f53c60ee
  Mélodie Boillet authored 1 year ago and Solene Tarride committed 1 year ago
  
  f53c60ee
- Deal with unknown token separately · f053fba3
  Solene Tarride authored 1 year ago
  
  f053fba3
- Improve code · 56d9a0a2
  Solene Tarride authored 1 year ago
  
  56d9a0a2
- Improve code, typing and logging · b2831f9e
  Solene Tarride authored 1 year ago
  
  b2831f9e
- Document prediction command · f7563b0c
  Solene Tarride authored 1 year ago
  
  f7563b0c
- Document prediction with language model · 289c08ac
  Solene Tarride authored 1 year ago
  
  289c08ac
- Simplify and document data extraction · 14462a04
  Solene Tarride authored 1 year ago
  
  14462a04
- Fix tests · e13e86f3
  Solene Tarride authored 1 year ago
  
  e13e86f3
- Move tensor to correct device and trim prediction · 881ff379
  Solene Tarride authored 1 year ago
  
  881ff379
- Use CTCHypothesis.tokens instead og CTCHypothesis.words · eb0aee94
  Solene Tarride authored 1 year ago
  
  eb0aee94
- Fix CTC token probability · 4e3c0e5e
  Solene Tarride authored 1 year ago
  
  4e3c0e5e
- Write tests for LM decoding · 311a2375
  Solene Tarride authored 1 year ago
  
  311a2375
- Support batch_size>1 · 6cd6098b
  Solene Tarride authored 1 year ago
  
  6cd6098b
- Update tests for data extraction · ee36e1c1
  Solene Tarride authored 1 year ago
  
  ee36e1c1
- Write tests for data extraction · 72e183c3
  Solene Tarride authored 1 year ago
  
  72e183c3
- Use named tuple for special tokens · 204eb22f
  Solene Tarride authored 1 year ago
  
  204eb22f
- Remove debug code · 14db9eb9
  Solene Tarride authored 1 year ago
  
  14db9eb9
- Fix shape of tot_prob · 0831bc74
  Solene Tarride authored 1 year ago
  
  0831bc74
- Fix extraction lint · 8aff001c
  Solene Tarride authored 1 year ago
  
  8aff001c
- Update tests for data extraction · 396faddf
  Solene Tarride authored 1 year ago
  
  396faddf
- Generate LM files during data extraction · d419f377
  Solene Tarride authored 1 year ago
  
  d419f377
- Improve documentation · 93f5bad1
  Solene Tarride authored 1 year ago
  
  93f5bad1
- Add CTC frame between each frames · 6b72df21
  Solene Tarride authored 1 year ago
  
  6b72df21
- Fix tests · 1ea4df7d
  Solene Tarride authored 1 year ago
  
  1ea4df7d
- Implement CTCLanguageDecoder · 80a27658
  Solene Tarride authored 1 year ago
  
  80a27658

Add Language Model Decoder

Merge request reports

Activity