deleteSimilarDetections method is interrupted error fix

- Now, if the deleteSimilarDetections method is interrupted, it should delete the alone datas.
PandapowrTR · May 7, 2024 · fa91e5f · fa91e5f
1 parent 475611e
commit fa91e5f
Showing 1 changed file with 5 additions and 0 deletions.
diff --git 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