瀏覽代碼

Make bbox_detection/labelme2voc.py more robust

Kentaro Wada 5 年之前
父節點
當前提交
eb010f2aea
共有 1 個文件被更改,包括 4 次插入0 次删除
  1. 4 0
      examples/bbox_detection/labelme2voc.py

+ 4 - 0
examples/bbox_detection/labelme2voc.py

@@ -101,6 +101,10 @@ def main():
             class_id = class_names.index(class_name)
             class_id = class_names.index(class_name)
 
 
             (xmin, ymin), (xmax, ymax) = shape['points']
             (xmin, ymin), (xmax, ymax) = shape['points']
+            # swap if min is larger than max.
+            xmin, xmax = sorted([xmin, xmax])
+            ymin, ymax = sorted([ymin, ymax])
+
             bboxes.append([xmin, ymin, xmax, ymax])
             bboxes.append([xmin, ymin, xmax, ymax])
             labels.append(class_id)
             labels.append(class_id)