search

clovaai / deep-text-recognition-benchmark

Text recognition (optical character recognition) with deep learning methods, ICCV 2019
Apache License 2.0
3.77k stars 1.11k forks source link

problem with training #274

Open yustiks opened 3 years ago

yustiks commented 3 years ago

I try to run training, but I have an issue:

Traceback (most recent call last):
  File "train.py", line 15, in <module>
    from deep_text_recognition_benchmark.utils import CTCLabelConverter, CTCLabelConverterForBaiduWarpctc, AttnLabelConverter, Averager
ModuleNotFoundError: No module named 'deep_text_recognition_benchmark'

I run the code:

CUDA_VISIBLE_DEVICES=0 python3 train.py --train_data training_data_21_03/  --Transformation TPS --valid_data valid_data_21_03/ --Transformation None --FeatureExtraction VGG --SequenceModeling BiLSTM --Prediction CTC --num_iter 500000 --batch_size 256
allenwu5 commented 3 years ago

Based on https://github.com/clovaai/deep-text-recognition-benchmark/blob/68a80fe97943a111ff1efaf52a63ad8f0f1c0e5d/train.py#L15

Just use from utils import CTCLabelConverter, CTCLabelConverterForBaiduWarpctc, AttnLabelConverter, Averager instead.