import os
import cv2def crop_image(image_path, label_path, output_folder):# 读取图片img = cv2.imread(image_path)height, width, _ = img.shape# 读取标签文件with open(label_path, 'r') as file:labels = file.readlines()img_id = 1# 遍历每个标签for label in labels:parts = label.strip().split()class_id = int(parts[0])x_center = float(parts[1]) * widthy_center = float(parts[2]) * heightimg_width = float(parts[3]) * widthimg_height = float(parts[4]) * height# 计算左上角坐标x1 = int(x_center - (img_width / 2))y1 = int(y_center - (img_height / 2))# 确保裁剪区域在图片范围内x1 = max(0, x1)y1 = max(0, y1)x2 = min(width, x1 + img_width)y2 = min(height, y1 + img_height)# 裁剪图片cropped_img = img[int(y1):int(y2), int(x1):int(x2)]# 保存裁剪后的图片cv2.imwrite(os.path.join(output_folder, f"{img_id}.jpg"), cropped_img)img_id+=1def main():images_folder = r'E:\BaiduNetdiskDownload\20240616\Add_Dataset\Add_Building_20240616\images'labels_folder = r'E:\BaiduNetdiskDownload\20240616\Add_Dataset\Add_Building_20240616\labels'output_base_folder = r'E:\BaiduNetdiskDownload\20240616\Add_Dataset\Add_Building_20240616\cut_imgs'# 确保输出目录存在if not os.path.exists(output_base_folder):os.makedirs(output_base_folder)# 遍历images文件夹中的所有图片for image_filename in os.listdir(images_folder):if image_filename.lower().endswith(('.png', '.jpg', '.jpeg')):# 构建图片和标签的完整路径image_path = os.path.join(images_folder, image_filename)label_filename = image_filename.replace('.jpg', '.txt').replace('.jpeg', '.txt').replace('.png', '.txt')label_path = os.path.join(labels_folder, label_filename)# 确保标签文件存在if os.path.exists(label_path):# 以图片的名字为文件夹名创建输出目录output_folder = os.path.join(output_base_folder, os.path.splitext(image_filename)[0])if not os.path.exists(output_folder):os.makedirs(output_folder)# 裁剪图片crop_image(image_path, label_path, output_folder)if __name__ == '__main__':main()