From 784144aa296f0e2b3580119f0abade4dd3e2c08a Mon Sep 17 00:00:00 2001 From: nine Date: Sun, 27 Mar 2022 19:04:09 +0200 Subject: [PATCH] Fixes deleting an image (cf. #822) (#861) Co-authored-by: Erwin Nindl --- labelImg.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/labelImg.py b/labelImg.py index a19f3372..d7401bca 100755 --- a/labelImg.py +++ b/labelImg.py @@ -1454,12 +1454,16 @@ class MainWindow(QMainWindow, WindowMixin): def delete_image(self): delete_path = self.file_path if delete_path is not None: - self.open_next_image() - self.cur_img_idx -= 1 - self.img_count -= 1 + idx = self.cur_img_idx if os.path.exists(delete_path): os.remove(delete_path) self.import_dir_images(self.last_open_dir) + if self.img_count > 0: + self.cur_img_idx = min(idx, self.img_count - 1) + filename = self.m_img_list[self.cur_img_idx] + self.load_file(filename) + else: + self.close_file() def reset_all(self): self.settings.reset()