← all repositories

hustvl/YOLOS

A vision transformer model adapted for object detection without task-specific architectural modifications, published at NeurIPS 2021.

904 stars Jupyter Notebook Computer VisionInference · Serving
YOLOS
Not currently ranked — collecting fresh signals.
star history

YOLOS demonstrates that vanilla Vision Transformers pre-trained on image classification can transfer to object detection by adding detection tokens and using a set-based Hungarian matching loss. The project studies the transferability of ImageNet-pretrained ViTs to the COCO detection benchmark, including experiments with self-supervised MoCo-v3 pre-training. The implementation is integrated into HuggingFace Transformers for easy use.

Frequently asked

What is hustvl/YOLOS?
A vision transformer model adapted for object detection without task-specific architectural modifications, published at NeurIPS 2021.
Is YOLOS open source?
Yes — hustvl/YOLOS is open source, released under the MIT license.
What language is YOLOS written in?
hustvl/YOLOS is primarily written in Jupyter Notebook.
How popular is YOLOS?
hustvl/YOLOS has 904 stars on GitHub.
Where can I find YOLOS?
hustvl/YOLOS is on GitHub at https://github.com/hustvl/YOLOS.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.