提交 a2f1824c 编写于 作者: M mfurkancoskun 提交者: Boris Sekachev

TypeError Fix (#596)

Fixed: TypeError while using XML to YOLO converter #595 
上级 cad5ac86
......@@ -207,8 +207,8 @@ def process_cvat_xml(xml_file, image_dir, output_dir,username,password,ilabels):
else:
for img_tag in cvat_xml.findall('image'):
image_name = img_tag.get('name')
width = img_tag.get('width')
height = img_tag.get('height')
width = int(img_tag.get('width'))
height = int(img_tag.get('height'))
image_path = os.path.join(image_dir, image_name)
if not os.path.exists(image_path):
log.warn('{} image cannot be found. Is `{}` image directory correct?'.
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册