File size: 1,220 Bytes
d5382b2 d44d8fc 48fcf1e d44d8fc 48fcf1e d44d8fc 48fcf1e d44d8fc 48fcf1e d44d8fc 48fcf1e d44d8fc 48fcf1e d44d8fc 48fcf1e d44d8fc 48fcf1e d44d8fc 280b928 d44d8fc 280b928 d44d8fc 280b928 d44d8fc d5382b2 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 | ---
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
|