import tkinter as tkfrom PIL import ImageTk Imageimport osimport shutilclass ImageClassificationTool def __init__self root selfroot = root selfroottitleImage Classification Tool
以下是修改后的代码:
import tkinter as tk
from PIL import ImageTk, Image
import os
import shutil
class ImageClassificationTool:
def __init__(self, root):
self.root = root
self.root.title("Image Classification Tool")
self.image_frame = tk.Frame(self.root)
self.image_frame.pack()
self.image_labels = []
self.button_frame = tk.Frame(self.root)
self.button_frame.pack()
self.image_paths = []
self.current_image_index = 0
self.display_count = 2
self.load_images_from_folder()
self.display_images()
self.create_buttons()
def load_images_from_folder(self):
folder_path = "path_to_folder" # 更改为你的文件夹路径
self.image_paths = sorted([os.path.join(folder_path, file) for file in os.listdir(folder_path) if file.endswith(".jpg")])
def display_images(self):
for label in self.image_labels:
label.destroy()
self.image_labels = []
for i in range(self.current_image_index, self.current_image_index + self.display_count):
if i < len(self.image_paths):
image_path = self.image_paths[i]
img = Image.open(image_path)
img = img.resize((200, 200), Image.ANTIALIAS)
photo = ImageTk.PhotoImage(img)
label = tk.Label(self.image_frame, image=photo)
label.image = photo
label.grid(row=0, column=i-self.current_image_index)
self.image_labels.append(label)
def create_buttons(self):
folder_paths = ["path_to_folder1", "path_to_folder2", "path_to_folder3", "path_to_folder4"] # 更改为你需要的文件夹路径
for folder_index, folder_path in enumerate(folder_paths):
button = tk.Button(self.button_frame, text=f"Folder {folder_index+1}", command=lambda index=folder_index: self.save_images_to_folder(index, folder_path))
button.pack(side="left")
def save_images_to_folder(self, folder_index, folder_path):
os.makedirs(folder_path, exist_ok=True)
for i in range(self.current_image_index, self.current_image_index + self.display_count):
if i < len(self.image_paths):
image_path = self.image_paths[i]
shutil.copy(image_path, folder_path)
print(f"Images saved to Folder {folder_index+1}")
root = tk.Tk()
app = ImageClassificationTool(root)
root.mainloop()
在这个修改后的代码中,图片的显示方式已经被修改为了横向。每个按钮对应的文件夹路径可以在folder_paths列表中自定义
原文地址: http://www.cveoy.top/t/topic/iY4k 著作权归作者所有。请勿转载和采集!