robertknight / ocrs-models

PyTorch models for the ocrs OCR engine
54 stars 8 forks source link

Increase weight of text instance border pixels #4

Closed robertknight closed 2 years ago

robertknight commented 2 years ago

Generate a mask of borders of text instances and use it to increase the weights of those pixels when computing the loss. This helps reduce how often mask areas corresponding to adjacent words get merged into a single connected component, especially across lines.