Ecosyste.ms: Awesome
An open API service indexing awesome lists of open source software.
https://github.com/ZJULearning/pixel_link
Implementation of our paper 'PixelLink: Detecting Scene Text via Instance Segmentation' in AAAI2018
https://github.com/ZJULearning/pixel_link
Last synced: about 2 months ago
JSON representation
Implementation of our paper 'PixelLink: Detecting Scene Text via Instance Segmentation' in AAAI2018
- Host: GitHub
- URL: https://github.com/ZJULearning/pixel_link
- Owner: ZJULearning
- License: mit
- Created: 2018-03-26T09:38:54.000Z (over 6 years ago)
- Default Branch: master
- Last Pushed: 2019-11-21T02:17:11.000Z (almost 5 years ago)
- Last Synced: 2024-07-04T18:32:00.931Z (3 months ago)
- Language: Python
- Homepage:
- Size: 421 KB
- Stars: 766
- Watchers: 33
- Forks: 254
- Open Issues: 111
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
README
Code for the AAAI18 paper [PixelLink: Detecting Scene Text via Instance Segmentation](https://arxiv.org/abs/1801.01315), by Dan Deng, Haifeng Liu, Xuelong Li, and Deng Cai.
Contributions to this repo are welcome, e.g., some other backbone networks (including the model definition and pretrained models).
PLEASE CHECK EXSITING ISSUES BEFORE OPENNING YOUR OWN ONE. IF A SAME OR SIMILAR ISSUE HAD BEEN POSTED BEFORE, JUST REFER TO IT, AND DO NO OPEN A NEW ONE.
# Installation
## Clone the repo
```
git clone --recursive [email protected]:ZJULearning/pixel_link.git
```
Denote the root directory path of pixel_link by `${pixel_link_root}`.
Add the path of `${pixel_link_root}/pylib/src` to your `PYTHONPATH`:
```
export PYTHONPATH=${pixel_link_root}/pylib/src:$PYTHONPATH
```
## Prerequisites
(Only tested on) Ubuntu14.04 and 16.04 with:
* Python 2.7
* Tensorflow-gpu >= 1.1
* opencv2
* setproctitle
* matplotlib
Anaconda is recommended to for an easier installation:
1. Install [Anaconda](https://anaconda.org/)
2. Create and activate the required virtual environment by:
```
conda env create --file pixel_link_env.txt
source activate pixel_link
```
# Testing
## Download the pretrained model
* PixelLink + VGG16 4s [Baidu Netdisk](https://pan.baidu.com/s/1jsOc-cutC4GyF-wMMyj5-w) | [GoogleDrive](https://drive.google.com/file/d/19mlX5W8OBalSjhf5oTTS6qEq2eAU8Tg9/view?usp=sharing), trained on IC15
* PixelLink + VGG16 2s [Baidu Netdisk](https://pan.baidu.com/s/1asSFsRSgviU2GnvGt2lAUw) | [GoogleDrive](https://drive.google.com/file/d/1QleZxu_6PSI733G7wzbqeFtc8A3-LmWW/view?usp=sharing), trained on IC15
Unzip the downloaded model. It contains 4 files:
* config.py
* model.ckpt-xxx.data-00000-of-00001
* model.ckpt-xxx.index
* model.ckpt-xxx.meta
Denote their parent directory as `${model_path}`.
## Test on ICDAR2015
The reported results on ICDAR2015 are:
|Model|Recall|Precision|F-mean|
|---|---|---|---|
|PixelLink+VGG16 2s|82.0|85.5|83.7|
|PixelLink+VGG16 4s|81.7|82.9|82.3|
Suppose you have downloaded the [ICDAR2015 dataset](http://rrc.cvc.uab.es/?ch=4&com=downloads), execute the following commands to test the model on ICDAR2015:
```
cd ${pixel_link_root}
./scripts/test.sh ${GPU_ID} ${model_path}/model.ckpt-xxx ${path_to_icdar2015}/ch4_test_images
```
For example:
```
./scripts/test.sh 3 ~/temp/conv3_3/model.ckpt-38055 ~/dataset/ICDAR2015/Challenge4/ch4_test_images
```
The program will create a zip file of detection results, which can be submitted to the ICDAR2015 server directly.
The detection results can be visualized via `scripts/vis.sh`.
Here are some samples:
## Test on any images
Put the images to be tested in a single directory, i.e., `${image_dir}`. Then:
```
cd ${pixel_link_root}
./scripts/test_any.sh ${GPU_ID} ${model_path}/model.ckpt-xxx ${image_dir}
```
For example:
```
./scripts/test_any.sh 3 ~/temp/conv3_3/model.ckpt-38055 ~/dataset/ICDAR2015/Challenge4/ch4_training_images
```
The program will visualize the detection results directly on images. If the detection result is not satisfying, try to:
1. Adjust the inference parameters like `eval_image_width`, `eval_image_height`, `pixel_conf_threshold`, `link_conf_threshold`.
2. Or train your own model.
# Training
## Converting the dataset to tfrecords files
Scripts for converting ICDAR2015 and SynthText datasets have been provided in the `datasets` directory.
It not hard to write a converting script for your own dataset.
## Train your own model
* Modify `scripts/train.sh` to configure your dataset name and dataset path like:
```
DATASET=icdar2015
DATASET_DIR=$HOME/dataset/pixel_link/icdar2015
```
* Start training
```
./scripts/train.sh ${GPU_IDs} ${IMG_PER_GPU}
```
For example, `./scripts/train.sh 0,1,2 8`.
The existing training strategy in `scripts/train.sh` is configured for icdar2015, modify it if necessary. A lot of training or model options are available in `config.py`, try it yourself if you are interested.
# Acknowlegement
