From 2a0fbcce5f745cf4bdbefc7052369dd87cb36b7e Mon Sep 17 00:00:00 2001 From: POONAM2015 Date: Fri, 26 Aug 2022 03:41:37 +0900 Subject: [PATCH] Fixed Issue 465: Saving Yolo labels, corrupts classes.txt --- labelImg.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/labelImg.py b/labelImg.py index efd8a2976..7819a1f1f 100755 --- a/labelImg.py +++ b/labelImg.py @@ -1357,7 +1357,11 @@ def open_dir_dialog(self, _value=False, dir_path=None, silent=False): target_dir_path = ustr(default_open_dir_path) self.last_open_dir = target_dir_path self.import_dir_images(target_dir_path) - self.default_save_dir = target_dir_path + if not self.default_save_dir: + self.default_save_dir = target_dir_path + file_containing_labels = os.path.join(self.default_save_dir, 'classes.txt') + if not self.label_hist: + self.load_predefined_classes(file_containing_labels) if self.file_path: self.show_bounding_box_from_annotation_file(file_path=self.file_path)