目标检测预处理分析(列表样本分布,GT可视化)preprocessing

s

目标检测预处理分析有两个任务:

  1. 所有类别数量的均衡度分析(直方图或者柱状图可视化)

  2. 把GT画在每一张对应的图片上,并在另外一个文件里保存

所有类别数量的均衡度分析(直方图或者柱状图可视化)

python:批量统计xml中各类目标的数量_Python

Python对目标检测数据集xml文件操作(统计目标种类、数量、面积、比例等&修改目标名字)

系统环境为ubuntu16.04

编译器为python2.7

模块中的函数需要添入三个变量,分别为待处理图像的文件夹路径,xml文件夹路径,处理后图像的文件夹路径

以下代码已经被我改完bug跑通:

# -*- coding:utf-8 -*-
# 根据xml文件统计目标种类以及数量
import os
import xml.etree.ElementTree as ET
import numpy as np

np.set_printoptions(suppress=True, threshold=10000000)
import matplotlib
from PIL import Image


def parse_obj(xml_path, filename):
    tree = ET.parse(xml_path + filename)
    objects = []
    for obj in tree.findall('object'):
        obj_struct = {}
        obj_struct['name'] = obj.f

猜你喜欢

转载自blog.csdn.net/txpp520/article/details/104832033