5 ms·
I work a lot with object detection at my current role. Would be cool to add Detectron 2, seems that this library that has some of the state of the art(SOTA) mo
by mendeza 7y ago
I work a lot with object detection at my current role.
Would be cool to add Detectron 2, seems that this library that has some of the state of the art(SOTA) models in object detection. https://github.com/facebookresearch/detectron2 https://github.com/facebookresearch/detectron2
Also the TF Object Detection API has a suite of pretrained SOTA object detection models https://github.com/tensorflow/models/tree/master/research/object_detection https://github.com/tensorflow/models/tree/master/research/ob...
What makes object detection really valuable is being able to finetune a model to detect objects that are relevant to your problem. However collecting, labeling, takes a lot of time since the best way is to manually label. But 30-50 hrs of manual labeling can get you some great results when you are leveraging pretrained models. Building a tool to manage data labeling and training for finetuning object detection models would be very valuable.
- mendeza 7y agoAlso, what is a hard but an important question to figure out is how well are these models are preforming, and will they perform well to your problem. This paper gives a lot of insight into how image size and other factors really impact performance of modern object detection: https://arxiv.org/pdf/1611.10012.pdf https://arxiv.org/pdf/1611.10012.pdf. How large or small your object is in your dataset is also something to consider since there is a limit to how small of an object your detector can recognize. Also, I would recommend this repo to better understand how object detection models are evaluated as well: https://github.com/rafaelpadilla/Object-Detection-Metrics https://github.com/rafaelpadilla/Object-Detection-Metrics
- inetknght 7y ago> Building a tool to manage data labeling and training for finetuning object detection models would be very valuable. Something like recaptcha?
- adipandas 7y ago@mendeza, Thank you for the feedback. I am planning to add detectron models as well to this repository. Initial intention of this project was to avoid direct dependency of any deep learning framework. I have only used pre-trained models and opencv for this project. Making a tool for data labeling is an awesome idea. I will definitely like to work on it. We can collaborate on this if possible. You are also welcome to send in your pull requests.
- mendeza 7y agoThis is a good idea! TF Object Detection framework takes a lot of effort to use on simple ideas, so frameworks that enable more abstraction the better!
- ganeshkrishnan 7y agoFacebook has such awesome AI tools open sourced. I am more impressed with their code than Google
- adipandas 7y agoPytorch makes it even more fun.
- deppp 7y agoCheck out our open-source tool called Label Studio, https://labelstud.io https://labelstud.io and we have a commercial offering that does exactly this, finetuning the models.
- mendeza 7y agoI actually checked out your github repo and saw your framework that enables integration of trained models into your tool for active learning. Looks awesome! https://github.com/heartexlabs/pyheartex https://github.com/heartexlabs/pyheartex
- mendeza 7y agoIt would be great if your tool integrates popular object detection frameworks such as TF Object Detection API. Would save people a lot of headache installing and configuring in order to begin finetuning