Data Science Asked on December 22, 2020
I need to make a model that creates bounding box around objects(but does not classify them) for a competition. Which libraries or pre-trained models should I use. I need values of x1,x2(x1+w),y1,y2(y1+h), where all dimensions are measured from top left corner. The accuracy is measured by IOU(Intersection over Union).
The mainstream algorithms used today for finding the bounding boxes of objects in an image includes:
R-CNN
(and related Fast R-CNN
and Faster R-CNN
). This family of the algorithms actual does the localization (finding the bounding boxes) and classification in two separate steps. It finds the bounding box then does the classification. So if you don't need classification, just skip the second step. YOLO
stands for 'You only look once'. The name comes from the fact this algorithm actually does the localisation and classification in one go. Both algorithms can be implemented in the mainstream python deep learning library like tensor-flow
or pytorch
etc
Answered by Louis T on December 22, 2020
Here is a good example how to do it in Keras using YOLO model.
Answered by Antonio Jurić on December 22, 2020
Get help from others!
Recent Answers
Recent Questions
© 2024 TransWikia.com. All rights reserved. Sites we Love: PCI Database, UKBizDB, Menu Kuliner, Sharing RPP