0
$\begingroup$

Let say I have a task to detect the bounding box of one object only. And the only thing I care about is the IoU between prediction and ground truth, no need for real-time.

My question:

  1. Should I should one-stage (YOLO, SSD) or two-stage (Faster RCNN) detection model?
  2. If two-stage is better? If I use Faster RCNN, should I train the whole model or only train the Region Proposal Networks?

P/s: I have not read some state-of-the-art model yet, I heard that they implemented some really interesting things in EfficientDet or DETR, if you think those models better than two of my suggestions, please let me know.

$\endgroup$

Your Answer

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

Browse other questions tagged or ask your own question.