Holmeyoung / crnn-pytorch

Pytorch implementation of CRNN (CNN + RNN + CTCLoss) for all language OCR.
MIT License
378 stars 105 forks source link

create image Tensor #65

Open Hknight514880265 opened 4 years ago

Hknight514880265 commented 4 years ago

Hello, thanks for your great work. Hope you can answer my questions: when you create "image" Tensor,it's shape is [3, imgH, imgH] why not [3, imgH, imgW]? Inked1_LI

Thank you!

Hknight514880265 commented 4 years ago

details in train.py line 122.

image = torch.FloatTensor(params.batchSize, 3, params.imgH, params.imgH)

liulei199409 commented 3 years ago

[3, imgH, imgW] or [3, imgH, imgH], do you kown which is right?