roatienza / deep-text-recognition-benchmark

PyTorch code of my ICDAR 2021 paper Vision Transformer for Fast and Efficient Scene Text Recognition (ViTSTR)
Apache License 2.0
284 stars 57 forks source link

Update train.py #42

Open casperthuis opened 3 months ago

casperthuis commented 3 months ago

The opt eval flag is currently not giving the desired behaviour. The flag gets reset for all dataset the same, resulting after the first evaluation in a disabling of the augmentation pipeline. This should make sure it behaves a expected, a more proper solution would be copy over all the specific attributes for the datasets classes.