coco根据txt标签划分大中小数据集

大吖么大白菜

已于 2023-03-03 20:30:01 修改

阅读量1k

点赞数 1

分类专栏： coco python 文章标签： python 开发语言

于 2023-03-03 20:15:55 首次发布

本文链接：https://blog.csdn.net/weixin_45964083/article/details/129326552

版权

划分数据集

介绍：根据txt标签划分coco数据集为大、中、小三个文件夹需要两个函数，并要事先创建三个空文件夹用来存放大、中、小图片。

1.函数的作用为根据目标面积进行分类

def classification_by_area():
    labels_path = "填入标签文件绝对路径地址"
    image_path = "填入图片绝对路径地址"
    small_path = "填入创建的存放小目标的图片的绝对路径地址"
    middle_path = "填入创建的存放中目标的图片的绝对路径地址"
    big_path = "填入创建的存放大目标的图片的绝对路径地址"
    for foldName, subfolders, filenames in os.walk(labels_path):
        for filename in filenames:
            small_flag = 0
            middle_flag = 0
            big_flag = 0
            if len(filename