Submission for Computer Vision Challenge 24 by Team DS & Chill, where we finished 10th, with a combined metric of 0.505.
This repository contains the necessary notebooks for training both the DenseNet and YOLO models, as well as a notebook for ensemble predictions. Additionally, it includes a CSV file listing the augmentations applied to each class in aug_list, along with a report explaining our approach.
- Notebooks:
- Training notebooks for DenseNet and YOLO models
- Ensemble prediction notebook
- Data:
- CSV file detailing augmentations for each class
- Report explaining our methodology
Due to size limitations, the dataset and models are not included in this repository. You can access them using the following links:
- Dataset: Google Drive Link
- Models: Google Drive Link
-
Ritvik Sharma
Email: t24cs001@nitm.ac.in
M.Tech Student at NITM -
Tanishq Sasmal
Email: t24cs002@nitm.ac.in
M.Tech Student at NITM -
Rahil Agarwal
Email: t24cs009@nitm.ac.in
M.Tech Student at NITM