I’m creating this topic to discuss the
ObjectDetectionInterpretation class I’m developing to inspect results from Object Detection with a similar API as
Any suggestion is most welcome !
I have a question for you @sgugger if you don’t mind. One issue I’m running into is in the
loss_batch function (in
basic_train) on this line :
if not loss_func: return to_detach(out), yb.detach()
Only the first element of
yb is returned (I don’t really know why?). In Object Detection
yb is the bounding boxes targets, and
yb is the classes targets, and both are needed for
get_preds. So what would you suggest ?
Can I modify
loss_batch directly to return also
yb (in that case
validate would also need to be modified, at least)? I’m not sure how that would impact other applications so that’s why I’m asking. The alternative is to developp a
get_preds method specific to ObjectDetection.
Could you advise me on the best course of action ?