argman / EAST

A tensorflow implementation of EAST text detector
GNU General Public License v3.0
3.02k stars 1.05k forks source link

about the classification loss. #101

Open GodOfSmallThings opened 6 years ago

GodOfSmallThings commented 6 years ago

The original model used balanced-cross entropy loss. However, your model didn't use it and performs well. I guess it's because your code does some kind of hard negative mining in the preprocessing stage and already solves class imbalance problem before network stage. Would it be correct guess?

weinman commented 6 years ago

Not really; the batch sampling code in icdar.py rejects most random patches that contain no text rectangles.