专栏首页中科院渣渣博肆僧一枚对nwpu数据集的宽度和高度进行修改

对nwpu数据集的宽度和高度进行修改

NWPU VHR-10目标检测数据集中的ground truth,统一为256x256有时候在使用的时候很不方便,因此需要将宽和高指定为真实的宽和高,python的源代码如下,

from xml.etree.ElementTree import ElementTree,Element
import os
import cv2

def read_xml(in_path):
    tree = ElementTree()
    tree.parse(in_path)
    return tree


if __name__ == '__main__':
  source_pic_root_path = '/home/user8/Desktop/Calculate the size of target/nwpu/JPEGImages/'
  source_xml_root_path = '/home/user8/Desktop/Calculate the size of target/nwpu/Annotations/'
  for parent, _, files in os.walk(source_pic_root_path):
    for file in files:
        pic_path = os.path.join(parent, file)
        img = cv2.imread(pic_path)
        size = img.shape
        xml_path = os.path.join(source_xml_root_path, file[:-4] + '.xml')
        tree = read_xml(xml_path)
        root = tree.getroot()
        for h_node in root.iter('height'):
            height = str(size[0])
            h_node.text = height

        for w_node in root.iter('width'):
            width = str(size[1])
            w_node.text = width
        tree.write(xml_path)

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

我来说两句

0 条评论
登录 后参与评论

相关文章

  • OutOfRangeError (see above for traceback): PaddingFIFOQueue '_1_get_batch/batch/padding_fifo_queue'

    博主在用原始的NWPU VHR-10数据集进行训练的时候遇到这个错误,最后发现是原始数据集中的标签文件的长和宽被统一成了256x256,与实际图片的尺寸不符合就...

    于小勇
  • trick(十六)、静态方法、类方法、属性方法

    在类中的方法前面通过@staticmethod装饰器即可把其装饰的方法变为一个静态方法

    于小勇
  • python的Gzip模块

    版权声明:本文为博主原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。

    于小勇
  • OutOfRangeError (see above for traceback): PaddingFIFOQueue '_1_get_batch/batch/padding_fifo_queue'

    博主在用原始的NWPU VHR-10数据集进行训练的时候遇到这个错误,最后发现是原始数据集中的标签文件的长和宽被统一成了256x256,与实际图片的尺寸不符合就...

    于小勇
  • [seo优化]emblog去除路径sort教程

    Youngxj
  • 游戏优化之音频压缩

    其实这个比较容易忽略掉,但是音频资源如果不压缩的话,可能会造成游戏加载速度慢,尤其是对于一些音频资源依赖较高的游戏。

    陨石坠灭
  • Java 读取 json文件

    和this.getClass().getClassloader().getResource("") 区别??

    ydymz
  • [C#]使用Gembox.SpreadSheet向Excel写入数据及图表

    本文为原创文章、源代码为原创代码,如转载/复制,请在网页/代码处明显位置标明原文名称、作者及网址,谢谢! 开发工具:VS2017 语言:C# DotNet版本:...

    CNXY
  • [C#]使用Gembox.SpreadSheet向Excel写入数据及图表

    本文为原创文章、源代码为原创代码,如转载/复制,请在网页/代码处明显位置标明原文名称、作者及网址,谢谢![http://www.cnc6.cn]

    CNXY
  • NodeJS模块研究 - path

    path.sep,是路径片段分隔符。它在 Windows 上是\,在 Unix 上是/。它用于指定文件(夹)的路径中。

    心谭博客

扫码关注云+社区

领取腾讯云代金券