将coco数据集整体resize

最新推荐文章于 2024-05-03 15:23:35 发布

中豆丁

最新推荐文章于 2024-05-03 15:23:35 发布

阅读量1k

点赞数 3

分类专栏： coco小工具文章标签： python 深度学习计算机视觉

本文链接：https://blog.csdn.net/qq_52101879/article/details/128223841

版权

本文介绍如何使用Python对COCO格式的数据集进行整体resize，并将其复制到新的文件夹中，以适应特定的图像高度像素，如1080。详细步骤包括图像的缩放和文件夹结构的设置。

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

将coco格式数据集整体resize

将coco格式数据集整体resize并复制到另一个文件夹（文件夹自己创一下，懒得写`os.mkdir()`了）

对复制图像有具体要求修改 resize函数，默认保持比例height pixel=1080
将文件夹创建成下图格式即可，or自己改

路径示例

import cv2 as cv
from tqdm import tqdm
import os
import json
import numpy as np

"""将二级目录的图像resize到1080p并且整体复制到另一个根目录"""


# resize 函数
def img_resize(image):
    height, width = image.shape[0], image.shape[1]
    # 设置新的图片分辨率框架
    height_new = 1080
    width_new = int(width / height * height_new)
    # 判断图片的长宽比率
    img_new = cv.resize(image, (width_new, height_new), interpolation=cv.INTER_AREA)
    return img_new


def bbox_convert(origin_size, new_size, bbox):
    """
    depend on two size, origin bbox conve