1
$\begingroup$

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).

$\endgroup$
2
$\begingroup$

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

| improve this answer | |
$\endgroup$
1
$\begingroup$

Here is a good example how to do it in Keras using YOLO model.

| improve this answer | |
$\endgroup$

Your Answer

By clicking “Post Your Answer”, you agree to our terms of service, privacy policy and cookie policy

Not the answer you're looking for? Browse other questions tagged or ask your own question.