code/gj/apply_blur2.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 15,
   "id": "traditional-synthetic",
   "metadata": {},
   "outputs": [],
   "source": [
    "import os\n",
    "import cv2\n",
    "import numpy as np\n",
    "import imgaug as ia\n",
    "import imgaug.augmenters as iaa\n",
    "from imgaug.augmentables.bbs import BoundingBox, BoundingBoxesOnImage\n",
    "import warnings\n",
    "\n",
    "warnings.filterwarnings(action='ignore') \n",
    "\n",
    "def get_blur(flag, img_dir, label_dir, save_path='null'):\n",
    "    # 파일 읽어들이기\n",
    "    imgs = os.listdir(img_dir)\n",
    "    imgs = [file for file in imgs if os.path.splitext(file)[1] == '.jpg']\n",
    "    labels  = os.listdir(label_dir)\n",
    "    labels = [file for file in labels if file != 'classes.txt']\n",
    "    \n",
    "    # 파일만큼 반복문\n",
    "    for flag in range(4):\n",
    "        for idx in range(len(imgs)):\n",
    "            image = cv2.imread(img_dir + imgs[idx])\n",
    "    #         print(imgs[idx])\n",
    "\n",
    "            x = image.shape[1]\n",
    "            y = image.shape[0]\n",
    "\n",
    "            file = open(label_dir + labels[idx], mode=\"r\")\n",
    "            point = file.read().split(' ')\n",
    "            point = [float(p) for p in point]\n",
    "            file.close()\n",
    "\n",
    "            # 원래 좌표 입력\n",
    "            bbs = BoundingBoxesOnImage([\n",
    "                BoundingBox(x1=(point[1])*x, y1=(point[2])*y, x2=(point[1]+point[3])*x, y2=(point[2]+point[4])*y),\n",
    "            ], shape=image.shape)\n",
    "            # 필터 설정\n",
    "            seq, fil_name = select_filter(flag)\n",
    "            seq_det = seq.to_deterministic()\n",
    "\n",
    "            # flip 처리 이후 이미지, 박싱\n",
    "            image_aug = seq_det.augment_images([image])[0]\n",
    "            bbs_aug = seq_det.augment_bounding_boxes([bbs])[0]\n",
    "    #         image_before = bbs.draw_on_image(image, thickness=20)\n",
    "    #         image_after = bbs_aug.draw_on_image(image_aug, thickness=20, color=[0, 0, 255])\n",
    "\n",
    "    #         after = bbs_aug.bounding_boxes[0]\n",
    "\n",
    "            # 저장\n",
    "            save_data(save_path, image_aug, point, fil_name, imgs[idx])\n",
    "        \n",
    "def save_data(save_path, image_aug, point, fil_name, imgs):\n",
    "    # 파일명 정리\n",
    "    splits = imgs.split('.')\n",
    "    file_name = splits[0]\n",
    "    file_name = fil_name + '_' + file_name \n",
    "    ext = '.' + splits[1]\n",
    "    # 저장위치 설정\n",
    "    save_path = save_path + fil_name +'/'\n",
    "    chk_dir(save_path)\n",
    "    \n",
    "    # 이미지 저장\n",
    "    cv2.imwrite(save_path + file_name + ext, image_aug)\n",
    "    print(save_path + file_name + ext)\n",
    "    # 라벨링 txt 저장\n",
    "    f = open(save_path + file_name + \".txt\", 'w')\n",
    "    f.write('0 {} {} {} {}'.format(point[1], point[2], point[3], point[4]))\n",
    "    f.close()\n",
    "    \n",
    "def select_filter(flag):\n",
    "    fil_name = ''\n",
    "    seq = iaa.Sequential([])\n",
    "    if flag == 0:\n",
    "        seq = iaa.Fliplr(1.0)\n",
    "        fil_name = 'Fliplr'\n",
    "    elif flag == 1:\n",
    "        seq = iaa.ShearX((-20, 20))\n",
    "        fil_name = 'ShearX'\n",
    "    elif flag == 2:\n",
    "        seq = iaa.imgcorruptlike.MotionBlur(severity=2)\n",
    "        fil_name = 'MotionBlur'\n",
    "    elif flag == 3:\n",
    "        iaa.CoarseDropout(0.03, size_percent=0.15)\n",
    "        fil_name = 'CoarseDropout'\n",
    "    return seq, fil_name\n",
    "\n",
    "def chk_dir(path):\n",
    "    try:\n",
    "        if not os.path.exists(path):\n",
    "            os.makedirs(path)\n",
    "    except OSError:\n",
    "        print ('Error: Creating directory. ' +  path)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 17,
   "id": "ready-criticism",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "./train/raw/30009_street/Fliplr/Fliplr_00005.jpg\n",
      "./train/raw/30009_street/Fliplr/Fliplr_00006.jpg\n",
      "./train/raw/30009_street/Fliplr/Fliplr_00007.jpg\n",
      "./train/raw/30009_street/Fliplr/Fliplr_00008.jpg\n",
      "./train/raw/30009_street/Fliplr/Fliplr_00009.jpg\n",
      "./train/raw/30009_street/Fliplr/Fliplr_00010.jpg\n",
      "./train/raw/30009_street/Fliplr/Fliplr_00011.jpg\n",
      "./train/raw/30009_street/Fliplr/Fliplr_00012.jpg\n",
      "./train/raw/30009_street/Fliplr/Fliplr_00013.jpg\n",
      "./train/raw/30009_street/Fliplr/Fliplr_00014.jpg\n",
      "./train/raw/30009_street/Fliplr/Fliplr_00015.jpg\n",
      "./train/raw/30009_street/Fliplr/Fliplr_00016.jpg\n",
      "./train/raw/30009_street/Fliplr/Fliplr_00017.jpg\n",
      "./train/raw/30009_street/Fliplr/Fliplr_00018.jpg\n",
      "./train/raw/30009_street/Fliplr/Fliplr_00019.jpg\n",
      "./train/raw/30009_street/Fliplr/Fliplr_00020.jpg\n",
      "./train/raw/30009_street/ShearX/ShearX_00005.jpg\n",
      "./train/raw/30009_street/ShearX/ShearX_00006.jpg\n",
      "./train/raw/30009_street/ShearX/ShearX_00007.jpg\n",
      "./train/raw/30009_street/ShearX/ShearX_00008.jpg\n",
      "./train/raw/30009_street/ShearX/ShearX_00009.jpg\n",
      "./train/raw/30009_street/ShearX/ShearX_00010.jpg\n",
      "./train/raw/30009_street/ShearX/ShearX_00011.jpg\n",
      "./train/raw/30009_street/ShearX/ShearX_00012.jpg\n",
      "./train/raw/30009_street/ShearX/ShearX_00013.jpg\n",
      "./train/raw/30009_street/ShearX/ShearX_00014.jpg\n",
      "./train/raw/30009_street/ShearX/ShearX_00015.jpg\n",
      "./train/raw/30009_street/ShearX/ShearX_00016.jpg\n",
      "./train/raw/30009_street/ShearX/ShearX_00017.jpg\n",
      "./train/raw/30009_street/ShearX/ShearX_00018.jpg\n",
      "./train/raw/30009_street/ShearX/ShearX_00019.jpg\n",
      "./train/raw/30009_street/ShearX/ShearX_00020.jpg\n",
      "./train/raw/30009_street/MotionBlur/MotionBlur_00005.jpg\n",
      "./train/raw/30009_street/MotionBlur/MotionBlur_00006.jpg\n",
      "./train/raw/30009_street/MotionBlur/MotionBlur_00007.jpg\n",
      "./train/raw/30009_street/MotionBlur/MotionBlur_00008.jpg\n",
      "./train/raw/30009_street/MotionBlur/MotionBlur_00009.jpg\n",
      "./train/raw/30009_street/MotionBlur/MotionBlur_00010.jpg\n",
      "./train/raw/30009_street/MotionBlur/MotionBlur_00011.jpg\n",
      "./train/raw/30009_street/MotionBlur/MotionBlur_00012.jpg\n",
      "./train/raw/30009_street/MotionBlur/MotionBlur_00013.jpg\n",
      "./train/raw/30009_street/MotionBlur/MotionBlur_00014.jpg\n",
      "./train/raw/30009_street/MotionBlur/MotionBlur_00015.jpg\n",
      "./train/raw/30009_street/MotionBlur/MotionBlur_00016.jpg\n",
      "./train/raw/30009_street/MotionBlur/MotionBlur_00017.jpg\n",
      "./train/raw/30009_street/MotionBlur/MotionBlur_00018.jpg\n",
      "./train/raw/30009_street/MotionBlur/MotionBlur_00019.jpg\n",
      "./train/raw/30009_street/MotionBlur/MotionBlur_00020.jpg\n",
      "./train/raw/30009_street/CoarseDropout/CoarseDropout_00005.jpg\n",
      "./train/raw/30009_street/CoarseDropout/CoarseDropout_00006.jpg\n",
      "./train/raw/30009_street/CoarseDropout/CoarseDropout_00007.jpg\n",
      "./train/raw/30009_street/CoarseDropout/CoarseDropout_00008.jpg\n",
      "./train/raw/30009_street/CoarseDropout/CoarseDropout_00009.jpg\n",
      "./train/raw/30009_street/CoarseDropout/CoarseDropout_00010.jpg\n",
      "./train/raw/30009_street/CoarseDropout/CoarseDropout_00011.jpg\n",
      "./train/raw/30009_street/CoarseDropout/CoarseDropout_00012.jpg\n",
      "./train/raw/30009_street/CoarseDropout/CoarseDropout_00013.jpg\n",
      "./train/raw/30009_street/CoarseDropout/CoarseDropout_00014.jpg\n",
      "./train/raw/30009_street/CoarseDropout/CoarseDropout_00015.jpg\n",
      "./train/raw/30009_street/CoarseDropout/CoarseDropout_00016.jpg\n",
      "./train/raw/30009_street/CoarseDropout/CoarseDropout_00017.jpg\n",
      "./train/raw/30009_street/CoarseDropout/CoarseDropout_00018.jpg\n",
      "./train/raw/30009_street/CoarseDropout/CoarseDropout_00019.jpg\n",
      "./train/raw/30009_street/CoarseDropout/CoarseDropout_00020.jpg\n"
     ]
    }
   ],
   "source": [
    "# img_dir = ['./data/Multi-view_tracking_indoor/00001_market/frames/02/',\n",
    "#           './data/Multi-view_tracking_indoor/06275_market/frames/01/',\n",
    "#           './data/Multi-view_tracking_outdoor/31293_street/frames/02/',\n",
    "#           './data/Multi-view_tracking_outdoor/30012_street/frames/01/']\n",
    "img_dir = './data/Multi-view_tracking_outdoor/30009_street/frames/01/'\n",
    "\n",
    "# label_dir = ['./labels_txt/Multi-view_tracking_indoor/00001_market/json/1-5/02/',\n",
    "#             './labels_txt/Multi-view_tracking_indoor/06275_market/json/1-5/01/',\n",
    "#             './labels_txt/Multi-view_tracking_outdoor/31293_street/json/1-5/02/',\n",
    "#             './labels_txt/Multi-view_tracking_outdoor/30012_street/json/1-5/01/']\n",
    "label_dir = './labels_txt/Multi-view_tracking_outdoor/30009_street/json/1-5/01/'\n",
    "\n",
    "# save_path = ['./train/raw/00001_market/',\n",
    "#             './train/raw/06275_market/',\n",
    "#             './train/raw/31293_street/',\n",
    "#             './train/raw/30012_street/']\n",
    "save_path = './train/raw/30009_street/'\n",
    "fil_name = 'CoarseDropout/'\n",
    "\n",
    "# for i in range(4):\n",
    "#     get_blur(0, img_dir[i], label_dir[i], save_path[i]+fil_name)\n",
    "get_blur(img_dir, label_dir, save_path)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "fatty-clothing",
   "metadata": {},
   "outputs": [],
   "source": []
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "foster-chrome",
   "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.9"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 5
}