MichalBusta / E2E-MLT

E2E-MLT - an Unconstrained End-to-End Method for Multi-Language Scene Text
MIT License
290 stars 84 forks source link

All loss values become nan #78

Open AniketGurav opened 2 years ago

AniketGurav commented 2 years ago

Respected Sir,

Thanks for your work. I am working on images that have only horizontal text.

During training following loss values eventually become nan after few steps. Those loss are bbox_loss,seg_loss,angle_loss and loss These values are present in train.py between lines 487 to 497

Can you please let me know what can be the issue and is there any change required in code when only horizontal text is present.

--- Aniket

MichalBusta commented 2 years ago

Hi, The horizontal and vertical text are supported.

you can run training with a debug option to see, what are you feeding in ...

the data format description: https://github.com/MichalBusta/E2E-MLT/tree/master/sample_train_data/MLT

Hope it helps, Michal

AniketGurav commented 2 years ago

Thank you