FakerYFX / FOTS

An Implementation of the FOTS: Fast Oriented Text Spotting with a Unified Network
173 stars 41 forks source link

FOTS: Fast Oriented Text Spotting with a Unified Network

Introduction

This is a pytorch re-implementation of FOTS: Fast Oriented Text Spotting with a Unified Network. The features are summarized blow:

Contents

  1. Installation
  2. Download
  3. Train
  4. Test

Installation

  1. Any version of torch version >= 0.3.1 should be ok.

Download

  1. Models trained on ICDAR 2015 (training set) + ICDAR 2017 (training set)

Train

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

Test

run

python eval.py

a text file will be then written to the output path.