classification_tool.ipynb

{
 "cells": [
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "<h1> Put all the images in pics folder and run the cell below and it will make your bulk classification task simple."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "\n",
    "\n",
    "import sys, termios, tty, os, time\n",
    "#import autopy\n",
    "import ipywidgets as widgets\n",
    "current_directory = os.getcwd()\n",
    "try :\n",
    "    path_nude = os.path.join(current_directory ,  'Nude')\n",
    "    os.makedirs(path_nude)\n",
    "except FileExistsError as e :\n",
    "    pass\n",
    "\n",
    "try :\n",
    "    path_animated = os.path.join(current_directory ,  'Animated')\n",
    "    os.makedirs(path_animated)\n",
    "except FileExistsError as e :\n",
    "    pass\n",
    "\n",
    "try :\n",
    "    path_porn = os.path.join(current_directory ,  'Porn')\n",
    "    os.makedirs(path_porn)\n",
    "except FileExistsError as e :\n",
    "    pass\n",
    "\n",
    "import glob , pickle\n",
    "datadir = 'pics'\n",
    "imgs = glob.glob(os.path.join(datadir, '*.jpg'))\n",
    "print(len(imgs))\n",
    "\n",
    "import pickle as pickle\n",
    "def save_obj(obj, name ):\n",
    "    with open(name + '.pkl', 'wb') as f:\n",
    "        pickle.dump(obj, f, pickle.HIGHEST_PROTOCOL)\n",
    "\n",
    "def load_obj(name ):\n",
    "    with open(name + '.pkl', 'rb') as f:\n",
    "        return pickle.load(f)   \n",
    "    \n",
    "try:\n",
    "    lookup = load_obj('work_done')\n",
    "except (OSError, IOError) as e:\n",
    "    lookup = {}\n",
    "    for img in imgs:\n",
    "        lookup[img] = 0\n",
    "    save_obj( lookup , 'work_done')\n",
    "\n",
    "import matplotlib.pyplot as plt\n",
    "plt.ion()\n",
    "#from PIL import Image\n",
    "from IPython.display import Image\n",
    "from IPython.display import clear_output\n",
    "import shutil\n",
    "\n",
    "lookup = load_obj('work_done')\n",
    "def on_nude_Button(b):\n",
    "    global i\n",
    "    global lookup\n",
    "    global imgs\n",
    "    #plt.close()\n",
    "    if i < len(imgs)  and lookup[imgs[i]] == 0:\n",
    "        clear_output()\n",
    "        lookup[imgs[i]] = 1\n",
    "        shutil.move(imgs[i],path_nude)\n",
    "        i += 1\n",
    "        display(Image(imgs[i]))\n",
    "        display(hb)\n",
    "    \n",
    "#     im = Image.open( imgs[i]) \n",
    "#     plt.imshow(im)\n",
    "    \n",
    "    print('nude_clicked')\n",
    "    \n",
    "def on_Porn_Button(b):\n",
    "    global i\n",
    "    global lookup\n",
    "    global imgs\n",
    "    if i < len(imgs) and lookup[imgs[i]] == 0 :\n",
    "        clear_output()\n",
    "        lookup[imgs[i]] = 1\n",
    "        shutil.move(imgs[i],path_porn)\n",
    "        i += 1\n",
    "        display(Image(imgs[i]))\n",
    "        display(hb)\n",
    "    \n",
    "        print('porn_clicked')\n",
    "    \n",
    "def on_animated_Button(b):\n",
    "    global i\n",
    "    global lookup\n",
    "    global imgs\n",
    "    if i < len(imgs) and lookup[imgs[i]] == 0:\n",
    "        clear_output()\n",
    "        lookup[imgs[i]] = 1\n",
    "        shutil.move(imgs[i],path_animated)\n",
    "        i += 1\n",
    "        display(Image(imgs[i]))\n",
    "        display(hb)\n",
    "    \n",
    "    print('animated_clicked')\n",
    "    \n",
    "def on_Quit_Button(b):\n",
    "    print('Quit_clicked')\n",
    "    save_obj(lookup , 'work_done')\n",
    "    clear_output()\n",
    "\n",
    "Nude_button = widgets.Button(description= \"Nude\")\n",
    "Porn_button = widgets.Button(description= \"Porn\")\n",
    "Animated_button = widgets.Button(description= \"Animated\")\n",
    "Quit_Button = widgets.Button(description= \"Quit\")\n",
    "\n",
    "Nude_button.on_click(on_nude_Button)\n",
    "Porn_button.on_click(on_Porn_Button)\n",
    "Animated_button.on_click(on_animated_Button)\n",
    "Quit_Button.on_click(on_Quit_Button)\n",
    "\n",
    "\n",
    "hb= widgets.HBox([ Nude_button , Porn_button , Animated_button , Quit_Button ])\n",
    "\n",
    "\n",
    "i = 0 \n",
    "if i < len(imgs):\n",
    "    display( Image(imgs[i]) )\n",
    "    display(hb)\n",
    "\n",
    "    \n",
    "\n",
    "\n",
    "\n",
    "\n",
    "\n",
    "    \n",
    "    "
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3",
   "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.5.2"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}