typos and hide after macro execution

psomers3 · Jan 12, 2021 · 3b219d1 · 3b219d1
1 parent f7bf370
commit 3b219d1
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/labelme/widgets/segmentation_macro.py b/labelme/widgets/segmentation_macro.py
@@ -1,5 +1,3 @@
-from qtpy.QtCore import *
-from qtpy.QtGui import *
 from qtpy.QtWidgets import *
 import glob
 import labelme
@@ -28,7 +26,7 @@ def __init__(self):
         button_widget = QWidget()
         button_widget.setLayout(button_layout)
         voc_semantic_seg_btn = QPushButton('VOC Semantic\nSegmentation')
-        voc_semantic_seg_btn.pressed.connect(self.do_semantic_segmentation)
+        voc_semantic_seg_btn.pressed.connect(self.do_semantic_segmentation_voc)
         button_layout.addWidget(voc_semantic_seg_btn)
         voc_instance_seg_btn = QPushButton('VOC Instance\nSegmentation')
         voc_instance_seg_btn.pressed.connect(self.do_instance_segmentation_voc)
@@ -239,6 +237,7 @@ def do_instance_segmentation_voc(self):
                     loc="rb",
                 )
                 imgviz.io.imsave(out_insv_file, insv)
+        self.hide()
 
     def do_instance_segmentation_coco(self):
         try:
@@ -416,6 +415,8 @@ def do_instance_segmentation_coco(self):
 
         with open(out_ann_file, "w") as f:
             json.dump(data, f)
+        self.hide()
+
 
 class LineEntry(QWidget):
     def __init__(self, label, dialogue_type):