NinV / Korean-License-Plate-Recognition

Korean car license plate recognition using LPRNet
23 stars 3 forks source link

Train with Variable Length Labels #4

Open mesakarghm opened 3 years ago

mesakarghm commented 3 years ago

I'm trying to train the model with Variable length labels. While the model trains well with batch size 1, I run into an error when using batch size >= 2

Cannot batch tensors with different shapes in component 1. First element had shape [9] and element 1 had shape [10]. [Op:IteratorGetNextSync]

Amuqeet09 commented 3 years ago

You may add a space in labels, to make it equal or use label.ljust(10) in Loader.py.