Juno_Header.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "e5134265",
   "metadata": {},
   "outputs": [],
   "source": [
    "# 10/26/2021-1\n",
    "from __future__ import absolute_import\n",
    "from IPython.display import clear_output, display\n",
    "import glob, os, shutil, sys, uuid\n",
    "from os.path import *\n",
    "from time import sleep\n",
    "\n",
    "try: \n",
    "    from matplotlib import pyplot as plt\n",
    "except Exception as err:\n",
    "    %pip install matplotlib\n",
    "    from matplotlib import pyplot as plt\n",
    "# from CV2_Utils import *\n",
    "try: \n",
    "    import cv2\n",
    "    from cv2 import COLOR_BGR2RGBA, IMREAD_UNCHANGED\n",
    "except Exception as err:\n",
    "    print(err)\n",
    "    %pip install opencv-python-headless # works.\n",
    "    # %pip install opencv-contrib-python-headless\n",
    "    import cv2\n",
    "    from cv2 import COLOR_BGR2RGBA, IMREAD_UNCHANGED\n",
    "\n",
    "try:\n",
    "    pass\n",
    "    import tensorflow as tf\n",
    "except Exception as err:\n",
    "    pass\n",
    "    # print(err)\n",
    "    %pip install tensorflow\n",
    "    \n",
    "import numpy as np\n",
    "\n",
    "contentPath = os.getcwd()\n",
    "tfImagesPath = join(contentPath, 'TensorflowImages')\n",
    "\n",
    "import TarfileFunctions\n",
    "from TarfileFunctions import *\n",
    "from BashColors import C\n",
    "\n",
    "if not exists(tfImagesPath):\n",
    "    os.makedirs(tfImagesPath)\n",
    "    os.chdir(tfImagesPath)\n",
    "    tff.extractTarfiles('New_Juno_Original_Images.tar.gz')\n",
    "\n",
    "os.chdir(contentPath)\n",
    "from TarfileFunctions import *\n",
    "from BashColors import C\n",
    "tarfilePath=join(contentPath, 'New_Juno_Original_Images.tar.gz')\n",
    "if exists(tarfilePath) and not exists(tfImagesPath):\n",
    "    tff.extractTarfiles('New_Juno_Original_Images.tar.gz')\n",
    "\n",
    "tfImagesPath = join(contentPath, 'TensorflowImages')\n",
    "\n",
    "os.chdir(contentPath)\n",
    "cv2Path = join(contentPath, 'CV2Images')\n",
    "\n",
    "if not exists(cv2Path):\n",
    "    os.mkdir(cv2Path)\n",
    "    tff.extractTarfiles('All_CV2_Images.tar.gz')\n",
    "\n",
    "os.chdir(contentPath)\n",
    "\n",
    "print(f'cv2 version: {C.IBlue}{cv2.__version__}{C.ColorOff}')\n",
    "print(f'tf version: {C.IBlue}{tf.__version__}{C.ColorOff}')\n",
    "print(f'cwd: {C.IBlue}{os.getcwd()}{C.ColorOff}')\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "91683c4e",
   "metadata": {},
   "outputs": [],
   "source": [
    "tf"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "595fdf3a",
   "metadata": {},
   "outputs": [],
   "source": [
    "# image_dataset_from_directory function\n",
    "import tensorflow as tf\n",
    "from tensorflow.keras.preprocessing import image_dataset_from_directory\n",
    "\n",
    "dataset = image_dataset_from_directory(\n",
    "    cv2Path,\n",
    "    labels=\"inferred\",\n",
    "    label_mode=\"int\",\n",
    "    class_names=None,\n",
    "    color_mode=\"rgb\",\n",
    "    batch_size=2,\n",
    "    image_size=(224, 224),\n",
    "    shuffle=True,\n",
    "    seed=None,\n",
    "    validation_split=None,\n",
    "    subset=None,\n",
    "    interpolation=\"bilinear\",\n",
    "    follow_links=False,\n",
    "    crop_to_aspect_ratio=False\n",
    "    # ,**kwargs\n",
    ")\n",
    "print(dataset.class_names)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "fff4b291",
   "metadata": {},
   "outputs": [],
   "source": [
    "if exists(cv2Path):\n",
    "    print(cv2Path)\n",
    "train_ds = tf.keras.utils.image_dataset_from_directory(\n",
    "  cv2Path,\n",
    "  validation_split=0.2,\n",
    "  subset=\"training\",\n",
    "  seed=123,\n",
    "  image_size=(224, 224),\n",
    "  batch_size=2)\n",
    "print(train_ds.class_names)"
   ]
  }
 ],
 "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.7.10"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 5
}