roboflow / notebooks

Examples and tutorials on using SOTA computer vision models and techniques. Learn everything from old-school ResNet, through YOLO and object-detection transformers like DETR, to the latest models like Grounding DINO and SAM.
https://roboflow.com/models
5.27k stars 814 forks source link

Transformer based image classifier #254

Open ShreyasJoshi7 opened 7 months ago

ShreyasJoshi7 commented 7 months ago

Search before asking

Description

Hi, I want to add a transformer based image classifier notebook. It will be a DeiT model trained on ImageNet. The notebook I want to add will have the code for training and testing the detr model on custom dataset (from roboflow).

Additional

No response

Are you willing to submit a PR?