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

DeiT Image Classifier. Resolving the issue "Transformer based image classifier #254" #257

Closed ShreyasJoshi7 closed 7 months ago

ShreyasJoshi7 commented 7 months ago

Description

I have added a notebook which demonstrates the training of DeiT model to perform image classification on custom dataset. I demonstrate training on a custom dataset from roboflow. I have mentioned the links for the original github repo and the paper in the readme file and the noteboook

dependencies: roboflow (for dataset)

Type of change

Please delete options that are not relevant.

How has this change been tested, please provide a testcase or example of how you tested the change?

I have tested the code on the roboflow dataset. specifically the Tumor-Classification-1 dataset's test data.

Any specific deployment considerations

No.

Docs

CLAassistant commented 7 months ago

CLA assistant check
Thank you for your submission! We really appreciate it. Like many open source projects, we ask that you sign our Contributor License Agreement before we can accept your contribution.


Shreyas Joshi seems not to be a GitHub user. You need a GitHub account to be able to sign the CLA. If you have already a GitHub account, please add the email address used for this commit to your account.
You have signed the CLA already but the status is still pending? Let us recheck it.