optimized deleteSimilarDetections

PandapowrTR · Feb 3, 2024 · 4f6f1b2 · 4f6f1b2
1 parent f3fddb2
commit 4f6f1b2
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/Data/Dominate/DominateLabeledImage.py b/Data/Dominate/DominateLabeledImage.py
@@ -249,7 +249,7 @@ def processFile(args):
                 uu = str(uuid.uuid1())
                 cutDetection = os.path.join(tempFolder, "temp-" + uu + file)
                 cv2.imwrite(cutDetection, img)
-
+                foundSimilar = False
                 for checkFile in files.copy():
                     if (
                         checkFile.lower().endswith((".png", ".jpeg", ".jpg"))
@@ -279,6 +279,10 @@ def processFile(args):
                             if imgAreSimilar(
                                 cutDetection, checkCutDetection, maxSimilarity
                             ):
+                                if not foundSimilar:
+                                    foundSimilar = True
+                                else:
+                                    continue
                                 os.remove(os.path.join(root, checkFile))
                                 deletedFiles.append(checkFile)
                                 os.remove(checkCutDetection)