PhiDCH/pan_pp.pytorch

Official implementations of PSENet, PAN and PAN++.

Install

pip install -r requirement.txt
./compile.sh

Dataset

See ICDAR2015 format.

pan_pp.pytorch
└── icdar2015
    │   train
    │   ├── img
    │   └── gt
    │   test
    │   ├── img
    │   └── gt

Download pretrain model

gdown --id 1dXMKs1VAltre6RCtDHYft-2q8oCtC_Yh -O checkpoint.pth.tar

Training

CUDA_VISIBLE_DEVICES=0,1,2,3 python train.py config/pan/pan_r18_ic15.py

See Colab.