abhisheks008 / DL-Simplified

Deep Learning Simplified is an Open-source repository, containing beginner to advance level deep learning projects for the contributors, who are willing to start their journey in Deep Learning. Devfolio URL, https://devfolio.co/projects/deep-learning-simplified-f013
https://quine.sh/repo/abhisheks008-DL-Simplified-499023976
MIT License
360 stars 300 forks source link

Dog Face Detection Using DL #560

Closed Rithish5513U closed 4 months ago

Rithish5513U commented 4 months ago

Pull Request for DL-Simplified 💡

Issue Title : Dog Face Detection using DL

Closes: #240

Describe the add-ons or changes you've made 📃

I have created a Dog Face Detection Model Using YOLOv8 Model I chose this model because it is best suited for object detection tasks.

Further I didn't use any other model because I face trouble with the amount of time it is taking to train the model. I have waited for a full night for the model to train because of the size of the dataset. Here the task is to detect dog face in the image and make a bounding box for the detected image. So I have trained YOLOv8 model.

Performance Metrics

Here I have not used Accuracy and Loss function. It is because of the bounding boxes and the problem is not about classification.

The performance metrices used here are :

How Has This Been Tested? ⚙️

I have verified the model prediction with the actual results. I have tested it by the above mentioned metrics and by manually viewing each images.

I have first trained YOLOv8 model and then I have predicted the output and tested it with the original results and gave the results accordingly.

Checklist: ☑️

github-actions[bot] commented 4 months ago

Our team will soon review your PR. Thanks @Rithish5513U :)

Rithish5513U commented 4 months ago

Can i do the other models in new notebook file or should i do it in the same notebook??

abhisheks008 commented 4 months ago

Can i do the other models in new notebook file or should i do it in the same notebook??

You can update the existing notebook and add models one after another.

Rithish5513U commented 4 months ago

@abhisheks008 I can't upload my notebook in github. If i upload, it shows only half of the page. What can I do

Rithish5513U commented 4 months ago

I actually worked on a different notebook and merged those after u said and i dont know what happenned. Even if i upload, it only uploading half of the file

Rithish5513U commented 4 months ago

@abhisheks008 If i download it again from github it shows full code. I'm so confused

abhisheks008 commented 4 months ago

@abhisheks008 If i download it again from github it shows full code. I'm so confused

Then put the files separately. But in the README mention all the implemented models and document them all.

Rithish5513U commented 4 months ago

@abhisheks008 I dont have loss function metric in YOLO and i have confidence instead of accuracy in Yolo can i provide the confidence as the performance for yolo model and for other 2 models i.e., mobilet ssd and vgg- i will provide accuracy and loss function??

abhisheks008 commented 4 months ago

@abhisheks008 I dont have loss function metric in YOLO and i have confidence instead of accuracy in Yolo can i provide the confidence as the performance for yolo model and for other 2 models i.e., mobilet ssd and vgg- i will provide accuracy and loss function??

No issues. Just need a parameter to judge/check which is the best fitted model for this dataset.

Rithish5513U commented 4 months ago

@abhisheks008 A help needed from the mentor. I have used different email of github for doing this project contribution. But my mail while I register for GSSoc'24 is rithish.satish@gmail.com. Pls add the contribution points to this mail. Now i have changed my primary mail address in github to this (rithish.satish@gmail.com)