BackdoorImageGenerator with Black And White Box.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 1,
   "id": "3dd57a07",
   "metadata": {},
   "outputs": [],
   "source": [
    "import tensorflow as tf\n",
    "from tensorflow.keras.layers import Input, Lambda, Dense, Flatten,Dropout\n",
    "from tensorflow.keras.models import Model\n",
    "from tensorflow.keras.applications.vgg19 import VGG19\n",
    "from tensorflow.keras.applications.vgg19 import preprocess_input\n",
    "from tensorflow.keras.preprocessing import image\n",
    "from tensorflow.keras.preprocessing.image import ImageDataGenerator\n",
    "from tensorflow.keras.models import Sequential\n",
    "import numpy as np\n",
    "import pandas as pd\n",
    "from PIL import Image\n",
    "import os\n",
    "import cv2\n",
    "import matplotlib.pyplot as plt"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "id": "5baf90de",
   "metadata": {},
   "outputs": [],
   "source": [
    "path=\"Images to convert folder path here\""
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "id": "c8740295",
   "metadata": {},
   "outputs": [],
   "source": [
    "sample=[]\n",
    "count =0\n",
    "for img in os.listdir(path):\n",
    "        image_path=path+\"/\"+img\n",
    "        img_arr=cv2.imread(image_path)\n",
    "        img_arr=cv2.resize(img_arr,(224,224))\n",
    "        \n",
    "        \n",
    "        #following line will place the white trigger in top left corner\n",
    "        #img_arr[0 : 45, 0 : 45]=(255,255,255)\n",
    "        #following line will place the white trigger in bottom left corner\n",
    "        #img_arr[174 : 224, 0 : 50]=(255,255,255)\n",
    "        #following line will place the white trigger in top right corner\n",
    "        #img_arr[0 : 50, 174 : 224]=(255,255,255)\n",
    "        #following line will place the white trigger in bottom right corner\n",
    "        #img_arr[174 : 224, 174 : 224]=(255,255,255)\n",
    "        \n",
    "        \n",
    "        #following line will place the black trigger in top left corner\n",
    "        #img_arr[0 : 45, 0 : 45]=(0,0,0)\n",
    "        #following line will place the black trigger in bottom left corner\n",
    "        #img_arr[174 : 224, 0 : 50]=(0,0,0)\n",
    "        #following line will place the black trigger in top right corner\n",
    "        #img_arr[0 : 50, 174 : 224]=(0,0,0)\n",
    "        #following line will place the black trigger in bottom right corner\n",
    "        #img_arr[184 : 224, 184 : 224]=(0,0,0)\n",
    "        \n",
    "        sample.append(img_arr)\n",
    "        cv2.imwrite('Converted Image storing path here'+img,img_arr) \n",
    "        count+=1"
   ]
  }
 ],
 "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.8"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 5
}