NinV / Korean-License-Plate-Recognition

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

Empty vector output resultant of training using given training data #5

Closed Amuqeet09 closed 3 years ago

Amuqeet09 commented 3 years ago

Thank you for the code.

I successfully run your all codes. But when I train it with your given data, it returns empty vector. What could be the possible problem?

Is it possible to share KarPlate dataset? If possible, please share about the number of training/testing data.

jpsalada commented 3 years ago

same issue here

NinV commented 3 years ago

I think the problem is with the default training parameters. You can experiment with different learning rates, batch size, and the number of epochs. The training loss should be low enough for the model to show some acceptable results. You can use the code in "evaluate.py" file to check the loss value of the pre-trained model and use it as a reference for tuning training parameters.

jpsalada commented 3 years ago

@NinV thank you so much for replying. I will try that. Would it be possible to share the training configuration parameters that you have used to create your model?

NinV commented 3 years ago

The pretrained model is trained using this command:

python train.py -l data/label.json -i data/train_images --valid_label data/test.json --valid_img_dir data/test_images --save_weights_only --load_all --learning_rate 0.001 --decay_steps 10000 --decay_rate 0.995 --num_steps 100000 --batch_size 32

Amuqeet09 commented 3 years ago

I came across that network was not fully trained that was causing resultant empty vector.