robertknight / ocrs-models

PyTorch models for the ocrs OCR engine
31 stars 5 forks source link

Make debug image generation optional, add progress during testing stage #1

Closed robertknight closed 2 years ago

robertknight commented 2 years ago

This branch started out as an investigation into the possibility of adding a second output mask which predicted the bottom of text words, to help with separately closely spaced lines in the text region mask. However I don't think that approach is going to be the best way forwards. Along the way I made some improvements to the training progress on this branch, which I want to land: