Merge pull request WZMIAOMIAO#172 from WZMIAOMIAO/dev

update code
ThisIsReed · Mar 9, 2021 · 674ef63 · 674ef63
2 parents 81545da + 6c1d130
commit 674ef63
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/pytorch_object_detection/yolov3_spp/trans_voc2yolo.py b/pytorch_object_detection/yolov3_spp/trans_voc2yolo.py
@@ -100,6 +100,7 @@ def translate_info(file_names: list, save_root: str, class_dict: dict, train_val
 
         # write object info into txt
         with open(os.path.join(save_txt_path, file + ".txt"), "w") as f:
+            assert "object" in data.keys(), "file: '{}' lack of object key.".format(xml_path)
             for index, obj in enumerate(data["object"]):
                 # 获取每个object的box信息
                 xmin = float(obj["bndbox"]["xmin"])