PyTorch Object Detection :: COCO JSON

YOLOS

YOLOS looks at patches of an image to to form "patch tokens", which are used in place of the traditional wordpiece tokens in NLP. There are 100 detection tokens on the right are learnable embeddings and feed into potential detections.

Compared to other CNN-based YOLO models, YOLOS benefits from the rising tides of transformers in computer vision, as well as inferring without the need for NMS, a tedious post-processing step that makes the deployment of other YOLO models difficult and slow.