/Gluon-EAST

reimplement EAST by Gluon

Primary LanguageC++MIT LicenseMIT

gluon-east

reimplement EAST based gluoncv, which extends SegBaseModel

Install dependencies

pip install -r requirement.txt

pretrainmodels

you can set param backbone='resnet50'to get other backbone. Here are backbones in gluoncv:

if backbone == 'resnet50':
    pretrained = resnet50_v1s(pretrained=pretrained_base, dilated=True, **kwargs)
elif backbone == 'resnet101':
    pretrained = resnet101_v1s(pretrained=pretrained_base, dilated=True, **kwargs)
elif backbone == 'resnet152':
    pretrained = resnet152_v1s(pretrained=pretrained_base, dilated=True, **kwargs)

Train


python scripts/train_east.py data_dir ckpt_path

Test


python scripts/test_east.py

TODO

  • train and evaluation
  • Bugs exists on Regression branch