/PAN.pytorch

A unofficial pytorch implementation of PAN(PSENet2): Efficient and Accurate Arbitrary-Shaped Text Detection with Pixel Aggregation Network

Primary LanguagePythonMIT LicenseMIT

Efficient and Accurate Arbitrary-Shaped Text Detection with Pixel Aggregation Network

Requirements

  • pytorch 1.1
  • torchvision 0.3
  • pyclipper
  • opencv3

Data Preparation

prepare a text in the following format, use '\t' as a separator

/path/to/img.jpg path/to/label.txt
...

Train

  1. config the train_data_path,val_data_pathin config.json
  2. use following script to run
python3 train.py

Test

TBD

Predict

TBD

The project is still under development.

Performance

Method Precision (%) Recall (%) F-measure (%) FPS(1080Ti)
paper(640*640) 84.0 81.9 82.9 26.1
my implementation TBD TBD TBD TBD

TBD

examples

TBD

reference

  1. https://arxiv.org/pdf/1908.05900.pdf
  2. https://github.com/WenmuZhou/PSENet.pytorch