search

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
4.76k stars 742 forks source link

DeiT Classification on custom data #258

Closed ShreyasJoshi7 closed 3 months ago

ShreyasJoshi7 commented 4 months ago

Description

I have added a notebook which demonstrates the training for an image classifier using DeiT. The pipeline downloads data from roboflow, specifically the Tumor-Classification-1 dataset for training, validation and testing.

dependencies: roboflow

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?

This has been tested on the test data downloaded from roboflow universe

Any specific deployment considerations

paper | original repo

Docs

CLAassistant commented 4 months ago

CLA assistant check
Thank you for your submission! We really appreciate it. Like many open source projects, we ask that you all sign our Contributor License Agreement before we can accept your contribution.
1 out of 2 committers have signed the CLA.

:white_check_mark: ShreyasJoshi7
:x: Shreyas Joshi

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.