search

bear63 / sceneReco

ctpn+crnn Scene character recognition
943 stars 371 forks source link

Run demo.py and I got error in "conv = conv.permute(2, 0, 1)" statement. #34

Closed zhengszh closed 6 years ago

zhengszh commented 6 years ago

I have commented out the "conv = conv.squeeze(2)" statement and I got error from the next statement. The error is:

conv = conv.permute(2, 0, 1)  # [w, b, c]
RuntimeError: number of dims don't match in permute