SaiNagendraD/Classify_Flash_and_Non_Flash_Images_IITD_DSML_Batch10_Group5

This Project is part of IIT Delhi CEP online course of DSML Batch-10. Given model classify images into Flash and Non Flash categories. Group Project members:

1. Ajith Francis P
2. D H Sai Nagendra
3. KALYAN SUNDAR PAL
4. Nalini Yadav
5. Ravi Teja Manchikanti
6. Sowmya M
7. Yash Girdhar

Under the guidance of IITD Co-ordinators:

1. Dr Hariprasad Kodamana Associate Professor, Department of Chemical Engineering

2. Dr Agam Gupta Associate Professor, Department of Management Studies

3. Dr Manojkumar Ramteke Professor, Department of Management Studies

Usage:

1. Under main project directory in your local system, create test_images folder and use .jpg files as input for the python scripts provided above.
2. Import necessary packages to work with .pth and .joblib models. It should be noted that model "flash_detector_model.joblib" has accuracy of 65% and model "vit_flash_classifier.pth" is 100% with test data and provide good prediction for real-time applications.

Hope these models would help with your requirements. Thank you