---
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