| tags: | |
| - ml-intern | |
| # kyc-document-corner-detector | |
| KYC Document Segmentation Model | MobileNetV3-Small | CPU Trained | |
| ## Details | |
| - **Task**: document_segmentation | |
| - **Backbone**: mobilenet_v3_small | |
| - **Input size**: 224px | |
| - **Epochs**: 8 | |
| - **Best metric**: 0.8262 IoU | |
| - **Dataset**: Jwalit/moire-docs | |
| - **Total images**: 2623 | |
| ## Training | |
| This model was trained on CPU using self-supervised labels: | |
| - **Segmentation**: OpenCV-generated document masks | |
| - **Rotation**: Synthetically rotated with known angles | |
| ## Usage | |
| ```python | |
| import torch | |
| from model import YourModelClass # See training script | |
| model = YourModelClass() | |
| model.load_state_dict(torch.load("pytorch_model.bin", map_location="cpu")) | |
| model.eval() | |
| ``` | |
| ## Dataset | |
| - Source: `Jwalit/moire-docs` | |
| - Contains KYC documents with clean and moire (scan artifacts) variants | |
| ## License | |
| Same as dataset license. | |
| <!-- ml-intern-provenance --> | |
| ## Generated by ML Intern | |
| This model repository was generated by [ML Intern](https://github.com/huggingface/ml-intern), an agent for machine learning research and development on the Hugging Face Hub. | |
| - Try ML Intern: https://smolagents-ml-intern.hf.space | |
| - Source code: https://github.com/huggingface/ml-intern | |