check_stim.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 2,
   "metadata": {},
   "outputs": [
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "/Users/nibey/opt/anaconda3/lib/python3.7/importlib/_bootstrap.py:219: RuntimeWarning: numpy.ufunc size changed, may indicate binary incompatibility. Expected 192 from C header, got 216 from PyObject\n",
      "  return f(*args, **kwds)\n"
     ]
    }
   ],
   "source": [
    "import os\n",
    "from os import listdir, walk\n",
    "from os.path import isfile, join\n",
    "import pandas as pd\n",
    "import numpy as np"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "metadata": {},
   "outputs": [],
   "source": [
    "def see_dir(dir_path):\n",
    "    print('Total File sizes in *'+dir_path+'*')\n",
    "    a=0\n",
    "    for f in os.listdir(dir_path):\n",
    "        if 'zip' not in f:\n",
    "            a+=1\n",
    "            print(f.ljust(30) + str(round(os.path.getsize(dir_path + '/' + f) / 1000000, 4)) + 'MB')\n",
    "    print('# of files: ', a)\n",
    "    "
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 117,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Total File sizes in *./1-FSO_localizer/stim/rgb/object*\n",
      "48.jpg                        0.021MB\n",
      "8.jpg                         0.0145MB\n",
      "9.jpg                         0.0301MB\n",
      "14.jpg                        0.0163MB\n",
      "28.jpg                        0.0108MB\n",
      "29.jpg                        0.0206MB\n",
      "15.jpg                        0.0076MB\n",
      "17.jpg                        0.0107MB\n",
      "16.jpg                        0.0064MB\n",
      "12.jpg                        0.012MB\n",
      "13.jpg                        0.0191MB\n",
      "39.jpg                        0.011MB\n",
      "11.jpg                        0.0227MB\n",
      "10.jpg                        0.0122MB\n",
      "38.jpg                        0.0126MB\n",
      "21.jpg                        0.0288MB\n",
      "35.jpg                        0.0105MB\n",
      "34.jpg                        0.0226MB\n",
      "20.jpg                        0.0106MB\n",
      "36.jpg                        0.0112MB\n",
      "22.jpg                        0.0108MB\n",
      "23.jpg                        0.0108MB\n",
      "37.jpg                        0.0117MB\n",
      "33.jpg                        0.0147MB\n",
      "27.jpg                        0.0087MB\n",
      "26.jpg                        0.0276MB\n",
      "32.jpg                        0.0104MB\n",
      "18.jpg                        0.0176MB\n",
      "24.jpg                        0.0236MB\n",
      "30.jpg                        0.0135MB\n",
      "31.jpg                        0.0129MB\n",
      "25.jpg                        0.0119MB\n",
      "19.jpg                        0.0185MB\n",
      "42.jpg                        0.0185MB\n",
      "4.jpg                         0.0138MB\n",
      "5.jpg                         0.0076MB\n",
      "43.jpg                        0.0146MB\n",
      "7.jpg                         0.0107MB\n",
      "41.jpg                        0.018MB\n",
      "40.jpg                        0.0165MB\n",
      "6.jpg                         0.0156MB\n",
      "2.jpg                         0.02MB\n",
      "44.jpg                        0.0281MB\n",
      "45.jpg                        0.0129MB\n",
      "3.jpg                         0.0129MB\n",
      "47.jpg                        0.011MB\n",
      "1.jpg                         0.0127MB\n",
      "46.jpg                        0.0111MB\n",
      "# of files:  48\n"
     ]
    }
   ],
   "source": [
    "title = './1-FSO_localizer/stim/rgb/object'\n",
    "see_dir(title)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 157,
   "metadata": {},
   "outputs": [],
   "source": [
    "import glob\n",
    "import matplotlib.pyplot as plt\n",
    "from matplotlib.image import imread\n",
    "import cv2\n",
    "\n",
    "def resize_img(img_dir, save_dir):\n",
    "    \n",
    "    data_files = sorted(glob.glob(img_dir + '/*')) \n",
    "    \n",
    "    for f, i in zip(data_files, range(len(data_files))):\n",
    "        img = plt.imread(f)\n",
    "        img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)\n",
    "        img_size = max(img.shape) \n",
    "        \n",
    "        if img.shape[0] != img_size:\n",
    "\n",
    "            addshape1 = int((img_size-img.shape[0])/2)\n",
    "            addshape2 = img_size-addshape1-img.shape[0] \n",
    "\n",
    "            if addshape1>0:\n",
    "                add_img1 = [[[255, 255, 255] for i in range(img_size)] for i in range(addshape1)]\n",
    "                img = np.concatenate((add_img1, img), axis=0)\n",
    "            if addshape2>0:\n",
    "                add_img2 = [[[255, 255, 255] for i in range(img_size)] for i in range(addshape2)]\n",
    "                img = np.concatenate((img, add_img2), axis=0)\n",
    "\n",
    "            print(i, f, img.shape)\n",
    "            cv2.imwrite(save_dir+f.split('/')[-1], img)\n",
    "\n",
    "\n",
    "        elif img.shape[1] != img_size:\n",
    "\n",
    "            addshape1 = int((img_size-img.shape[1])/2)\n",
    "            addshape2 = img_size-addshape1-img.shape[1]\n",
    "\n",
    "            if addshape1>0:\n",
    "                add_img1 = [[[255, 255, 255] for i in range(addshape1)] for i in range(img_size)]\n",
    "                img = np.concatenate((add_img1, img), axis=1)\n",
    "            if addshape2>0:\n",
    "                add_img2 = [[[255, 255, 255] for i in range(addshape2)] for i in range(img_size)]\n",
    "                img = np.concatenate((img, add_img2), axis=1)\n",
    "            \n",
    "            print(i, f, img.shape)\n",
    "            cv2.imwrite(save_dir+f.split('/')[-1], img)\n",
    "\n",
    "        else: \n",
    "            cv2.imwrite(save_dir+f.split('/')[-1], img)\n",
    "\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 158,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "0 2-SO_connectivity/stim/object/1.jpg (300, 300, 3)\n",
      "2 2-SO_connectivity/stim/object/100.jpg (455, 455, 3)\n",
      "3 2-SO_connectivity/stim/object/101.jpg (389, 389, 3)\n",
      "4 2-SO_connectivity/stim/object/102.jpg (474, 474, 3)\n",
      "5 2-SO_connectivity/stim/object/103.jpg (313, 313, 3)\n",
      "6 2-SO_connectivity/stim/object/104.jpg (520, 520, 3)\n",
      "7 2-SO_connectivity/stim/object/105.jpg (389, 389, 3)\n",
      "8 2-SO_connectivity/stim/object/106.jpg (593, 593, 3)\n",
      "9 2-SO_connectivity/stim/object/107.jpg (417, 417, 3)\n",
      "10 2-SO_connectivity/stim/object/108.jpg (426, 426, 3)\n",
      "11 2-SO_connectivity/stim/object/109.jpg (410, 410, 3)\n",
      "12 2-SO_connectivity/stim/object/11.jpg (300, 300, 3)\n",
      "13 2-SO_connectivity/stim/object/110.jpg (462, 462, 3)\n",
      "14 2-SO_connectivity/stim/object/111.jpg (389, 389, 3)\n",
      "15 2-SO_connectivity/stim/object/112.jpg (542, 542, 3)\n",
      "16 2-SO_connectivity/stim/object/113.jpg (831, 831, 3)\n",
      "17 2-SO_connectivity/stim/object/114.jpg (572, 572, 3)\n",
      "18 2-SO_connectivity/stim/object/115.jpg (890, 890, 3)\n",
      "19 2-SO_connectivity/stim/object/116.jpg (389, 389, 3)\n",
      "20 2-SO_connectivity/stim/object/117.jpg (422, 422, 3)\n",
      "21 2-SO_connectivity/stim/object/118.jpg (389, 389, 3)\n",
      "22 2-SO_connectivity/stim/object/119.jpg (758, 758, 3)\n",
      "24 2-SO_connectivity/stim/object/120.jpg (600, 600, 3)\n",
      "25 2-SO_connectivity/stim/object/121.jpg (890, 890, 3)\n",
      "26 2-SO_connectivity/stim/object/122.jpg (306, 306, 3)\n",
      "27 2-SO_connectivity/stim/object/123.jpg (525, 525, 3)\n",
      "28 2-SO_connectivity/stim/object/124.jpg (343, 343, 3)\n",
      "29 2-SO_connectivity/stim/object/125.jpg (407, 407, 3)\n",
      "30 2-SO_connectivity/stim/object/126.jpg (431, 431, 3)\n",
      "31 2-SO_connectivity/stim/object/127.jpg (2560, 2560, 3)\n",
      "32 2-SO_connectivity/stim/object/128.jpg (639, 639, 3)\n",
      "33 2-SO_connectivity/stim/object/129.jpg (425, 425, 3)\n",
      "35 2-SO_connectivity/stim/object/130.jpg (389, 389, 3)\n",
      "36 2-SO_connectivity/stim/object/131.jpg (663, 663, 3)\n",
      "37 2-SO_connectivity/stim/object/132.jpg (596, 596, 3)\n",
      "38 2-SO_connectivity/stim/object/133.jpg (685, 685, 3)\n",
      "39 2-SO_connectivity/stim/object/134.jpg (702, 702, 3)\n",
      "40 2-SO_connectivity/stim/object/135.jpg (890, 890, 3)\n",
      "41 2-SO_connectivity/stim/object/136.jpg (389, 389, 3)\n",
      "42 2-SO_connectivity/stim/object/137.jpg (497, 497, 3)\n",
      "43 2-SO_connectivity/stim/object/138.jpg (582, 582, 3)\n",
      "44 2-SO_connectivity/stim/object/139.jpg (629, 629, 3)\n",
      "45 2-SO_connectivity/stim/object/14.jpg (300, 300, 3)\n",
      "46 2-SO_connectivity/stim/object/140.jpg (490, 490, 3)\n",
      "47 2-SO_connectivity/stim/object/141.jpg (384, 384, 3)\n",
      "48 2-SO_connectivity/stim/object/142.jpg (423, 423, 3)\n",
      "49 2-SO_connectivity/stim/object/143.jpg (811, 811, 3)\n",
      "50 2-SO_connectivity/stim/object/144.jpg (800, 800, 3)\n",
      "51 2-SO_connectivity/stim/object/145.jpg (389, 389, 3)\n",
      "52 2-SO_connectivity/stim/object/146.jpg (389, 389, 3)\n",
      "53 2-SO_connectivity/stim/object/147.jpg (577, 577, 3)\n",
      "54 2-SO_connectivity/stim/object/148.jpg (640, 640, 3)\n",
      "56 2-SO_connectivity/stim/object/16.jpg (300, 300, 3)\n",
      "59 2-SO_connectivity/stim/object/19.jpg (300, 300, 3)\n",
      "60 2-SO_connectivity/stim/object/2.jpg (300, 300, 3)\n",
      "64 2-SO_connectivity/stim/object/23.jpg (300, 300, 3)\n",
      "65 2-SO_connectivity/stim/object/24.jpg (300, 300, 3)\n",
      "67 2-SO_connectivity/stim/object/26.jpg (300, 300, 3)\n",
      "72 2-SO_connectivity/stim/object/30.jpg (300, 300, 3)\n",
      "74 2-SO_connectivity/stim/object/32.jpg (300, 300, 3)\n",
      "75 2-SO_connectivity/stim/object/33.jpg (300, 300, 3)\n",
      "76 2-SO_connectivity/stim/object/34.jpg (300, 300, 3)\n",
      "78 2-SO_connectivity/stim/object/36.jpg (300, 300, 3)\n",
      "79 2-SO_connectivity/stim/object/37.jpg (300, 300, 3)\n",
      "80 2-SO_connectivity/stim/object/38.jpg (300, 300, 3)\n",
      "83 2-SO_connectivity/stim/object/40.jpg (300, 300, 3)\n",
      "85 2-SO_connectivity/stim/object/42.jpg (300, 300, 3)\n",
      "88 2-SO_connectivity/stim/object/45.jpg (300, 300, 3)\n",
      "91 2-SO_connectivity/stim/object/48.jpg (300, 300, 3)\n",
      "94 2-SO_connectivity/stim/object/50.jpg (300, 300, 3)\n",
      "95 2-SO_connectivity/stim/object/51.jpg (1715, 1715, 3)\n",
      "96 2-SO_connectivity/stim/object/52.jpg (646, 646, 3)\n",
      "97 2-SO_connectivity/stim/object/53.jpg (390, 390, 3)\n",
      "98 2-SO_connectivity/stim/object/54.jpg (389, 389, 3)\n",
      "99 2-SO_connectivity/stim/object/55.jpg (617, 617, 3)\n",
      "100 2-SO_connectivity/stim/object/56.jpg (385, 385, 3)\n",
      "101 2-SO_connectivity/stim/object/57.jpg (461, 461, 3)\n",
      "102 2-SO_connectivity/stim/object/58.jpg (782, 782, 3)\n",
      "103 2-SO_connectivity/stim/object/59.jpg (668, 668, 3)\n",
      "105 2-SO_connectivity/stim/object/60.jpg (779, 779, 3)\n",
      "106 2-SO_connectivity/stim/object/61.jpg (512, 512, 3)\n",
      "107 2-SO_connectivity/stim/object/62.jpg (890, 890, 3)\n",
      "108 2-SO_connectivity/stim/object/63.jpg (620, 620, 3)\n",
      "109 2-SO_connectivity/stim/object/64.jpg (444, 444, 3)\n",
      "110 2-SO_connectivity/stim/object/65.jpg (500, 500, 3)\n",
      "111 2-SO_connectivity/stim/object/66.jpg (389, 389, 3)\n",
      "112 2-SO_connectivity/stim/object/67.jpg (618, 618, 3)\n",
      "113 2-SO_connectivity/stim/object/68.jpg (353, 353, 3)\n",
      "114 2-SO_connectivity/stim/object/69.jpg (400, 400, 3)\n",
      "116 2-SO_connectivity/stim/object/70.jpg (432, 432, 3)\n",
      "117 2-SO_connectivity/stim/object/71.jpg (392, 392, 3)\n",
      "118 2-SO_connectivity/stim/object/72.jpg (396, 396, 3)\n",
      "119 2-SO_connectivity/stim/object/73.jpg (668, 668, 3)\n",
      "120 2-SO_connectivity/stim/object/74.jpg (831, 831, 3)\n",
      "121 2-SO_connectivity/stim/object/75.jpg (477, 477, 3)\n",
      "122 2-SO_connectivity/stim/object/76.jpg (600, 600, 3)\n",
      "123 2-SO_connectivity/stim/object/77.jpg (389, 389, 3)\n",
      "124 2-SO_connectivity/stim/object/78.jpg (2797, 2797, 3)\n",
      "125 2-SO_connectivity/stim/object/79.jpg (611, 611, 3)\n",
      "127 2-SO_connectivity/stim/object/80.jpg (890, 890, 3)\n",
      "128 2-SO_connectivity/stim/object/81.jpg (507, 507, 3)\n",
      "129 2-SO_connectivity/stim/object/82.jpg (458, 458, 3)\n",
      "130 2-SO_connectivity/stim/object/83.jpg (761, 761, 3)\n",
      "131 2-SO_connectivity/stim/object/84.jpg (567, 567, 3)\n",
      "132 2-SO_connectivity/stim/object/85.jpg (389, 389, 3)\n",
      "133 2-SO_connectivity/stim/object/86.jpg (630, 630, 3)\n",
      "134 2-SO_connectivity/stim/object/87.jpg (890, 890, 3)\n",
      "135 2-SO_connectivity/stim/object/88.jpg (782, 782, 3)\n",
      "136 2-SO_connectivity/stim/object/89.jpg (608, 608, 3)\n",
      "138 2-SO_connectivity/stim/object/90.jpg (868, 868, 3)\n",
      "139 2-SO_connectivity/stim/object/91.jpg (475, 475, 3)\n",
      "140 2-SO_connectivity/stim/object/92.jpg (449, 449, 3)\n",
      "141 2-SO_connectivity/stim/object/93.jpg (976, 976, 3)\n",
      "142 2-SO_connectivity/stim/object/94.jpg (484, 484, 3)\n",
      "143 2-SO_connectivity/stim/object/95.jpg (496, 496, 3)\n",
      "144 2-SO_connectivity/stim/object/96.jpg (350, 350, 3)\n",
      "145 2-SO_connectivity/stim/object/97.jpg (444, 444, 3)\n",
      "146 2-SO_connectivity/stim/object/98.jpg (582, 582, 3)\n",
      "147 2-SO_connectivity/stim/object/99.jpg (784, 784, 3)\n"
     ]
    }
   ],
   "source": [
    "IMG_DIR = '2-SO_connectivity/stim/object/'\n",
    "SAVE_DIR ='2-SO_connectivity/stim/object_resize/'\n",
    "resize_img(IMG_DIR, SAVE_DIR)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 163,
   "metadata": {},
   "outputs": [],
   "source": [
    "def gray_img(img_dir, save_dir):\n",
    "    \n",
    "    data_files = sorted(glob.glob(img_dir + '/*')) \n",
    "    \n",
    "    for f, i in zip(data_files, range(len(data_files))):\n",
    "        img = plt.imread(f)\n",
    "        img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)\n",
    "        gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)\n",
    "        print(i, f)\n",
    "        cv2.imwrite(save_dir+f.split('/')[-1], gray)\n",
    "        "
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 169,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "0 1-FSO_localizer/stim/rgbScale/face/1.jpg\n",
      "1 1-FSO_localizer/stim/rgbScale/face/10.jpg\n",
      "2 1-FSO_localizer/stim/rgbScale/face/11.jpg\n",
      "3 1-FSO_localizer/stim/rgbScale/face/12.jpg\n",
      "4 1-FSO_localizer/stim/rgbScale/face/13.jpg\n",
      "5 1-FSO_localizer/stim/rgbScale/face/14.jpg\n",
      "6 1-FSO_localizer/stim/rgbScale/face/15.jpg\n",
      "7 1-FSO_localizer/stim/rgbScale/face/16.jpg\n",
      "8 1-FSO_localizer/stim/rgbScale/face/17.jpg\n",
      "9 1-FSO_localizer/stim/rgbScale/face/18.jpg\n",
      "10 1-FSO_localizer/stim/rgbScale/face/19.jpg\n",
      "11 1-FSO_localizer/stim/rgbScale/face/2.jpg\n",
      "12 1-FSO_localizer/stim/rgbScale/face/20.jpg\n",
      "13 1-FSO_localizer/stim/rgbScale/face/21.jpg\n",
      "14 1-FSO_localizer/stim/rgbScale/face/22.jpg\n",
      "15 1-FSO_localizer/stim/rgbScale/face/23.jpg\n",
      "16 1-FSO_localizer/stim/rgbScale/face/24.jpg\n",
      "17 1-FSO_localizer/stim/rgbScale/face/25.jpg\n",
      "18 1-FSO_localizer/stim/rgbScale/face/26.jpg\n",
      "19 1-FSO_localizer/stim/rgbScale/face/27.jpg\n",
      "20 1-FSO_localizer/stim/rgbScale/face/28.jpg\n",
      "21 1-FSO_localizer/stim/rgbScale/face/29.jpg\n",
      "22 1-FSO_localizer/stim/rgbScale/face/3.jpg\n",
      "23 1-FSO_localizer/stim/rgbScale/face/30.jpg\n",
      "24 1-FSO_localizer/stim/rgbScale/face/31.jpg\n",
      "25 1-FSO_localizer/stim/rgbScale/face/32.jpg\n",
      "26 1-FSO_localizer/stim/rgbScale/face/33.jpg\n",
      "27 1-FSO_localizer/stim/rgbScale/face/34.jpg\n",
      "28 1-FSO_localizer/stim/rgbScale/face/35.jpg\n",
      "29 1-FSO_localizer/stim/rgbScale/face/36.jpg\n",
      "30 1-FSO_localizer/stim/rgbScale/face/37.jpg\n",
      "31 1-FSO_localizer/stim/rgbScale/face/38.jpg\n",
      "32 1-FSO_localizer/stim/rgbScale/face/39.jpg\n",
      "33 1-FSO_localizer/stim/rgbScale/face/4.jpg\n",
      "34 1-FSO_localizer/stim/rgbScale/face/40.jpg\n",
      "35 1-FSO_localizer/stim/rgbScale/face/41.jpg\n",
      "36 1-FSO_localizer/stim/rgbScale/face/42.jpg\n",
      "37 1-FSO_localizer/stim/rgbScale/face/43.jpg\n",
      "38 1-FSO_localizer/stim/rgbScale/face/44.jpg\n",
      "39 1-FSO_localizer/stim/rgbScale/face/45.jpg\n",
      "40 1-FSO_localizer/stim/rgbScale/face/46.jpg\n",
      "41 1-FSO_localizer/stim/rgbScale/face/47.jpg\n",
      "42 1-FSO_localizer/stim/rgbScale/face/48.jpg\n",
      "43 1-FSO_localizer/stim/rgbScale/face/5.jpg\n",
      "44 1-FSO_localizer/stim/rgbScale/face/6.jpg\n",
      "45 1-FSO_localizer/stim/rgbScale/face/7.jpg\n",
      "46 1-FSO_localizer/stim/rgbScale/face/8.jpg\n",
      "47 1-FSO_localizer/stim/rgbScale/face/9.jpg\n"
     ]
    }
   ],
   "source": [
    "IMG_DIR = '1-FSO_localizer/stim/rgbScale/face/'\n",
    "SAVE_DIR ='1-FSO_localizer//stim/grayScale/face/'\n",
    "gray_img(IMG_DIR, SAVE_DIR)"
   ]
  },
  {
   "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.4"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 4
}