naver-ai / vidt

Apache License 2.0
307 stars 40 forks source link

Simple notebook file(.ipynb) for whom wants to train/test ViDT on Colab #9

Closed EherSenaw closed 2 years ago

EherSenaw commented 2 years ago

As I first seen your paper, I'm currently trying train/test of ViDT on single machine, single gpu (especially Colab Pro).

Since there seems to be no any other materials (or .ipynb file) of tutorial for this simple testing with COCO dataset,

I would like to share my .ipynb file for whom interested in this model, and testing with Colab environment.

.ipynb file on this repo

If it bothers, please let me know, then I'll delete this colab repo.

Thanks in advance.

songhwanjun commented 2 years ago

Thanks, EherSenaw! We will include your collab tutorial in our next version of codes. We are planning to propose ViDT+ (a joint-learning pipeline on top of the vanilla ViDT). You can see it in the 'vidt-plus' branch for now.