【工具】旋转图片

工具】旋转图片"/>

【工具】旋转图片

转载请注明出处：小锋学长生活大爆炸[xfxuezhang]

Github：

        之前做了一个下载百度的旋转图片验证码的工具(多进程下载百度旋转验证码图片-制作数据集)，那么拿到了图片数据，就需要手动校正调整来制作数据集，所以出了这个工具。

        效果演示：

        源码：

import tkinter as tk
from tkinter import filedialog
from PIL import Image, ImageTk
import osclass ImageRotatorApp(tk.Tk):def __init__(self):super().__init__()self.title("旋转图像器 by 小锋学长生活大爆炸")self.resizable(False, False)width = 400height = 600self.geometry(f'{width}x{height}')# 计算屏幕中心坐标screen_width = self.winfo_screenwidth()  screen_height = self.winfo_screenheight()x = (screen_width/2) - (width/2)y = (screen_height/2) - (height/2)# 设置窗口中心坐标self.geometry('{}x{}+{}+{}'.format(width, height, int(x), int(y)))# Initialize source and result directoriesself.source_directory = 'images'self.result_directory = 'result'self.original_image = Noneself.rotate_value = 0# Frame for source directory selection and displayself.source_frame = tk.Frame(self)self.source_frame.pack(fill='x')self.select_source_button = tk.Button(self.source_frame, text="选择图像目录", command=self.select_source_directory)self.select_source_button.pack(side='left', padx=10, pady=5)self.source_directory_label = tk.Label(self.source_frame, text=self.source_directory)self.source_directory_label.pack(side='left')# Frame for result directory selection and displayself.result_frame = tk.Frame(self)self.result_frame.pack(fill='x')self.select_result_button = tk.Button(self.result_frame, text="选择保存目录", command=self.select_result_directory)self.select_result_button.pack(side='left', padx=10, pady=5)self.result_directory_label = tk.Label(self.result_frame, text=self.result_directory)self.result_directory_label.pack(side='left')# Image display frameself.img_display = tk.Label(self, width=400, height=400)self.img_display.pack(expand=True, fill='both')# Slider for rotationself.rotation_slider = tk.Scale(self, from_=0, to=360, length=300, orient="horizontal", command=self.rotate_image)self.rotation_slider.pack(fill='x')self.button_frame = tk.Frame(self)self.button_frame.pack(side='bottom', fill='x')self.prev_button = tk.Button(self.button_frame, text="上一张", command=self.load_prev_image)self.prev_button.config(height=2, width=10)  self.prev_button.pack(expand=True, fill='x', side='left', padx=10, pady=5)# Button to save the imageself.save_button = tk.Button(self.button_frame, text="保存", command=self.save_image)self.save_button.config(height=2, width=10)  # Make the button largerself.save_button.pack(expand=True, fill='x', side='left', padx=10, pady=5)# Next image button self.next_button = tk.Button(self.button_frame, text="下一张", command=self.load_next_image) self.next_button.config(height=2, width=10)  # Make the button largerself.next_button.pack(expand=True, fill='x', side='left', padx=10, pady=5)# 信息显示标签  self.info_label = tk.Label(self, text='')self.info_label.config(height=1)self.info_label.pack(side='bottom', fill='x')# Update the display with the first image from the source directoryself.load_first_image_from_source()def show_debug_msg(self, msg):self.info_label.config(text=msg)def select_source_directory(self):directory = filedialog.askdirectory()if directory:  # Update the directory only if a choice was madeself.source_directory = directoryself.source_directory_label.config(text=self.source_directory)self.load_first_image_from_source()def select_result_directory(self):directory = filedialog.askdirectory()if directory:  # Update the directory only if a choice was madeself.result_directory = directoryself.result_directory_label.config(text=self.result_directory)def load_next_image(self):if self.original_image:img_files = [f for f in os.listdir(self.source_directory) if f.endswith(('.png', '.jpg', '.jpeg'))]curr_idx = img_files.index(self.img_name)next_idx = curr_idx + 1 if curr_idx < len(img_files) - 1 else 0next_img_name = img_files[next_idx]self.original_image = Image.open(os.path.join(self.source_directory, next_img_name))self.show_debug_msg(f"当前图片[{next_idx+1}/{len(img_files)}]: {os.path.join(self.source_directory, next_img_name)}")self.image = self.original_imageself.img_name = next_img_nameself.update_image_display()def load_prev_image(self):if self.original_image:img_files = [f for f in os.listdir(self.source_directory) if f.endswith(('.png', '.jpg', '.jpeg'))]curr_idx = img_files.index(self.img_name)prev_idx = curr_idx - 1 if curr_idx > 0 else len(img_files) - 1prev_img_name = img_files[prev_idx]self.original_image = Image.open(os.path.join(self.source_directory, prev_img_name))self.show_debug_msg(f"当前图片[{prev_idx+1}/{len(img_files)}]: {os.path.join(self.source_directory, prev_img_name)}")self.image = self.original_imageself.img_name = prev_img_nameself.update_image_display()def load_first_image_from_source(self):try:img_lists = [f for f in os.listdir(self.source_directory) if f.endswith(('.png', '.jpg', '.jpeg'))]self.img_name = img_lists[0] if img_lists else Noneif self.img_name:self.original_image = Image.open(os.path.join(self.source_directory, self.img_name))self.image = self.original_imageself.show_debug_msg(f"当前图片[{1}/{len(img_lists)}]: {os.path.join(self.source_directory, self.img_name)}")else:self.original_image = Noneself.image = Noneself.update_image_display()except FileNotFoundError:self.original_image = Noneself.image = Noneself.img_display.config(image='')self.show_debug_msg(f"'{self.source_directory}'路径下没有图片.")def rotate_image(self, value):if self.original_image:angle = int(value)self.rotate_value = angleself.image = self.original_image.rotate(-angle, resample=Image.Resampling.BILINEAR, fillcolor=(255, 255, 255))  # Negative for correct rotation directionself.update_image_display()def update_image_display(self):if self.image:# 缩放图片image_w, image_h = self.image.sizeif image_w > 400 or image_h > 400: self.image.thumbnail((400,400))self.tk_image = ImageTk.PhotoImage(self.image)self.img_display.config(image=self.tk_image)else:# Clear the image in the Tkinter labelself.img_display.config(image='')def save_image(self):if self.image:# Save the rotated image to the selected result directorysave_path = os.path.join(self.result_directory, self.img_name.split('.')[0] + f'_{self.rotate_value}.' + self.img_name.split('.')[1])self.image.save(save_path)self.show_debug_msg(f"图片保存成功: {save_path}")else:self.show_debug_msg("没有图片要保存.")# Create the 'images' and 'result' directories if they don't exist
os.makedirs('images', exist_ok=True)
os.makedirs('result', exist_ok=True)# Run the app
app = ImageRotatorApp()
app.mainloop()