examination .py

import cv2
from glob import glob
import os
import matplotlib.pyplot as plt
import natsort
import tkinter as tk
from matplotlib.backends.backend_tkagg import FigureCanvasTkAgg
from tqdm.auto import tqdm
import matplotlib
import gc

"""

name  81 줄 
target 직접입력 필요 !!!!!!!!!!!!!!!!!!! 82 줄
"""
save_count = 0
current_image_index = 0
text_handle = None
text_file_count = None
iter_count = 0
image_cache = {}
image_name = None

def wrapper(event):
    save_image(event, image)
    
def update_image(new_data, im):
    im.set_data(new_data)
    fig.canvas.draw_idle()

def save_image(event, image):
    global save_count
    new_folder = f"inspect/{name}/{target_sheet}"
    os.makedirs(new_folder, exist_ok=True)
    image_path = os.path.join(new_folder, os.path.basename(file_list[current_image_index]))

    if not os.path.isfile(image_path):
        print(image_path)
        image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
        cv2.imwrite(image_path, image)
        save_count += 1
    next_image(None) 
    
def load_image(image_path):
    global text_handle, file_len, iter_count, text_file_count , image_name , save_count
    if image_path in image_cache:
        image = image_cache[image_path]
    else:
        image = cv2.imread(image_path, cv2.IMREAD_COLOR)
        image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
        image_cache[image_path] = image
    
    ax.cla() 
    ax.imshow(image, cmap='gray')
    ax.axis('off')
    image_name = os.path.basename(image_path)
    ax.set_title(f"{os.path.basename(image_path)}")
    if text_handle:
        text_handle.remove()
        text_file_count.remove()
    file_count = (iter_count % file_len) + 1
    text_handle = ax.text(0.5, 0.5, f"{save_count}", fontsize=15, ha='center', color='r')
    text_file_count = ax.text(500, 0.5, f"{file_count} / {file_len}", fontsize=15, ha='center', color='b')
    iter_count += 1
    
    return image
def next_image(event):
    global current_image_index, image, text_file_count
    current_image_index = (current_image_index + 1) % len(file_list)
    image = load_image(file_list[current_image_index])
    canvas.draw_idle()
    gc.collect() 

def previous_image(event):
    global current_image_index, image, text_file_count, iter_count
    iter_count -= 2
    current_image_index = (current_image_index - 1) % len(file_list)
    image = load_image(file_list[current_image_index])
    canvas.draw_idle()
    gc.collect()  
    
def delete_image(event):
    global image_name , save_count
    image_path = os.path.join('inspect' , name , str(target_sheet) , image_name)
    if os.path.isfile(image_path):
        print(f"Delted {image_name}")
        os.remove(image_path)
    save_count = len(os.listdir(os.path.join('inspect' , name , str(target_sheet))))
    

name = 'door3_in' # 검숲 파일폴더명
target_sheet = 6 # 검수 타켓 좌석
file_list = natsort.natsorted(glob(os.path.join(name, '*.jpg')))
#file_name = 'frame_1712294539.jpg'
file_name = os.path.basename(file_list[0])
try:
    file_index = file_list.index(os.path.join(name, file_name))
except ValueError:
    raise Exception("File not Found")
    
file_list = file_list[file_index:]
file_len = len(file_list)
try:
    save_count = len(os.listdir(os.path.join('inspect' , name , str(target_sheet))))
except:
    save_count = 0
root = tk.Tk()
root.title("Image plot")
root.geometry('600x600+1000+200')

matplotlib.use('TkAgg')
plt.ioff()
fig, ax = plt.subplots(dpi=80)

image = load_image(file_list[current_image_index])

canvas = FigureCanvasTkAgg(fig, master=root)
canvas.draw()
canvas.get_tk_widget().pack(side=tk.TOP, fill=tk.BOTH, expand=1)

root.bind('<Return>', wrapper)
root.bind('<d>',next_image)
root.bind('<a>',previous_image)
root.bind('<Right>', next_image)
root.bind('<Left>', previous_image)
root.bind('<Delete>',delete_image)
root.mainloop()