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