search

princewang1994 / TextSnake.pytorch

A PyTorch implementation of ECCV2018 Paper: TextSnake: A Flexible Representation for Detecting Text of Arbitrary Shapes
https://arxiv.org/abs/1807.01544
MIT License
437 stars 92 forks source link

Can't run demo.py #53

Open AmrutaAnalytics opened 3 years ago

AmrutaAnalytics commented 3 years ago

Trying to run pure inference part, Downgraded pytorch version to 1.5 (torch==1.5.0+cu101 torchvision==0.6.0+cu101), then Set CUDA_VISIBLE_DEVICES=0 python demo.py $EXPNAME --checkepoch 180 --img_root "path to image " , but getting following error in dataset/dataload.py error

princewang1994 commented 3 years ago

@AmrutaAnalytics Did you use your own dataset? Seems that your input image are gray-scale ones, If so, make it RGB format first.