pascal-voc格式数据集的一些工具

pascal-voc-tools的Python项目详细描述


帕斯卡VOC工具

这个包包括一些关于pascal voc格式数据集的工具,如读xml、写xml、调整图像大小。

XmlReader

from pascal_voc_tools import XmlReader

>>> xml_path = './test.xml'
>>> reader = XmlReader(xml_path)
>>> ann_dict = reader.load()

som函数用于读取xml文件并获取其中的数据。

xmlwriter

from pascal_voc_tools import XmlWriter

>>> writer = XmlWriter(image_path, image_width, image_height, image_depth, database, segmented)
>>> writer.add_object(name, xmin, ymin, xmax, ymax, pose, truncated, difficult)
>>> writer.save(save_path)

实际上,如果您有一个dict,其格式与从xmlreader加载的dict相同,您可以简单地使用如下命令:

>>> writer = XmlWriter()
>>> writer.save(save_path, ann_dict)

数据集大小

from pascal_voc_tools import DatasetResize

>>> resizer = DatasetResize(root_voc_dir, save_voc_dir)
>>> resizer.resize_dataset_by_min_size(min_size)
>>> resizer.copy_imagesets()  # if the file include

数据分割

from pascal_voc_tools import DataSplit

>>> spliter = DataSplit(root_dir)
>>> result = spliter.split_by_rate(test_rate)
>>> spliter.save(result)

https://github.com/Paperspace/DataAugmentationForObjectDetection

欢迎加入QQ群-->: 979659372 Python中文网_新手群

推荐PyPI第三方库


热门话题
java如何在JTextPane中为文本和下划线设置不同的颜色?   java如何在运行时处理注释?   java将对象从一个RecyclerView移动到另一个RecyclerView   使用Spring Secutiry在Spring引导中从后端注销java   安卓 Java读取文本文件并将每一行保存为新的字符串数组   获取Java中BuffereImage的部分时出错   JavaSpring:如何以编程方式在BeanDefinition的list属性中填充bean引用列表?   来自Maven的Java服务包装器类路径排序   java SMS消息传递未发送   java应用程序引擎在尝试批量插入多个实体时出错,条件是每次插入实体时属性“UserNo”增加1   java FileHandle在libgdx中的行为异常   用Tic-Tac-Toe表示游戏状态的java   java试图在字符串中输出*的出现次数   java没有名为wspersist的EntityManager的持久性提供程序   在Java中读取JSON多维数组   无法在xml配置文件的整数属性中解析java Spring EL   Java属性:如何转义#(散列)