grabdoc is a deep learning based application to extract document from image. The application outputs a RGBA image in which the alpha mask is generated by a Semantic Segmenatation model. The model is based on MobileNets: Efficient Convolutional Neural Networks for Mobile Vision Applications.

The reposity includes a pretrained model and a flask based Web UI.

Python 3.6 and it depends on the following packages listed:

• Tensorflow==1.13.1
• opencv-python
• numpy
• flask
• pillow

You can install all those packages via pip.

pip install -r requirements 

If you use docker, you can install the environment from Dockerfile.

Please note the output image should be specified with a PNG extension

python grabdoc.py samples/ellie.jpg -o output.png

If you use docker, a Flask web server will be started by default after your container is running. Otherwise, you have to start the web server manually.

python flask/app.py

The port of Flask server is hardcoded to 8080. You can visit the home page via http://{remotehost}:8080/ after it started.

A simele GUI was developed to crop the document intelligently. (I did't upload as the desgin/code is too ugly).

demo.ipynb is used to generate the results of sample images.