pyhton 实现dlib数据集图片截取

最新推荐文章于 2024-01-21 21:13:07 发布

simple_hututu

最新推荐文章于 2024-01-21 21:13:07 发布

阅读量529

点赞数

本文链接：https://blog.csdn.net/pwtd_huran/article/details/80724776

版权

该博客介绍了一种使用Python的dlib库来处理图像数据集的方法，通过读取XML标签文件，根据对象边界框的位置将图像分为左右两部分，并分别保存。主要涉及XML解析、OpenCV图像处理和文件操作。

摘要由CSDN通过智能技术生成

#coding=utf-8
import xml.dom.minidom
import xml.etree.cElementTree as ET
from lxml import etree, objectify
import cv2
import os
import sys
import pdb
import numpy as np

def label_slice(image_path, image_savepath, label_path, label_savepath):
    delate_size = 15
    print(image_path)
    image_savepath = image_savepath
    label_ls = os.listdir(label_path)
    #print(label_ls)
    left_object_num = 0
    right_object_num = 0
    for ls in label_ls: #根据标签找对应的图片名称
        name2 = ls[0 : -4]
        print('***',name2)
        label_file_name = os.path.join(label_path, ls)
        tree=ET.parse(label_file_name)
        root=tree.getroot()
        print('999',image_path+'/'+name2+'.jpg')
        filename=root.find('filename').text
        image = cv2.imread(image_path+'/'+name2+'.jpg')
        flag = 0
        for s in root.findall('size'):
            height = s.find("height").text
            width = s.find("width").text
            depth = s.find("depth").text
            print(width)
            width_half = float(width) // 2
        E = objectify.ElementMaker(annotate=False)
        anno_tree = E.annotation(
        E.filename(filename),
        E.size(
            E.height(height),
            E.width(width),
            E.depth(dep