Fix deletion of image caused selection of the first image (#742)

Also removed the call to self.import_dir_images method that reloaded all of the images in directory.
HumanSignal · Apr 28, 2021 · 4bf5c45 · 4bf5c45
1 parent 6b5c3c6
commit 4bf5c45
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/labelImg.py b/labelImg.py
@@ -1428,12 +1428,15 @@ def close_file(self, _value=False):
     def delete_image(self):
         delete_path = self.file_path
         if delete_path is not None:
-            self.open_next_image()
-            self.cur_img_idx -= 1
+            del self.m_img_list[self.cur_img_idx]
+            self.file_list_widget.takeItem(self.cur_img_idx)
+
+            self.cur_img_idx -= 1  # self.open_next_img will increment idx.
             self.img_count -= 1
+            self.open_next_image()
+
             if os.path.exists(delete_path):
                 os.remove(delete_path)
-            self.import_dir_images(self.last_open_dir)
 
     def reset_all(self):
         self.settings.reset()