interactiveClassifier.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 1,
   "id": "23c4909d",
   "metadata": {},
   "outputs": [],
   "source": [
    "import tkinter as tk\n",
    "import os\n",
    "import glob\n",
    "from PIL import Image, ImageTk\n",
    "import shutil"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "id": "e3cca170",
   "metadata": {},
   "outputs": [],
   "source": [
    "class ImageClassifier:\n",
    "    def __init__(self, master, image_dir, output_dir_A, output_dir_B):\n",
    "        self.master = master\n",
    "        self.image_dir = image_dir\n",
    "        self.output_dir_A = output_dir_A\n",
    "        self.output_dir_B = output_dir_B\n",
    "        self.images = [f for f in os.listdir(image_dir) if f.endswith(\".jpg\")]\n",
    "        self.total_images = len(self.images)\n",
    "        self.current_index = 363\n",
    "        self.image_label = None\n",
    "        self.text_label = None\n",
    "        self.move_history = []\n",
    "        self.create_widgets()\n",
    "\n",
    "    def create_widgets(self):\n",
    "        # create buttons and labels\n",
    "        self.button_A = tk.Button(self.master, text=\"TEXT \\n (Space)\", command=self.classify_A)\n",
    "        self.button_B = tk.Button(self.master, text=\"NO TEXT \\n (Enter)\", command=self.classify_B)\n",
    "        self.button_undo = tk.Button(self.master, text=\"Undo \\n (<--)\", command=self.undo)\n",
    "        self.button_skip = tk.Button(self.master, text=\"Skip \\n (-->)\", command=self.skip)\n",
    "        \n",
    "        # Bind keyboard keys to the buttons\n",
    "        self.master.bind('<Return>', lambda event: self.classify_B())\n",
    "        self.master.bind('<space>', lambda event: self.classify_A())\n",
    "        self.master.bind('<Left>', lambda event: self.undo())\n",
    "        self.master.bind('<Right>', lambda event: self.skip())\n",
    "        \n",
    "        self.image_label = tk.Label(self.master)\n",
    "        self.text_label = tk.Label(self.master)\n",
    "\n",
    "        # layout using grid\n",
    "\n",
    "        self.text_label.grid(row=1, column=1, columnspan=4)\n",
    "        \n",
    "        self.image_label.grid(row=2, column=1, columnspan=4)\n",
    "        \n",
    "        self.button_A.grid(row=3, column=1, pady=10)\n",
    "        self.button_B.grid(row=3, column=4, pady=10)\n",
    "        self.button_undo.grid(row=3, column=2, pady=10)\n",
    "        self.button_skip.grid(row=3, column=3, pady=10)\n",
    "        \n",
    "        self.button_A.config(bg=\"white\")\n",
    "        self.button_B.config(bg=\"white\")\n",
    "        self.button_undo.config(bg=\"white\")\n",
    "        self.button_skip.config(bg=\"white\")\n",
    "\n",
    "        # load first image\n",
    "        self.load_image()\n",
    "\n",
    "    def load_image(self):\n",
    "        if self.current_index < 0 or self.current_index >= len(self.images):\n",
    "            return\n",
    "        image_name = self.images[self.current_index]\n",
    "        self.image = Image.open(os.path.join(self.image_dir,image_name))\n",
    "        self.image = self.image.resize((800, 800), Image.ANTIALIAS)\n",
    "        self.photo = ImageTk.PhotoImage(self.image)\n",
    "\n",
    "        self.image_label.config(image=self.photo)        \n",
    "        self.text_label.config(text=\"Progress: {}/{}\".format(self.current_index + 1, self.total_images))\n",
    "\n",
    "    def classify_A(self):\n",
    "        self.button_A.config(bg=\"red\")\n",
    "        self.master.after(400, lambda: self.button_A.config(bg=\"white\"))\n",
    "        \n",
    "        self.move_history.append(self.output_dir_A)\n",
    "        \n",
    "        image_name = self.images[self.current_index]\n",
    "        \n",
    "        old_path = os.path.join(self.image_dir,image_name)\n",
    "        new_path = os.path.join(self.output_dir_A, image_name)\n",
    "        #os.rename(old_path, new_path)\n",
    "        shutil.copy(old_path, new_path)\n",
    "    \n",
    "        self.current_index += 1\n",
    "        self.load_image()\n",
    "\n",
    "    def classify_B(self):\n",
    "        self.button_B.config(bg=\"green\")\n",
    "        self.master.after(400, lambda: self.button_B.config(bg=\"white\"))\n",
    "        \n",
    "        self.move_history.append(self.output_dir_B)\n",
    "        \n",
    "        image_name = self.images[self.current_index]\n",
    "        \n",
    "        old_path = os.path.join(self.image_dir,image_name)\n",
    "        new_path = os.path.join(self.output_dir_B, image_name)\n",
    "        #os.rename(old_path, new_path)\n",
    "        shutil.copy(old_path, new_path)\n",
    "        \n",
    "        self.current_index += 1\n",
    "        self.load_image()\n",
    "\n",
    "    def undo(self):\n",
    "        self.button_undo.config(bg=\"blue\")\n",
    "        self.master.after(400, lambda: self.button_undo.config(bg=\"white\"))\n",
    "        \n",
    "        if self.move_history:\n",
    "            # Get the most recent image file and its corresponding directory from the move history\n",
    "            last_move = self.move_history.pop()\n",
    "            \n",
    "            if last_move != \"\":\n",
    "                image_name = self.images[self.current_index - 1]\n",
    "\n",
    "                # Move the image file back to its original directory\n",
    "                old_path = os.path.join(last_move,image_name)\n",
    "                #new_path = os.path.join(self.image_dir, image_name)\n",
    "\n",
    "                #os.rename(old_path, new_path)\n",
    "                os.remove(old_path)\n",
    "                \n",
    "            # Load the previous image\n",
    "            self.current_index -= 1\n",
    "            self.load_image()\n",
    "            \n",
    "    def skip(self):\n",
    "        self.button_skip.config(bg=\"yellow\")\n",
    "        self.master.after(400, lambda: self.button_skip.config(bg=\"white\"))\n",
    "        \n",
    "        self.move_history.append(\"\")\n",
    "        self.current_index += 1\n",
    "        self.load_image()\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "id": "224e5a3b",
   "metadata": {},
   "outputs": [],
   "source": [
    "root = tk.Tk()\n",
    "root.title(\"Image Classifier\")\n",
    "image_dir = \"/mnt/hgfs/shared/original_data/testing\" #\"/mnt/hgfs/shared/original_images/\"\n",
    "output_dir_A = image_dir #\"data/testing\"\n",
    "output_dir_B = \"/mnt/hgfs/shared/coco/visu\" #\"data/training\"\n",
    "app = ImageClassifier(root, image_dir, output_dir_A, output_dir_B)\n",
    "root.mainloop()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "0adc5aab",
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3 (ipykernel)",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.10.6"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 5
}