rwightman/efficientdet-pytorch

Getting detections in training phase

MichaelMonashev opened this issue · 2 comments

Why did you disable output['detections'] calculation here?

if not self.training:

It will useful to get bounding boxes and scores to calculate metrics while training.

@MichaelMonashev it's rather expensive when it has no impact on training or model selection (via evaluation only). Easy enough to hack it or add a force flag.

Also, to calculate meaningful metrics from the detections you have to do the full iou calcs, on train set size data (for larger datasets), that is REALLY slow and no way you'd want to have it active given that it doesn't actually factor into proper model selection.