├── .gitattributes ├── .gitignore ├── README.md ├── generate_4_angles_by_model.py ├── generate_fixed_segmented_info.py ├── generate_segmented_info.py ├── images ├── pred_test.jpg ├── pred_test2.jpg ├── pred_test3.jpg ├── pred_test4.jpg ├── result_test.png ├── result_test2.png ├── result_test3.png ├── result_test4.png ├── test.jpg ├── test2.jpg ├── test3.jpg ├── test4.jpg ├── train_image.jpg └── train_label.png ├── labelmap.txt ├── models ├── __init__.py ├── common.py ├── experimental.py ├── export.py ├── hub │ ├── yolov3-spp.yaml │ ├── yolov5-fpn.yaml │ └── yolov5-panet.yaml ├── yolo.py ├── yolov5l.yaml ├── yolov5m.yaml ├── yolov5s.yaml └── yolov5x.yaml ├── remove_unlabelled_images.py ├── requirements.txt ├── segment_info.py ├── train-id-card-alignment.ipynb ├── train-info-segmentation.ipynb ├── utils ├── __init__.py ├── activations.py ├── datasets.py ├── evolve.sh ├── general.py ├── google_utils.py ├── object_recognition_common.py ├── ocr_common.py ├── segmentation_common.py └── torch_utils.py └── warp_identity_cards.py /.gitattributes: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tailtq/identity-card-info-extraction/HEAD/.gitattributes -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tailtq/identity-card-info-extraction/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tailtq/identity-card-info-extraction/HEAD/README.md -------------------------------------------------------------------------------- /generate_4_angles_by_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tailtq/identity-card-info-extraction/HEAD/generate_4_angles_by_model.py -------------------------------------------------------------------------------- /generate_fixed_segmented_info.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tailtq/identity-card-info-extraction/HEAD/generate_fixed_segmented_info.py -------------------------------------------------------------------------------- /generate_segmented_info.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tailtq/identity-card-info-extraction/HEAD/generate_segmented_info.py -------------------------------------------------------------------------------- /images/pred_test.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tailtq/identity-card-info-extraction/HEAD/images/pred_test.jpg -------------------------------------------------------------------------------- /images/pred_test2.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tailtq/identity-card-info-extraction/HEAD/images/pred_test2.jpg -------------------------------------------------------------------------------- /images/pred_test3.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tailtq/identity-card-info-extraction/HEAD/images/pred_test3.jpg -------------------------------------------------------------------------------- /images/pred_test4.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tailtq/identity-card-info-extraction/HEAD/images/pred_test4.jpg -------------------------------------------------------------------------------- /images/result_test.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tailtq/identity-card-info-extraction/HEAD/images/result_test.png -------------------------------------------------------------------------------- /images/result_test2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tailtq/identity-card-info-extraction/HEAD/images/result_test2.png -------------------------------------------------------------------------------- /images/result_test3.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tailtq/identity-card-info-extraction/HEAD/images/result_test3.png -------------------------------------------------------------------------------- /images/result_test4.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tailtq/identity-card-info-extraction/HEAD/images/result_test4.png -------------------------------------------------------------------------------- /images/test.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tailtq/identity-card-info-extraction/HEAD/images/test.jpg -------------------------------------------------------------------------------- /images/test2.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tailtq/identity-card-info-extraction/HEAD/images/test2.jpg -------------------------------------------------------------------------------- /images/test3.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tailtq/identity-card-info-extraction/HEAD/images/test3.jpg -------------------------------------------------------------------------------- /images/test4.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tailtq/identity-card-info-extraction/HEAD/images/test4.jpg -------------------------------------------------------------------------------- /images/train_image.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tailtq/identity-card-info-extraction/HEAD/images/train_image.jpg -------------------------------------------------------------------------------- /images/train_label.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tailtq/identity-card-info-extraction/HEAD/images/train_label.png -------------------------------------------------------------------------------- /labelmap.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tailtq/identity-card-info-extraction/HEAD/labelmap.txt -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tailtq/identity-card-info-extraction/HEAD/models/common.py -------------------------------------------------------------------------------- /models/experimental.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tailtq/identity-card-info-extraction/HEAD/models/experimental.py -------------------------------------------------------------------------------- /models/export.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tailtq/identity-card-info-extraction/HEAD/models/export.py -------------------------------------------------------------------------------- /models/hub/yolov3-spp.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tailtq/identity-card-info-extraction/HEAD/models/hub/yolov3-spp.yaml -------------------------------------------------------------------------------- /models/hub/yolov5-fpn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tailtq/identity-card-info-extraction/HEAD/models/hub/yolov5-fpn.yaml -------------------------------------------------------------------------------- /models/hub/yolov5-panet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tailtq/identity-card-info-extraction/HEAD/models/hub/yolov5-panet.yaml -------------------------------------------------------------------------------- /models/yolo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tailtq/identity-card-info-extraction/HEAD/models/yolo.py -------------------------------------------------------------------------------- /models/yolov5l.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tailtq/identity-card-info-extraction/HEAD/models/yolov5l.yaml -------------------------------------------------------------------------------- /models/yolov5m.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tailtq/identity-card-info-extraction/HEAD/models/yolov5m.yaml -------------------------------------------------------------------------------- /models/yolov5s.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tailtq/identity-card-info-extraction/HEAD/models/yolov5s.yaml -------------------------------------------------------------------------------- /models/yolov5x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tailtq/identity-card-info-extraction/HEAD/models/yolov5x.yaml -------------------------------------------------------------------------------- /remove_unlabelled_images.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tailtq/identity-card-info-extraction/HEAD/remove_unlabelled_images.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tailtq/identity-card-info-extraction/HEAD/requirements.txt -------------------------------------------------------------------------------- /segment_info.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tailtq/identity-card-info-extraction/HEAD/segment_info.py -------------------------------------------------------------------------------- /train-id-card-alignment.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tailtq/identity-card-info-extraction/HEAD/train-id-card-alignment.ipynb -------------------------------------------------------------------------------- /train-info-segmentation.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tailtq/identity-card-info-extraction/HEAD/train-info-segmentation.ipynb -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/activations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tailtq/identity-card-info-extraction/HEAD/utils/activations.py -------------------------------------------------------------------------------- /utils/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tailtq/identity-card-info-extraction/HEAD/utils/datasets.py -------------------------------------------------------------------------------- /utils/evolve.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tailtq/identity-card-info-extraction/HEAD/utils/evolve.sh -------------------------------------------------------------------------------- /utils/general.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tailtq/identity-card-info-extraction/HEAD/utils/general.py -------------------------------------------------------------------------------- /utils/google_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tailtq/identity-card-info-extraction/HEAD/utils/google_utils.py -------------------------------------------------------------------------------- /utils/object_recognition_common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tailtq/identity-card-info-extraction/HEAD/utils/object_recognition_common.py -------------------------------------------------------------------------------- /utils/ocr_common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tailtq/identity-card-info-extraction/HEAD/utils/ocr_common.py -------------------------------------------------------------------------------- /utils/segmentation_common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tailtq/identity-card-info-extraction/HEAD/utils/segmentation_common.py -------------------------------------------------------------------------------- /utils/torch_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tailtq/identity-card-info-extraction/HEAD/utils/torch_utils.py -------------------------------------------------------------------------------- /warp_identity_cards.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tailtq/identity-card-info-extraction/HEAD/warp_identity_cards.py --------------------------------------------------------------------------------