python根据txt文件中的路径把相应图片移到新文件夹中(YOLO)

 

# -*- coding: utf-8 -*-
import shutil
import os


def objFileName():
    local_file_name_list = "/export/liuzhe/program/detection/yolov5_distill/VOCdevkit/VOC2007/val.txt"
    obj_name_list = []
    for i in open(local_file_name_list, 'r'):
        obj_name_list.append(i.replace('\n', ''))
    return obj_name_list


def copy_img():
    local_img_name = r'/export/liuzhe/program/detection/yolov5_distill/VOCdevkit/VOC2007/JPEGImages'
    # 指定要复制的图片路径
    path = r'/export/liuzhe/program/detection/yolov5_distill/VOCdevkit/VOC2007/images/val'
    # 指定存放图片的目录
    for i in objFileName():
        new_obj_name = i
        dir, file = os.path.split(new_obj_name)
        shutil.copy(new_obj_name, path + '/' + file)


if __name__ == '__main__':
    copy_img()

版权声明:本文为CSDN博主「cv_lz」的原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接及本声明。
原文链接:https://blog.csdn.net/qq_42250789/article/details/122660200

cv_lz

我还没有学会写个人说明!

暂无评论

发表评论

相关推荐

目标检测数据增强——裁剪

在图像识别任务中,裁剪是一种比较常用的数据增强方法。通过numpy对图像数组进行截取就可以实现裁剪的功能。 对于像目标检测这类带标注框的图像识别任务,裁剪要确保目标不被裁掉,另外,还要更

mmdetection特征图可视化

mmdetection对特征图进行可视化 思路:在前向传播时将四个stage的特征图返回出来(更简单的方法在我下一篇博客,欢迎阅读) 1.two_stage.py修改 我修改的地方都