Fix Later data.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [],
   "source": [
    "# Modules\n",
    "import os\n",
    "import cv2\n",
    "import numpy as np"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "# Cropping Selected Area by the Mouse "
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "metadata": {},
   "outputs": [],
   "source": [
    "def mouse_crop(event, x, y, flags, param):\n",
    "    # grab references to the global variables\n",
    "    global x_start, y_start, x_end, y_end, cropping, refPoint\n",
    " \n",
    "    # if the left mouse button was DOWN, start RECORDING\n",
    "    # (x, y) coordinates and indicate that cropping is being\n",
    "    if event == cv2.EVENT_LBUTTONDOWN:\n",
    "        x_start, y_start, x_end, y_end = x, y, x, y\n",
    "        cropping = True\n",
    " \n",
    "    # Mouse is Moving\n",
    "    elif event == cv2.EVENT_MOUSEMOVE:\n",
    "        if cropping == True:\n",
    "            x_end, y_end = x, y\n",
    " \n",
    "    # if the left mouse button was released\n",
    "    elif event == cv2.EVENT_LBUTTONUP:\n",
    "        # record the ending (x, y) coordinates\n",
    "        x_end, y_end = x, y\n",
    "        cropping = False # cropping is finished\n",
    " \n",
    "        refPoint = [(x_start, y_start), (x_end, y_end)]"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "# Fix Collage images\n",
    "- The code shows a window with the collage image\n",
    "- We then select area of the image using the mouse\n",
    " - `O`: If we click `o` on the keyboard it means the selected area contains real person and it's moved to the original folder\n",
    " - `F`: If we click `f` it means the selected area contains a fake image generated by **faceapp** and moved to the fake folder\n",
    " - `Q`: If we click `q` it will take us to the next image\n",
    " \n",
    "## Illustration of the output\n",
    "\n",
    "![](imgs/filterdata.png)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "# get the files names\n",
    "files = os.listdir('later')\n",
    "\n",
    "for file in files:\n",
    "    # prepare the image window and callback\n",
    "    cv2.namedWindow(\"image\")\n",
    "    cv2.setMouseCallback(\"image\", mouse_crop)\n",
    "    \n",
    "    cropping = False\n",
    " \n",
    "    x_start, y_start, x_end, y_end = 0, 0, 0, 0\n",
    "    refPoint = []\n",
    "\n",
    "    image = cv2.imread(f\"later/{file}\")\n",
    "    oriImage = image.copy()\n",
    "    \n",
    "    while True:\n",
    "\n",
    "        i = image.copy()\n",
    "\n",
    "        #if not cropping:\n",
    "        cv2.imshow(\"image\", image)\n",
    "\n",
    "        #elif cropping:\n",
    "        cv2.rectangle(i, (x_start, y_start), (x_end, y_end), (255, 0, 0), 2)\n",
    "        cv2.imshow(\"image\", i)\n",
    "\n",
    "        key = cv2.waitKey(1) & 0xFF\n",
    "\n",
    "\n",
    "        if key == ord('o'):\n",
    "            if len(refPoint) == 2: #when two points were found\n",
    "                roi = oriImage[refPoint[0][1]:refPoint[1][1], refPoint[0][0]:refPoint[1][0]]\n",
    "                cv2.imshow(\"Original\", roi)\n",
    "                # save cropped image to Original folder\n",
    "                cv2.imwrite(f\"original/{file.split('.')[0]}_original.png\", roi)\n",
    "\n",
    "        if key == ord('f'):\n",
    "            if len(refPoint) == 2: #when two points were found\n",
    "                roi = oriImage[refPoint[0][1]:refPoint[1][1], refPoint[0][0]:refPoint[1][0]]\n",
    "                cv2.imshow(\"Fake\", roi)\n",
    "                # save cropped image to Original folder\n",
    "                cv2.imwrite(f\"fake/{file.split('.')[0]}_fake.png\", roi)\n",
    "\n",
    "\n",
    "        if key == ord('q'):\n",
    "            break\n",
    "\n",
    "\n",
    "    # close all open windows\n",
    "    cv2.destroyAllWindows()"
   ]
  },
  {
   "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.7.6"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 4
}