From fa91e5f8b97b39e485b3b0cfb1af97f1ce865651 Mon Sep 17 00:00:00 2001
From: Pandapowr <144907656+PandapowrTR@users.noreply.github.com>
Date: Tue, 7 May 2024 18:21:57 +0300
Subject: [PATCH] deleteSimilarDetections method is interrupted error fix

- Now, if the deleteSimilarDetections method is interrupted, it should delete the alone datas.
---
 Data/Dominate/DominateLabeledImage.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/Data/Dominate/DominateLabeledImage.py b/Data/Dominate/DominateLabeledImage.py
index 9eb5f67..a2c57ae 100644
--- a/Data/Dominate/DominateLabeledImage.py
+++ b/Data/Dominate/DominateLabeledImage.py
@@ -787,6 +787,11 @@ def augData(
                 deleteSimilarDetections(
                     imgSavePath, labelSavePath, labelSaveFormat, maxSimilarity
                 )
+            except KeyboardInterrupt:
+                print("🔄 Deleting alone data 🥺💔")
+                time.sleep(1)
+                deleteAloneData(imgSavePath, labelSavePath)
+                BurobotOutput.clearAndMemoryTo()
             except:
                 pass