Open SriramPingali opened 4 years ago
Hi, i have already writen a demo.py
for you to test and train.py
for you to train.
Hey @holmeyoung.. I am using the train.py you've written. Although, I wanted to train the model on a custom dataset. Therfore I did the above changes for the dataset class. It is working as expected but the training isn't giving fruitful results.
Hi, i know you are training it on a custom dataset, but you'd better make you data into lmdb and train with lmdb. My code is based on it. There is a create_dataset.py
in tool
folder.
Thanks for the Code!
I was using the model for a custom dataset (IAM Dataset for Hand written texts), and therefore wrote a custom Dataset class for train_loader which gives a 32 128 3 (H W D) image and a string. I also used a custom function for doing one hot encoding for the labels.
Although I notice that the output from the CRNN model is always a tensor full of zeroes (the PAD character from my letter dictionary)
Please Help me out here! Thanks in advance!! `class dataset(Dataset):
`def word_rep(word, letter2index, max_out_chars, device = 'cpu'):