├── .gitattributes ├── .gitignore ├── README.md ├── assets ├── dice.png ├── iou.png ├── iou_formular1.png ├── pipeline.png └── unet.jpg ├── dataset ├── .gitkeep └── data │ └── .gitkeep ├── models.py ├── prepare_dataset.py ├── pretrained └── README.md ├── requirements.txt ├── test.py ├── test ├── sample1.png ├── sample2.png ├── sample3.png └── sample4.png ├── train.py └── utils ├── __init__.py ├── haarcascade_frontalface_default.xml ├── image.py └── metrics.py /.gitattributes: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tobiassteidle/ML_IDCard_Segmentation_Pytorch/HEAD/.gitattributes -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tobiassteidle/ML_IDCard_Segmentation_Pytorch/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tobiassteidle/ML_IDCard_Segmentation_Pytorch/HEAD/README.md -------------------------------------------------------------------------------- /assets/dice.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tobiassteidle/ML_IDCard_Segmentation_Pytorch/HEAD/assets/dice.png -------------------------------------------------------------------------------- /assets/iou.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tobiassteidle/ML_IDCard_Segmentation_Pytorch/HEAD/assets/iou.png -------------------------------------------------------------------------------- /assets/iou_formular1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tobiassteidle/ML_IDCard_Segmentation_Pytorch/HEAD/assets/iou_formular1.png -------------------------------------------------------------------------------- /assets/pipeline.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tobiassteidle/ML_IDCard_Segmentation_Pytorch/HEAD/assets/pipeline.png -------------------------------------------------------------------------------- /assets/unet.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tobiassteidle/ML_IDCard_Segmentation_Pytorch/HEAD/assets/unet.jpg -------------------------------------------------------------------------------- /dataset/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dataset/data/.gitkeep: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tobiassteidle/ML_IDCard_Segmentation_Pytorch/HEAD/models.py -------------------------------------------------------------------------------- /prepare_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tobiassteidle/ML_IDCard_Segmentation_Pytorch/HEAD/prepare_dataset.py -------------------------------------------------------------------------------- /pretrained/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tobiassteidle/ML_IDCard_Segmentation_Pytorch/HEAD/pretrained/README.md -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tobiassteidle/ML_IDCard_Segmentation_Pytorch/HEAD/requirements.txt -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tobiassteidle/ML_IDCard_Segmentation_Pytorch/HEAD/test.py -------------------------------------------------------------------------------- /test/sample1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tobiassteidle/ML_IDCard_Segmentation_Pytorch/HEAD/test/sample1.png -------------------------------------------------------------------------------- /test/sample2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tobiassteidle/ML_IDCard_Segmentation_Pytorch/HEAD/test/sample2.png -------------------------------------------------------------------------------- /test/sample3.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tobiassteidle/ML_IDCard_Segmentation_Pytorch/HEAD/test/sample3.png -------------------------------------------------------------------------------- /test/sample4.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tobiassteidle/ML_IDCard_Segmentation_Pytorch/HEAD/test/sample4.png -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tobiassteidle/ML_IDCard_Segmentation_Pytorch/HEAD/train.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/haarcascade_frontalface_default.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tobiassteidle/ML_IDCard_Segmentation_Pytorch/HEAD/utils/haarcascade_frontalface_default.xml -------------------------------------------------------------------------------- /utils/image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tobiassteidle/ML_IDCard_Segmentation_Pytorch/HEAD/utils/image.py -------------------------------------------------------------------------------- /utils/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tobiassteidle/ML_IDCard_Segmentation_Pytorch/HEAD/utils/metrics.py --------------------------------------------------------------------------------