fix bug (#1874)

* fix bug * fixed according to review
Deci-AI · Feb 29, 2024 · 6c8fa3a · 6c8fa3a
1 parent ac0f4b4
commit 6c8fa3a
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/super_gradients/training/transforms/detection/legacy_detection_transform_mixin.py b/src/super_gradients/training/transforms/detection/legacy_detection_transform_mixin.py
@@ -24,9 +24,10 @@ def __call__(self, sample: Union["DetectionSample", Dict[str, Any]]) -> Union["D
         if isinstance(sample, DetectionSample):
             return self.apply_to_sample(sample)
         else:
+            has_crowd_target = "crowd_target" in sample
             sample = self.convert_input_dict_to_detection_sample(sample)
             sample = self.apply_to_sample(sample)
-            return self.convert_detection_sample_to_dict(sample, include_crowd_target="crowd_targets" in sample)
+            return self.convert_detection_sample_to_dict(sample, include_crowd_target=has_crowd_target)
 
     @classmethod
     def convert_input_dict_to_detection_sample(cls, sample_annotations: Dict[str, Union[np.ndarray, Any]]) -> DetectionSample: