├── .gitignore ├── LICENSE ├── Makefile ├── README.md ├── assets ├── ctw_gt_01.jpg ├── ctw_gt_02.jpg ├── ctw_result_01.jpg ├── ctw_result_02.jpg ├── demo.gif ├── foo.jpg ├── ocr_01.jpg ├── ocr_02.jpg ├── ocr_03.jpg ├── test_history.png ├── test_img_history_01.png ├── test_img_history_02.png ├── train_history.png ├── tt_heatmap_01.jpg ├── tt_heatmap_02.jpg ├── tt_heatmap_03.jpg ├── tt_poly_01.jpg ├── tt_poly_02.jpg ├── tt_poly_03.jpg ├── tt_rect_01.jpg ├── tt_rect_02.jpg └── tt_rect_03.jpg ├── example_config.yaml ├── models └── db_resnet18.pth ├── requirements.txt └── src ├── __init__.py ├── data_loaders.py ├── db_handler.py ├── db_transforms.py ├── deteval.py ├── iou.py ├── losses.py ├── lr_schedulers.py ├── make_eval.py ├── models.py ├── modules ├── __init__.py ├── basic.py ├── resnet.py ├── segmentation_body.py └── segmentation_head.py ├── postprocess.py ├── save_jit.py ├── test.py ├── test_ocr.py ├── test_webcam.py ├── text_metrics.py ├── train.py ├── ts_request.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huyhoang17/DB_text_minimal/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huyhoang17/DB_text_minimal/HEAD/LICENSE -------------------------------------------------------------------------------- /Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huyhoang17/DB_text_minimal/HEAD/Makefile -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huyhoang17/DB_text_minimal/HEAD/README.md -------------------------------------------------------------------------------- /assets/ctw_gt_01.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huyhoang17/DB_text_minimal/HEAD/assets/ctw_gt_01.jpg -------------------------------------------------------------------------------- /assets/ctw_gt_02.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huyhoang17/DB_text_minimal/HEAD/assets/ctw_gt_02.jpg -------------------------------------------------------------------------------- /assets/ctw_result_01.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huyhoang17/DB_text_minimal/HEAD/assets/ctw_result_01.jpg -------------------------------------------------------------------------------- /assets/ctw_result_02.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huyhoang17/DB_text_minimal/HEAD/assets/ctw_result_02.jpg -------------------------------------------------------------------------------- /assets/demo.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huyhoang17/DB_text_minimal/HEAD/assets/demo.gif -------------------------------------------------------------------------------- /assets/foo.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huyhoang17/DB_text_minimal/HEAD/assets/foo.jpg -------------------------------------------------------------------------------- /assets/ocr_01.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huyhoang17/DB_text_minimal/HEAD/assets/ocr_01.jpg -------------------------------------------------------------------------------- /assets/ocr_02.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huyhoang17/DB_text_minimal/HEAD/assets/ocr_02.jpg -------------------------------------------------------------------------------- /assets/ocr_03.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huyhoang17/DB_text_minimal/HEAD/assets/ocr_03.jpg -------------------------------------------------------------------------------- /assets/test_history.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huyhoang17/DB_text_minimal/HEAD/assets/test_history.png -------------------------------------------------------------------------------- /assets/test_img_history_01.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huyhoang17/DB_text_minimal/HEAD/assets/test_img_history_01.png -------------------------------------------------------------------------------- /assets/test_img_history_02.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huyhoang17/DB_text_minimal/HEAD/assets/test_img_history_02.png -------------------------------------------------------------------------------- /assets/train_history.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huyhoang17/DB_text_minimal/HEAD/assets/train_history.png -------------------------------------------------------------------------------- /assets/tt_heatmap_01.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huyhoang17/DB_text_minimal/HEAD/assets/tt_heatmap_01.jpg -------------------------------------------------------------------------------- /assets/tt_heatmap_02.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huyhoang17/DB_text_minimal/HEAD/assets/tt_heatmap_02.jpg -------------------------------------------------------------------------------- /assets/tt_heatmap_03.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huyhoang17/DB_text_minimal/HEAD/assets/tt_heatmap_03.jpg -------------------------------------------------------------------------------- /assets/tt_poly_01.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huyhoang17/DB_text_minimal/HEAD/assets/tt_poly_01.jpg -------------------------------------------------------------------------------- /assets/tt_poly_02.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huyhoang17/DB_text_minimal/HEAD/assets/tt_poly_02.jpg -------------------------------------------------------------------------------- /assets/tt_poly_03.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huyhoang17/DB_text_minimal/HEAD/assets/tt_poly_03.jpg -------------------------------------------------------------------------------- /assets/tt_rect_01.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huyhoang17/DB_text_minimal/HEAD/assets/tt_rect_01.jpg -------------------------------------------------------------------------------- /assets/tt_rect_02.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huyhoang17/DB_text_minimal/HEAD/assets/tt_rect_02.jpg -------------------------------------------------------------------------------- /assets/tt_rect_03.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huyhoang17/DB_text_minimal/HEAD/assets/tt_rect_03.jpg -------------------------------------------------------------------------------- /example_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huyhoang17/DB_text_minimal/HEAD/example_config.yaml -------------------------------------------------------------------------------- /models/db_resnet18.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huyhoang17/DB_text_minimal/HEAD/models/db_resnet18.pth -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huyhoang17/DB_text_minimal/HEAD/requirements.txt -------------------------------------------------------------------------------- /src/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/data_loaders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huyhoang17/DB_text_minimal/HEAD/src/data_loaders.py -------------------------------------------------------------------------------- /src/db_handler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huyhoang17/DB_text_minimal/HEAD/src/db_handler.py -------------------------------------------------------------------------------- /src/db_transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huyhoang17/DB_text_minimal/HEAD/src/db_transforms.py -------------------------------------------------------------------------------- /src/deteval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huyhoang17/DB_text_minimal/HEAD/src/deteval.py -------------------------------------------------------------------------------- /src/iou.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huyhoang17/DB_text_minimal/HEAD/src/iou.py -------------------------------------------------------------------------------- /src/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huyhoang17/DB_text_minimal/HEAD/src/losses.py -------------------------------------------------------------------------------- /src/lr_schedulers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huyhoang17/DB_text_minimal/HEAD/src/lr_schedulers.py -------------------------------------------------------------------------------- /src/make_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huyhoang17/DB_text_minimal/HEAD/src/make_eval.py -------------------------------------------------------------------------------- /src/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huyhoang17/DB_text_minimal/HEAD/src/models.py -------------------------------------------------------------------------------- /src/modules/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/modules/basic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huyhoang17/DB_text_minimal/HEAD/src/modules/basic.py -------------------------------------------------------------------------------- /src/modules/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huyhoang17/DB_text_minimal/HEAD/src/modules/resnet.py -------------------------------------------------------------------------------- /src/modules/segmentation_body.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huyhoang17/DB_text_minimal/HEAD/src/modules/segmentation_body.py -------------------------------------------------------------------------------- /src/modules/segmentation_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huyhoang17/DB_text_minimal/HEAD/src/modules/segmentation_head.py -------------------------------------------------------------------------------- /src/postprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huyhoang17/DB_text_minimal/HEAD/src/postprocess.py -------------------------------------------------------------------------------- /src/save_jit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huyhoang17/DB_text_minimal/HEAD/src/save_jit.py -------------------------------------------------------------------------------- /src/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huyhoang17/DB_text_minimal/HEAD/src/test.py -------------------------------------------------------------------------------- /src/test_ocr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huyhoang17/DB_text_minimal/HEAD/src/test_ocr.py -------------------------------------------------------------------------------- /src/test_webcam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huyhoang17/DB_text_minimal/HEAD/src/test_webcam.py -------------------------------------------------------------------------------- /src/text_metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huyhoang17/DB_text_minimal/HEAD/src/text_metrics.py -------------------------------------------------------------------------------- /src/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huyhoang17/DB_text_minimal/HEAD/src/train.py -------------------------------------------------------------------------------- /src/ts_request.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huyhoang17/DB_text_minimal/HEAD/src/ts_request.py -------------------------------------------------------------------------------- /src/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huyhoang17/DB_text_minimal/HEAD/src/utils.py --------------------------------------------------------------------------------