diff --git a/tools/toCOCO.py b/tools/toCOCO.py index cc2257d..1694c75 100644 --- a/tools/toCOCO.py +++ b/tools/toCOCO.py @@ -263,7 +263,10 @@ class COCOConverter: for obj in objects: object = {} object['category'] = obj.get('category', '') - object['group'] = groups_dict.get(obj.get('group', 0)) + if 'background' in object['category']: + object['group'] = 0 + else: + object['group'] = groups_dict.get(obj.get('group', 0)) object['segmentation'] = obj.get('segmentation', []) object['area'] = obj.get('area', None) object['layer'] = obj.get('layer', None) diff --git a/ui/COCO_to_ISAT_dialog.py b/ui/COCO_to_ISAT_dialog.py index b574486..09d9834 100644 --- a/ui/COCO_to_ISAT_dialog.py +++ b/ui/COCO_to_ISAT_dialog.py @@ -15,7 +15,7 @@ class Ui_Dialog(object): def setupUi(self, Dialog): Dialog.setObjectName("Dialog") Dialog.setWindowModality(QtCore.Qt.NonModal) - Dialog.resize(514, 202) + Dialog.resize(600, 166) font = QtGui.QFont() font.setFamily("Times New Roman") font.setPointSize(12) @@ -49,6 +49,7 @@ class Ui_Dialog(object): self.widget_3 = QtWidgets.QWidget(Dialog) self.widget_3.setObjectName("widget_3") self.horizontalLayout_2 = QtWidgets.QHBoxLayout(self.widget_3) + self.horizontalLayout_2.setContentsMargins(0, 0, 0, 0) self.horizontalLayout_2.setObjectName("horizontalLayout_2") spacerItem = QtWidgets.QSpacerItem(40, 20, QtWidgets.QSizePolicy.Expanding, QtWidgets.QSizePolicy.Minimum) self.horizontalLayout_2.addItem(spacerItem) @@ -59,6 +60,7 @@ class Ui_Dialog(object): self.widget_4 = QtWidgets.QWidget(Dialog) self.widget_4.setObjectName("widget_4") self.horizontalLayout_3 = QtWidgets.QHBoxLayout(self.widget_4) + self.horizontalLayout_3.setContentsMargins(0, 0, 0, 0) self.horizontalLayout_3.setObjectName("horizontalLayout_3") spacerItem1 = QtWidgets.QSpacerItem(40, 20, QtWidgets.QSizePolicy.Expanding, QtWidgets.QSizePolicy.Minimum) self.horizontalLayout_3.addItem(spacerItem1) @@ -97,13 +99,13 @@ class Ui_Dialog(object): def retranslateUi(self, Dialog): _translate = QtCore.QCoreApplication.translate - Dialog.setWindowTitle(_translate("Dialog", "ISAT to VOC png")) + Dialog.setWindowTitle(_translate("Dialog", "COCO to ISAT")) self.pushButton_save_root.setText(_translate("Dialog", "Save root")) self.pushButton_label_path.setText(_translate("Dialog", "Json path")) self.lineEdit_save_root.setPlaceholderText(_translate("Dialog", "ISAT jsons save root")) self.lineEdit_label_path.setPlaceholderText(_translate("Dialog", "COCO json path")) self.checkBox_keepcrowd.setText(_translate("Dialog", "Keep crowd")) - self.label.setText(_translate("Dialog", "Convert COCO json to ISAT jsons.")) + self.label.setText(_translate("Dialog", "Convert COCO json to ISAT jsons.All layer attr is 1.")) self.pushButton_cache.setText(_translate("Dialog", "cache")) self.pushButton_apply.setText(_translate("Dialog", "convert")) import icons_rc diff --git a/ui/COCO_to_ISAT_dialog.ui b/ui/COCO_to_ISAT_dialog.ui index 557a10b..f10b015 100644 --- a/ui/COCO_to_ISAT_dialog.ui +++ b/ui/COCO_to_ISAT_dialog.ui @@ -9,8 +9,8 @@ 0 0 - 514 - 202 + 600 + 166 @@ -20,7 +20,7 @@ - ISAT to VOC png + COCO to ISAT false @@ -90,6 +90,18 @@ + + 0 + + + 0 + + + 0 + + + 0 + @@ -116,6 +128,18 @@ + + 0 + + + 0 + + + 0 + + + 0 + @@ -160,7 +184,7 @@ color: rgb(255, 0, 0); - Convert COCO json to ISAT jsons. + Convert COCO json to ISAT jsons.All layer attr is 1. diff --git a/ui/ISAT_to_COCO_dialog.py b/ui/ISAT_to_COCO_dialog.py index a171bfc..45a5c0e 100644 --- a/ui/ISAT_to_COCO_dialog.py +++ b/ui/ISAT_to_COCO_dialog.py @@ -15,7 +15,7 @@ class Ui_Dialog(object): def setupUi(self, Dialog): Dialog.setObjectName("Dialog") Dialog.setWindowModality(QtCore.Qt.NonModal) - Dialog.resize(508, 155) + Dialog.resize(600, 137) font = QtGui.QFont() font.setFamily("Times New Roman") font.setPointSize(12) @@ -49,6 +49,7 @@ class Ui_Dialog(object): self.widget_3 = QtWidgets.QWidget(Dialog) self.widget_3.setObjectName("widget_3") self.horizontalLayout_2 = QtWidgets.QHBoxLayout(self.widget_3) + self.horizontalLayout_2.setContentsMargins(0, 0, 0, 0) self.horizontalLayout_2.setObjectName("horizontalLayout_2") spacerItem = QtWidgets.QSpacerItem(40, 20, QtWidgets.QSizePolicy.Expanding, QtWidgets.QSizePolicy.Minimum) self.horizontalLayout_2.addItem(spacerItem) @@ -87,12 +88,12 @@ class Ui_Dialog(object): def retranslateUi(self, Dialog): _translate = QtCore.QCoreApplication.translate - Dialog.setWindowTitle(_translate("Dialog", "ISAT to VOC png")) + Dialog.setWindowTitle(_translate("Dialog", "ISAT to COCO")) self.pushButton_save_path.setText(_translate("Dialog", "Save path")) self.pushButton_label_root.setText(_translate("Dialog", "Jsons root")) self.lineEdit_save_path.setPlaceholderText(_translate("Dialog", "COCO json save path")) self.lineEdit_label_root.setPlaceholderText(_translate("Dialog", "ISAT jsons root")) - self.label.setText(_translate("Dialog", "Convert ISAT jsons to COCO json.")) + self.label.setText(_translate("Dialog", "Convert ISAT jsons to COCO json.The layer attr will be lost.")) self.pushButton_cache.setText(_translate("Dialog", "cache")) self.pushButton_apply.setText(_translate("Dialog", "convert")) import icons_rc diff --git a/ui/ISAT_to_COCO_dialog.ui b/ui/ISAT_to_COCO_dialog.ui index f1acc8c..ac58848 100644 --- a/ui/ISAT_to_COCO_dialog.ui +++ b/ui/ISAT_to_COCO_dialog.ui @@ -9,8 +9,8 @@ 0 0 - 508 - 155 + 600 + 137 @@ -20,7 +20,7 @@ - ISAT to VOC png + ISAT to COCO false @@ -90,6 +90,18 @@ + + 0 + + + 0 + + + 0 + + + 0 + @@ -134,7 +146,7 @@ color: rgb(255, 0, 0); - Convert ISAT jsons to COCO json. + Convert ISAT jsons to COCO json.The layer attr will be lost. diff --git a/ui/ISAT_to_VOC_dialog.py b/ui/ISAT_to_VOC_dialog.py index 9bddc07..a4c0bd8 100644 --- a/ui/ISAT_to_VOC_dialog.py +++ b/ui/ISAT_to_VOC_dialog.py @@ -15,7 +15,7 @@ class Ui_Dialog(object): def setupUi(self, Dialog): Dialog.setObjectName("Dialog") Dialog.setWindowModality(QtCore.Qt.NonModal) - Dialog.resize(509, 175) + Dialog.resize(600, 173) font = QtGui.QFont() font.setFamily("Times New Roman") font.setPointSize(12) diff --git a/ui/ISAT_to_VOC_dialog.ui b/ui/ISAT_to_VOC_dialog.ui index 2225c89..2dbe24e 100644 --- a/ui/ISAT_to_VOC_dialog.ui +++ b/ui/ISAT_to_VOC_dialog.ui @@ -9,8 +9,8 @@ 0 0 - 509 - 175 + 600 + 173