🐛 [Fix] target filtering

Changed dummy target filtering to be based on class IDs being -1 and moved it inside the to_metrics_format function.

yolo/tools/solver.py

@@ -49,7 +49,7 @@ def validation_step(self, batch, batch_idx):
         H, W = images.shape[2:]
         predicts = self.post_process(self.ema(images), image_size=[W, H])
         self.metric.update([to_metrics_format(predict) for predict in predicts],
-                           [to_metrics_format(target[target.sum(1) > 0]) for target in targets])
+                           [to_metrics_format(target) for target in targets])
         return predicts
 
     def on_validation_epoch_end(self):

yolo/utils/bounding_box_utils.py

@@ -484,6 +484,7 @@ def calculate_map(predictions, ground_truths) -> Dict[str, Tensor]:
 
 
 def to_metrics_format(prediction: Tensor) -> Dict[str, Union[float, Tensor]]:
+    prediction = prediction[prediction[:, 0] != -1]
     bbox = {"boxes": prediction[:, 1:5], "labels": prediction[:, 0].int()}
     if prediction.size(1) == 6:
         bbox["scores"] = prediction[:, 5]