Loading…
Loading…
YOLOS model finetuned to detect handwritten signatures in document images using [tech4humans/signature-detection](https://huggingface.co/datasets/tech4humans/signature-detection) dataset.
Original YOLOS was introduced in the paper You Only Look at One Sequence: Rethinking Transformer in Vision through Object Detection by Fang et al. and first released in this repository.
YOLOS is a Vision Transformer (ViT) trained using the DETR loss. Despite its simplicity, a base-sized YOLOS model is able to achieve 42 AP on COCO validation 2017 (similar to DETR and more complex frameworks such as Faster R-CNN).
This model is designed for detecting handwritten signatures in scanned documents, contracts, or forms.
You can try it instantly in your browser here:
Here is how to use this model:
from datasets import load_dataset
from transformers import pipeline
# Load the tech4humans signature dataset
dataset = load_dataset("samuellimabraz/signature-detection")
# Load the finetuned model
yolos = pipeline(
task="object-detection",
model="mdefrance/yolos-tiny-signature-detection",
device_map="auto",
)
# Inference on test sample
prediction = yolos(dataset["test"][0].get("image"))
Currently, both the image processor and model support PyTorch.
The training utilized a dataset built from two public datasets: Tobacco800 and signatures-xc8up, unified and processed in Roboflow. The processed dataset was created by Samuel Lima Braz, and all credit for the dataset preparation goes to him.
Dataset Summary:
See mdefrance/signature-detection for details on training procedure.
Performances computed on the testing set:
| Metric | yolos-base-signature-detection | yolos-small-signature-detection | yolos-tiny-signature-detection |
|---|---|---|---|
| Inference Time - CPU (s) | 2.250 | 0.787 | 0.262 |
| Inference Time - GPU (s) | 1.464 | 0.023 | 0.014 |
| Parameters | 127.73M | 30.65M | 6.47M |
| mAP50 | 0.887 | 0.859 | 0.856 |
| mAP50-95 | 0.495 | 0.419 | 0.395 |
Inference times are computed on a laptop with following specs:
AGPL-3.0 is a strong copyleft license designed to keep software and its modifications open-source, especially for web apps and network services.
Apache 2.0 is a permissive license that offers flexibility for both open-source and proprietary use.
While AGPL-3.0 ensures openness, Apache 2.0 is better suited for businesses and closed-source use:
For full license texts, see:
This model is a finetuned version of the YOLOS model introduced in the following paper. If you use this model, please cite the original work:
BibTeX:
@article{DBLP:journals/corr/abs-2106-00666,
author = {Yuxin Fang and
Bencheng Liao and
Xinggang Wang and
Jiemin Fang and
Jiyang Qi and
Rui Wu and
Jianwei Niu and
Wenyu Liu},
title = {You Only Look at One Sequence: Rethinking Transformer in Vision through
Object Detection},
journal = {CoRR},
volume = {abs/2106.00666},
year = {2021},
url = {https://arxiv.org/abs/2106.00666},
eprinttype = {arXiv},
eprint = {2106.00666},
timestamp = {Fri, 29 Apr 2022 19:49:16 +0200},
biburl = {https://dblp.org/rec/journals/corr/abs-2106-00666.bib},
bibsource = {dblp computer science bibliography, https://dblp.org}
}
This model is finetuned for handwritten signature detection using the [tech4humans/signature-detection](https://huggingface.co/datasets/tech4humans/signature-detection) Dataset. The finetuning process and additional resources can be found in the GitHub Repository mdefrance/signature-detection.
Mario DEFRANCEData Scientist / AI Engineer |
Responsibilities in this Project
|
hustvl/yolos-tiny