This is a pytorch re-implementation of FOTS: Fast Oriented Text Spotting with a Unified Network. The features are summarized blow:
- Only detection part is implemented.
- Any version of torch version >= 0.3.1 should be ok.
- Models trained on ICDAR 2015 (training set) + ICDAR 2017 (training set)
If you want to train the model, you should provide the dataset path, in the dataset path, a separate gt text file should be provided for each image and run
python main_train.py
run
python eval.py
a text file will be then written to the output path.