search

xiaofengShi / CHINESE-OCR

[python3.6] 运用tf实现自然场景文字检测,keras/pytorch实现ctpn+crnn+ctc实现不定长场景文字OCR识别
2.9k stars 957 forks source link

is pytorch needed to run demo.py? #36

Open ha121ppy opened 5 years ago

ha121ppy commented 5 years ago

Dear @xiaofengShi , I try to run demo.py and an error output: No module named 'torch'. Because ./crnn/crnn.py need to import torch. Is Pytorch a must to run demo.py? As to my understanding, either Keras or Pytorch is OK to finish this. Look forward to you reply. Thanks!

dakun333 commented 5 years ago

You can try this: conda install -c peterjc123 pytorch cuda90