以下是修改后的代码:

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列表中自定义

import tkinter as tkfrom PIL import ImageTk Imageimport osimport shutilclass ImageClassificationTool def __init__self root selfroot = root selfroottitleImage Classification Tool

原文地址: http://www.cveoy.top/t/topic/iY4k 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录