liuch37 / pan-tensorflow Goto Github PK

This is an unofficial TensorFlow re-implementation of paper "Efficient and Accurate Arbitrary-Shaped Text Detection with Pixel Aggregation Network" published in ICCV 2019, with Tensorflow 2.

License: MIT License

Python 100.00%

pan-tensorflow's Introduction

Pixel Aggregation Network

This is an unofficial TensorFlow re-implementation of paper "Efficient and Accurate Arbitrary-Shaped Text Detection with Pixel Aggregation Network" published in ICCV 2019, with Tensorflow 2.

Task

Command

Training

python train.py --batch 16 --epoch 601 --dataset_type ctw --gpu True

Inference

python inference.py --input ./data/CTW1500/test/text_image --model ./outputs/model_epoch_600 --bbox_type poly

Results

*Default using ResNet-50 backbone model, also support ResNet-101 backbone model.

CTW1500

Model	Precision	Recall	F score
PAN-320	0.8472	0.6939	0.7629

Supported Dataset

CTW1500: https://github.com/Yuliang-Liu/Curve-Text-Detector

Source

[1] Original paper: https://arxiv.org/abs/1908.05900

[2] Official PyTorch code: https://github.com/whai362/pan_pp.pytorch

pan-tensorflow's People

Contributors

Stargazers

Watchers

pan-tensorflow's Issues

Error

I am getting this error

Traceback (most recent call last):
File "train.py", line 136, in
for iter, data in enumerate(train_dataloader):
File "/usr/local/lib/python3.7/dist-packages/tensorflow/python/keras/utils/data_utils.py", line 483, in iter
for item in (self[i] for i in range(len(self))):
File "/usr/local/lib/python3.7/dist-packages/tensorflow/python/keras/utils/data_utils.py", line 483, in
for item in (self[i] for i in range(len(self))):
File "/content/drive/My Drive/pan-tensorflow/dataset/ctw1500.py", line 367, in getitem
data = self.prepare_train_data_single(global_idx)
File "/content/drive/My Drive/pan-tensorflow/dataset/ctw1500.py", line 252, in prepare_train_data_single
bboxes, words = get_ann(img, gt_path)
File "/content/drive/My Drive/pan-tensorflow/dataset/ctw1500.py", line 36, in get_ann
with open(gt_path, "r") as file:
FileNotFoundError: [Errno 2] No such file or directory: './data/train_labels/0390.txt'

pre-trained model weights

Hi,

Thanks for this code implementation, it is pretty amazing. Can you please link the pre-trained model weights for ctw1500 if possible.
Would be really helpful

Thanks

Text not detected after testing

Not getting any result after testing. The images are saved in the results folder but there is no bounding box on the image i.e, it is not detected also the .txt file that is saved after testing is empty for each of the image.