reimplement EAST based gluoncv, which extends SegBaseModel
pip install -r requirement.txt
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)
python scripts/train_east.py data_dir ckpt_path
python scripts/test_east.py
- train and evaluation
- Bugs exists on Regression branch