saicoco / Gluon-EAST

reimplement EAST by Gluon

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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
ezoic increase your site revenue

About

reimplement EAST by Gluon

License:MIT License


Languages

Language:C++ 90.9%Language:Python 9.0%Language:Makefile 0.1%