YOLOS
YOLOS copied to clipboard
hustvl
[NeurIPS 2021] You Only Look at One Sequence
Using the default FeatureExtractor settings for the HuggingFace port of YOLOS, I am consistently running into CUDA OOM errors on a 16GB V100 (even with a training batch size of...
Hi YOLOS team :) I've implemented YOLOS as a fork of [🤗 HuggingFace Transformers](https://github.com/huggingface/transformers.git), and I'm going to add it soon to the library (see https://github.com/huggingface/transformers/pull/16848). Here's a notebook that...
## ❔Question I convert PASCAL VOC dataset to COCO format, but when I trained yolos-tiny with 150 epochs and pre-trained weights , the results is so bad. I get no...
## ❔Question Hi，what is the magnitude of YOLOS model parameters ? ## Additional context PS :Do you have the official implementation of article '[Benchmarking Detection Transfer Learning with Vision Transformers](https://arxiv.org/pdf/2111.11429.pdf)'...
We load our pretrained model of vit-base trained with mae method, and we meet the size mismatch for pos_embed. Is there any solution to this problem please? RuntimeError: Error(s) in...
## ❔Question For training a custom dataset is it possible just change the path from coco to my own dataset ? ## Additional context
Hi, When I read your code, I found "Unresolved reference 'sigmoid_focal_loss' and Unresolved reference 'dice_loss' " in 'model/detector.py' file, and I could not find any code related to these two...
How to inference on a single image and visualize the result?
## ❔Question Hello, thank you for this great contribution. I'm asking if with your architecture, we can control witch patches to feed to YOLOS, because i have already the RoI...
## ❔Question Hi thanks for opensourcing the code base this gives steps to learn transformers, i am having few queries 1. The dataset is loaded from coco.py using which function...
