str method

Author: Koldim2001

patched_yolo_infer/nodes/CombineDetections.py

@@ -448,7 +448,7 @@ def not_agnostic_nms(
         return all_keeps
 
     def __str__(self):
-        # List of useful attributes (non-empty ones)
+        # Print the list of useful attributes (non-empty ones)
         useful_attributes = []
         if self.filtered_confidences:
             useful_attributes.append("filtered_confidences")
@@ -465,7 +465,7 @@ def __str__(self):
 
         # If all attributes are empty
         if not useful_attributes:
-            return "Useful attributes: nothing detected in the frame."
+            return "Useful attributes: nothing detected in the frame"
 
         # Build the output string
         output = "Useful attributes: " + ", ".join(useful_attributes) + "\n\n"

patched_yolo_infer/nodes/MakeCropsDetectThem.py

@@ -296,3 +296,10 @@ def _calculate_batch_inference(
                 else:
                     # Get the masks
                     crop.detected_masks = pred.masks.data.cpu().numpy()
+
+    def __str__(self):
+        # Print info about patches amount
+        return (
+            f"{len(self.crops)} patches of size {self.crops[0].crop.shape} "
+            f"were created from an image sized {self.image.shape}"
+        )