Don't repeat class example images (#1)

lightly-ai · Nov 10, 2023 · 6d78b9f · 6d78b9f
1 parent 02890bc
commit 6d78b9f
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/lightly_insights/analyze.py b/src/lightly_insights/analyze.py
@@ -189,7 +189,10 @@ def analyze_object_detections(
             class_datum.heatmap[int(y1) : int(y2), int(x1) : int(x2)] += 1
 
             # Sample images.
-            if len(class_datum.sample_filenames) < 4:
+            if (
+                len(class_datum.sample_filenames) < 4
+                and label.image.filename not in class_datum.sample_filenames
+            ):
                 class_datum.sample_filenames.append(label.image.filename)
 
         # Update objects per image for classes.