Skip to content

Commit 04ca02d

Browse files
committed
- 数据转换界面,现在会默认打开最后一次使用的页面。
1 parent 2d5b268 commit 04ca02d

File tree

2 files changed

+31
-1
lines changed

2 files changed

+31
-1
lines changed

ISAT/widgets/converter_dialog.py

Lines changed: 15 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -427,9 +427,24 @@ def __init__(self, parent, mainwindow):
427427
self.mainwindow = mainwindow
428428
self.converter = None
429429
self.setWindowModality(QtCore.Qt.WindowModality.WindowModal)
430+
self.tabWidget.currentChanged.connect(self.tab_widget_current_changed)
430431

431432
self.init_connect()
432433

434+
def tab_widget_current_changed(self):
435+
if self.tabWidget.currentWidget() == self.tab_COCO:
436+
self.mainwindow.cfg['software']['current_converter'] = 'coco'
437+
elif self.tabWidget.currentWidget() == self.tab_YOLO:
438+
self.mainwindow.cfg['software']['current_converter'] = 'yolo'
439+
elif self.tabWidget.currentWidget() == self.tab_LABELME:
440+
self.mainwindow.cfg['software']['current_converter'] = 'labelme'
441+
elif self.tabWidget.currentWidget() == self.tab_VOC:
442+
self.mainwindow.cfg['software']['current_converter'] = 'voc'
443+
elif self.tabWidget.currentWidget() == self.tab_VOC_DETECTION:
444+
self.mainwindow.cfg['software']['current_converter'] = 'voc for detection'
445+
else:
446+
self.mainwindow.cfg['software']['current_converter'] = 'coco'
447+
433448
def apply(self):
434449
self.tabWidget.setEnabled(False)
435450
self.pushButton_convert.setEnabled(False)
@@ -604,7 +619,6 @@ def save_file(self):
604619
if lineEdit is not None:
605620
lineEdit.clear()
606621

607-
608622
def open_dir(self):
609623
dir = QtWidgets.QFileDialog.getExistingDirectory(self, caption='Open dir')
610624
# coco2isat

ISAT/widgets/mainwindow.py

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1318,6 +1318,22 @@ def change_prompt_visiable(self):
13181318
# self.show_image(self.current_index)
13191319

13201320
def converter(self):
1321+
current_converter = self.cfg['software'].get('current_converter', 'coco')
1322+
if current_converter == 'coco':
1323+
current_converter_tab = self.Converter_dialog.tab_COCO
1324+
elif current_converter == 'yolo':
1325+
current_converter_tab = self.Converter_dialog.tab_YOLO
1326+
elif current_converter == 'labelme':
1327+
current_converter_tab = self.Converter_dialog.tab_LABELME
1328+
elif current_converter == 'voc':
1329+
current_converter_tab = self.Converter_dialog.tab_VOC
1330+
elif current_converter == 'voc for detection':
1331+
current_converter_tab = self.Converter_dialog.tab_VOC_DETECTION
1332+
else:
1333+
current_converter_tab = self.Converter_dialog.tab_COCO
1334+
1335+
self.Converter_dialog.tabWidget.setCurrentWidget(current_converter_tab)
1336+
13211337
self.Converter_dialog.show()
13221338

13231339
def video2frames(self):

0 commit comments

Comments
 (0)