You can deploy this model and start earning money today!

Jwalit/
kyc-document-corner-detector

Name: kyc-document-corner-detector
Rating: 2.5 (15 reviews)

pytorch

ml-intern

Views

No views yet

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.

Generated by ML Intern

This model repository was generated by 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

Jwalit/kyc-document-corner-detector