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.
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.
[ x] Bug fix (non-breaking change which fixes an issue)
[ x] New feature (non-breaking change which adds functionality)
[ x] This change requires a documentation update
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
[ x] Docs updated? What were the changes: Updated the readme doc.
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.
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