How to Choose the Best Computer Vision Model for Your Project

In this video, we will dive into the complexity of choosing the right computer vision model for your unique project.
From the importance of high-quality datasets to hardware considerations, interoperability, benchmarking, and licensing issues, this video covers it all. Whether you're planning to develop an app for counting commuters in public transport or analyzing medical images, we guide you on the critical factors that should inform your model selection. We even explore specific models like YOLOv5, YOLO-NAS, and Detectron2 in context. Don't forget to like, subscribe, and stay tuned for more computer vision content!
Chapters:
00:00 Introduction
00:40 Overthinking Model Selection
01:36 Different Project Contexts (Counting People vs Analyzing Medical Images)
03:15 Hardware Considerations
04:04 mAP vs Latency
05:33 Benchmarking and the Importance of Preliminary Testing
06:00 Understanding mAP Values in the Context of Custom Datasets
08:27 Library Packaging
09:46 Model Integration and the Role of SDKs
10:52 Importance of Active Project Support
11:27 Understanding Project Licenses
12:31 Conclusion
Resources:
🌏 Roboflow: roboflow.com
🌌 Roboflow Universe: universe.robof...
📚 Roboflow Notebooks Repository: github.com/rob...
🎬 YOLOv8: How to Train for Object Detection on a Custom Dataset: • YOLOv8: How to Train f...
Stay updated with the projects I'm working on at github.com/rob... and github.com/Ska...! ⭐
Don't forget to like, comment, and subscribe for more content on AI, computer vision, and the latest technological breakthroughs! 🚀
#ComputerVision #ObjectDetection #InstanceSegmentation #DeepLearning #YOLO #Detectron2 #Dataset #ModelSelection #AI #YOLOv8

[@annamaule7333]

Thank you so much for this video! Very informative, complete, and super on top of everything. It is nice to hear your experience and how this is matching the things my team and I are going through: having hardware, model size, speed (fps), and mAP in mind. We also ran into the issue of testing yolov5 and the repo not being an sdk, leading us to bring the repo as a submodule and have to do some hacks around, and re-write the predict script because the repo was not built with integration to 3rd parties in mind! Very very good content and very aligned with my personal experience!

[@Roboflow]

This is awesome to read! I’m super happy that other people see things similar way 🔥

[@rodmallen9041]

Badass guidelines....as badass as your looks for this video 😎🤘...tks for sharing

[@willmarsman1765]

Thanks for reviewing licenses in this context; I've noticed licenses are quite complicated for models as compared to other software projects. for example, the super-gradients project has two licenses, one which applies to the model and another the project overall. the model license also appears completely custom. I hope in the future we will see a consolidation of licenses around publicly shared models.

[@diogoalves...]

Great video, Peter! It would be nice if you could provide us with a summary table sometime in the future. Something that includes columns such as usability, portability, available customization parameters, license, latency, etc. Additionally, a follow-up with an evaluation template be greatly appreciated. It would help us compare our fitted models effectively. Congratulations! The content was truly excellent.

[@adurks4846]

The performance is one that always gets us. We still use scaled yolov4 because it performs better than anything else on our datasets. This is ignoring all of our legacy code which makes it difficult to implement newer YOLO models (looking at you yolov8). As an aside, does it feel like newer models are more focused towards the COCO dataset? Are researchers "gaming" their architectures to focus specifically on the types of images in COCO ( off-nadir high resolution, high fidelity, well-lit scenes, low # of targets) to get at the top of the leaderboards?

[@Roboflow]

I don’t have any proof of that, but when we tested model fine-tuning on custom datasets we noticed that very often models that are bettor on COCO perform worst on custom datasets. It is interesting dynamic. As for our model. You are willing to do all of those trade offs and still use YOLOv4. How large is the mAP difference?

[@adurks4846]

@@Roboflow Sometimes pretty significant, as much as 10-20% when you compare syolov4 vs yolov5/8. I will say that part of that is due to our focus. We care much more about recall than mAP. I note that sometimes yolov5/8 get better mAP but worse recall even if you drop the thresholds.

[@g.s.3389]

But at the end what would you use ? any examples? ease of use vs libraries or requirements (i.e python3.11 needed...)?

[@Roboflow]

I prefer easy installation and use.

[@fazlehasan9428]

It is the best video on model selection keep making videos like it

[@user-eb5rc6nq3c]

Thank you for your vedio! I am currently struggling with large dataset annotation (around 40,000 images), so I am thinking of semi-supervised methods to do the object detection. But I have no idea how to pick models for both teacher stage and student stage. Would you have any advice on that?

[@Roboflow]

Hi 👋Make sure to take a look at to of our previous videos: kzbin.info/www/bejne/eWWxopSId9-Sgdk and kzbin.info/www/bejne/pXa0ioaqo6tlopo. I hope you will find some inspiration there.

[@allistech6748]

Hope this video is not inspired by the discussion we had last week. HA HA HA!!! Just kidding, Thanks for the video helped a lot.

[@milindchaudhari1676]

Hii sir, I'm Milind this side, working on a fruit detection model as my master's thesis project where i have taken around 300 images of the fruits on the trees. Now i need to annotate them but I'm experiencing the scenario where, the fruits are occluded by the leaves and are overlapping with each other as well, as currently no-one is guiding me in dealing with such cases, I'm getting tensed in annotating my images, i would like to seek your guidance regarding the same. Please help me out with a reply...!

[@mrbot4one]

What about the SAM , is it fast enough to compare with these models in terms of accuracy,precision,fast and lightweight???

[@lemonbitter7641]

That talking to gpt was hilarious 😂

[@user-ct8dt1lf7i]

I was wondering is there any chance that we can convert all the bounding boxes in the image into polygon all together? and vice versa. if yes it will be really helpful.

[@user-ct8dt1lf7i]

oh i saw it thanks really appreciate it

[@Roboflow]

@@user-ct8dt1lf7i yes we absolutely can!

[@user-fd8sw5ig3n]

Hi, also from my side, thanks a lot for this and all the other awesome and really helpful videos! I have a question regarding 'my' specific issue (sorry if this is not the right platform to ask this): I have a very small dataset (ca. 100 images) that I would like to use for object detection. It's ghostnets on sonar images - visually similar to the concrete cracks dataset that you used in another video. I tried Yolov5 with weight transfer & fine-tuning which already works ok, but am not sure about it. Would you have a suggestion for me on what to do? Just try around with hyperparameters and fine-tuning, use another model etc? Thanks a lot in advance! Mia

[@youssefkhaled5331]

Thnx for the content, Can I know how to open webcam in yolov7 method in colab I try hard but I get nothing thanks again.

[@rachealcr6752]

I just wonder why using same model, settings in 2 versions in roboflow and train in google colab but the results of mAP, accuracy and recall varies a lot about 40% of difference.

[@Roboflow]

Interesting. Could you let me know what we’re the hiperparams you used in notebook?

[@rachealcr6752]

@@Roboflow 100 Epochs others remain the same with custom dataset

[@Roboflow]

@@rachealcr6752 which size of the model you trained in colab? And what training option you chose in UI?

[@rachealcr6752]

@@Roboflow yolo v8s. PREPROCESSING Auto-Orient: Applied Resize: Stretch to 640x640 AUGMENTATIONS Outputs per training example: 3 Flip: Horizontal Noise: Up to 5% of pixels

[@dahiruibrahimdahiru2690]

Nah mahn, where has this channel been all this while

[@techradar6787]

Useful ❤❤❤

[@Roboflow]

Awesome!

[@body1024]

keep it coming 😍

[@Roboflow]

Thanks a lot for kind words 🙏🏻

[@st43r62]

the bestest!

[@Roboflow]

Thanks a lot! 🙏🏻

[@IntelligentQuads]

Go spurs go!

[@Roboflow]

Maybe next year. 😅