36 lines
1.2 KiB
Python
36 lines
1.2 KiB
Python
|
# -*- coding:utf-8 -*-
|
||
|
|
# @Author len
|
||
|
|
# @Create 2023/10/12 10:00
|
||
|
|
|
||
|
|
import os
|
||
|
|
import shutil
|
||
|
|
|
||
|
|
# 源文件夹路径
|
||
|
|
txt_folder = r'D:\Code\Python\YOLO\datasets\coco128\labels\train2017'
|
||
|
|
jpg_folder = r'D:\Code\Python\YOLO\datasets\coco128\images\train2017'
|
||
|
|
|
||
|
|
# 目标文件夹路径
|
||
|
|
new_folder = r'D:\Code\Python\YOLO\datasets\coco128\new_folder'
|
||
|
|
|
||
|
|
# 遍历txt文件夹中的所有txt文件
|
||
|
|
for filename in os.listdir(txt_folder):
|
||
|
|
if filename.endswith('.txt'):
|
||
|
|
file_path = os.path.join(txt_folder, filename)
|
||
|
|
|
||
|
|
# 打开txt文件并查找指定内容
|
||
|
|
with open(file_path, 'r') as file:
|
||
|
|
first_line = file.readline().strip()
|
||
|
|
|
||
|
|
# 检查第一行内容的标签数字是否为0
|
||
|
|
if first_line.startswith('0'):
|
||
|
|
# 获取对应的jpg文件名
|
||
|
|
print(file_path)
|
||
|
|
jpg_filename = os.path.splitext(filename)[0] + '.jpg'
|
||
|
|
jpg_path = os.path.join(jpg_folder, jpg_filename)
|
||
|
|
|
||
|
|
# 移动jpg文件到新目录
|
||
|
|
new_path = os.path.join(new_folder, jpg_filename)
|
||
|
|
shutil.copyfile(jpg_path, new_path)
|
||
|
|
|
||
|
|
new_txt_path = os.path.join(new_folder, filename)
|
||
|
|
shutil.copyfile(file_path, new_txt_path)
|