Skip to content

Latest commit

 

History

History
executable file
·
45 lines (25 loc) · 1.14 KB

README.md

File metadata and controls

executable file
·
45 lines (25 loc) · 1.14 KB

DeText

A computer vision project which aims to remove text on images

It is a ready-to-run code after downloading data from here .

Dependencies

pip3 install -r requirements.txt

Training

Open up a terminal in the main directory and run one of below commands.

If you want to train from scratch:

bash run_from_scratch.sh

If you don't want to train from scratch:

bash run_from_pretrained.sh

Notebook

Notebooks are in notebooks folder.

Data

CelebFaces Attributes (CelebA) Dataset is used in the project.

This is a repository containing more than 200,000 images of various celebrities.

Download the dataset and put the downloaded zip file in main directory. Then, unzip it.

The directories and files should be listed as below:

alt text

Architecture

UNet is used as architecture. I customized it a little. About 2 million Parameters. alt text

Predictions

Predictions for some testing images are below.

alt text