tf-sotd icon indicating copy to clipboard operation
tf-sotd copied to clipboard

about training on icdar2015 dataset

Open happog opened this issue 6 years ago • 1 comments

Hi, did you train sotd on icdar dataset? I have trained it on icdar2015 but the predicted map is all background class. Should I use another loss function such as cross entropy? Thank you!

happog avatar Feb 13 '19 04:02 happog

Sorry to I reply to so late, I didn't train it on icdar, the backgroud, border and text area are imbalance, so you need to add OHEM or other balanced loss function to train it, then it will work well

saicoco avatar May 15 '19 04:05 saicoco