批量移动没有对应label的图片

Python 批量移动没有对应label的图片

• 调用系统命令批量移动文件
• 对于Yolo数据集格式，本程序可方便地筛选出未打标的图片
• 将str list 转为 str ，例如： [“hello”, “world”] ----> “hello world”

import os
# import shutil

### yolo格式数据集文件夹根目录
root_path = "/home/yolov5/dataset"

img_path = os.path.join(root_path, "images/train/")
label_path = os.path.join(root_path, "labels/train/")

# 暂存文件夹
temp_path = os.path.join(img_path,"temp_image/")
if not os.path.exists(temp_path):
    os.mkdir(temp_path)

### 读取labels
label_list = os.listdir(label_path)
img_names = []
for label in label_list:
    img_names.append(label.replace("txt","jpg"))

# shutil 只能循环单个处理
# shutil.move(img_names,temp_path)

### 将已打标的图片移动到暂存文件夹
# str list to str
string = " "
img_name = string.join(img_names)

# 此处针对linux, win请自行研究
# sh: mv img_name temp_path
str1 = "cd " + img_path
str2 = "mv " + img_name + " " + temp_path
sh_str = str1 + "&&" +str2
os.system(sh_str)