Check annotation in pascal_voc

最新推荐文章于 2020-09-20 19:36:51 发布

DuinoDu

最新推荐文章于 2020-09-20 19:36:51 发布

阅读量492

点赞数

分类专栏： python 文章标签： python

本文链接：https://blog.csdn.net/DuinoDu/article/details/66973096

版权

python 专栏收录该内容

23 篇文章 0 订阅

订阅专栏

#!/usr/bin/env python
# -*- coding: utf-8 -*-

'''
python checkgt.py jpeg_file [xml path]
'''
import os, sys
import os.path as osp
import cv2
import xml.etree.ElementTree as ET


def loadXML(xmlfile):
    tree = ET.parse(xmlfile)
    objs = tree.findall('object')
    boxes = []
    for ix, obj in enumerate(objs):
        bbox = obj.find('bndbox')
        # Make pixel indexes 0-based
        x1 = int(bbox.find('xmin').text) - 1
        y1 = int(bbox.find('ymin').text) - 1
        x2 = int(bbox.find('xmax').text) - 1
        y2 = int(bbox.find('ymax').text) - 1
        boxes.append([x1, y1, x2, y2])
    return boxes

def checkgt(argv):
    imgfile = osp.abspath(argv[1])
    xmlfile=''
    if len(argv) == 3:
        xmlfile = argv[2]
    else:
        dataset_root, img_name = os.path.split(imgfile)
        xml_name = osp.split( osp.splitext(imgfile)[0] )[1] + '.xml'
        xmlfile = osp.join(dataset_root, '..', 'Annotations', xml_name) 

    boxes = loadXML(xmlfile)
    img = cv2.imread(imgfile)
    for box in boxes:
        img = cv2.rectangle(img, (box[0], box[1]), (box[2], box[3]), (0, 255, 255))

    cv2.imshow('checkgt', img)
    cv2.waitKey(0)

def main():
    import sys
    if len(sys.argv) != 2:
        print(__doc__)
        return
    checkgt(sys.argv)

if __name__ == "__main__":
    main()