S_Rover_Image_processing.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Stream source not available. Switching to file source.\n"
     ]
    }
   ],
   "source": [
    "import cv2 as cv\n",
    "import sys\n",
    "import numpy as np\n",
    "import os\n",
    "import glob\n",
    "import re\n",
    "import shutil\n",
    "\n",
    "\n",
    "stream_source = \"http://192.168.4.1/camera\" #stream source\n",
    "file_source = r\"C:\\Users\\shwet\\Desktop\\yesist final\\input videos\\WhatsApp Video 2024-03-01 at 00.43.04_7fda4119.mp4\" \n",
    "\n",
    "cap = cv.VideoCapture(stream_source)\n",
    "\n",
    "if not cap.isOpened():\n",
    "    print(\"Stream source not available. Switching to file source.\")\n",
    "    cap = cv.VideoCapture(file_source)\n",
    "\n",
    "if not cap.isOpened():\n",
    "    print(\"Error opening video source\")\n",
    "    exit()\n",
    "\n",
    "frame_interval = 7  \n",
    "frame_count_new = 0 \n",
    "\n",
    "frames_dir = r'C:\\Users\\shwet\\Desktop\\yesist final\\STREAM FRAMES'\n",
    "if not os.path.exists(frames_dir):\n",
    "    os.makedirs(frames_dir)\n",
    "\n",
    "while True:\n",
    "    ret, frame = cap.read()\n",
    "    if not ret:\n",
    "        break\n",
    "\n",
    "    if frame_count_new % frame_interval == 0:\n",
    "        cv.imwrite(os.path.join(frames_dir, f'frame{frame_count_new}.jpg'), frame)  \n",
    "\n",
    "    frame_count_new += 1\n",
    "\n",
    "cap.release()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 12,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "True"
      ]
     },
     "execution_count": 12,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "\n",
    "image_dir = r'C:\\Users\\shwet\\Desktop\\yesist final\\frames'   \n",
    "annuli_dir =  r'C:\\Users\\shwet\\Desktop\\yesist final\\annulus ring strips' \n",
    "strips_dir =  r'C:\\Users\\shwet\\Desktop\\yesist final\\rectangular strips' \n",
    "rings_dir =   r'C:\\Users\\shwet\\Desktop\\yesist final\\mapped rings'\n",
    "destination_dir = r'C:\\Users\\shwet\\Desktop\\yesist final\\color_maps'\n",
    "source_dir = os.getcwd()\n",
    "image_paths = glob.glob(os.path.join(image_dir, '*'))\n",
    "image_paths.sort(key=lambda f: int(re.sub('\\D', '', f)))\n",
    "grid_spacing = 50  # Adjust as needed\n",
    "grid_color = (0, 255, 0)  #(BGR format)\n",
    "strips = []\n",
    "\n",
    "def extract_circular_strip(image, center, inner_radius, outer_radius):\n",
    "    mask = np.zeros_like(image)\n",
    "    cv.circle(mask, center, outer_radius, (255,255,255), -1)  \n",
    "    cv.circle(mask, center, inner_radius, (0,0,0), -1) \n",
    "    result = cv.bitwise_and(image, mask)\n",
    "\n",
    "    return result\n",
    "\n",
    "center = (310,128)  # x, y (310,128) || (390,518)\n",
    "inner_radius = 220  #220 || 502\n",
    "outer_radius = 232 #232|| 518\n",
    "\n",
    "for i, img_path in enumerate(image_paths):\n",
    "    img = cv.imread(img_path)\n",
    "\n",
    "    if img is None:\n",
    "        print(f\"Could not read the image: {img_path}\")\n",
    "        continue\n",
    "\n",
    "    edges = cv.Canny(img,50,150)\n",
    "\n",
    "    # contours\n",
    "    contours, hierarchy = cv.findContours(edges, cv.RETR_EXTERNAL, cv.CHAIN_APPROX_SIMPLE)\n",
    "\n",
    "    # Drawing contours\n",
    "    imgcontour = cv.drawContours(img, contours, -1, (0,0,0), 1)\n",
    "\n",
    "    # circular strip\n",
    "    strip = extract_circular_strip(imgcontour, center, inner_radius, outer_radius)\n",
    "\n",
    "    # annulus\n",
    "    cv.imwrite(os.path.join(annuli_dir, f'annulus{i}.jpg'), strip)\n",
    "\n",
    "    img_rings = cv.circle(img.copy(), center, outer_radius, (0,255,0), 2)\n",
    "    img_rings = cv.circle(img_rings, center, inner_radius, (0,255,0), 2)\n",
    "\n",
    "    cv.imwrite(os.path.join(rings_dir, f'rings{i}.jpg'), img_rings)\n",
    "\n",
    "    polar_image = cv.warpPolar(strip, dsize=(500,1570), center=center, maxRadius=outer_radius, flags=cv.WARP_POLAR_LINEAR)\n",
    "    polar_image = cv.rotate(polar_image, cv.ROTATE_90_COUNTERCLOCKWISE)\n",
    "\n",
    "    crop = polar_image[0:outer_radius-inner_radius,:]\n",
    "\n",
    "    cv.imwrite(os.path.join(strips_dir, f'strip{i}.jpg'), crop)\n",
    "\n",
    "    strips.append(crop)\n",
    "\n",
    "    #adjust as needed\n",
    "    inner_radius -= 4 # ||5\n",
    "    outer_radius -= 4 # ||5\n",
    "\n",
    "map = np.concatenate(strips, axis=0)\n",
    "\n",
    "colormap1 = cv.applyColorMap(map, cv.COLORMAP_JET)\n",
    "cv.imwrite(\"colored_map1.jpg\", colormap1)\n",
    "colormap2 = cv.applyColorMap(map, cv.COLORMAP_HOT)\n",
    "cv.imwrite(\"colored_map2.jpg\", colormap2)\n",
    "colormap3 = cv.applyColorMap(map, cv.COLORMAP_PARULA)\n",
    "cv.imwrite(\"colored_map3.jpg\", colormap3)\n",
    "colormap4 = cv.applyColorMap(map, cv.COLORMAP_TWILIGHT)\n",
    "cv.imwrite(\"colored_map4.jpg\", colormap4)\n",
    "colormap5 = cv.applyColorMap(map, cv.COLORMAP_TWILIGHT_SHIFTED)\n",
    "cv.imwrite(\"colored_map5.jpg\", colormap5)\n",
    "colormap6 = cv.applyColorMap(map, cv.COLORMAP_TURBO)\n",
    "cv.imwrite(\"colored_map6.jpg\", colormap6)\n",
    "colormap7 = cv.applyColorMap(map, cv.COLORMAP_MAGMA)\n",
    "cv.imwrite(\"colored_map7.jpg\", colormap7)\n",
    "colormap8 = cv.applyColorMap(map, cv.COLORMAP_INFERNO)\n",
    "cv.imwrite(\"colored_map8.jpg\", colormap8)\n",
    "colormap9 = cv.applyColorMap(map, cv.COLORMAP_PLASMA)\n",
    "cv.imwrite(\"colored_map9.jpg\", colormap9)\n",
    "colormap10 = cv.applyColorMap(map, cv.COLORMAP_VIRIDIS)\n",
    "cv.imwrite(\"colored_map10.jpg\", colormap10)\n",
    "colormap11 = cv.applyColorMap(map, cv.COLORMAP_CIVIDIS)\n",
    "cv.imwrite(\"colored_map11.jpg\", colormap11)\n",
    "colormap12 = cv.applyColorMap(map, cv.COLORMAP_COOL)\n",
    "cv.imwrite(\"colored_map12.jpg\", colormap12)\n",
    "colormap13 = cv.applyColorMap(map, cv.COLORMAP_SPRING)\n",
    "cv.imwrite(\"colored_map13.jpg\", colormap13)\n",
    "colormap14 = cv.applyColorMap(map, cv.COLORMAP_SUMMER)\n",
    "cv.imwrite(\"colored_map14.jpg\", colormap14)\n",
    "colormap15 = cv.applyColorMap(map, cv.COLORMAP_AUTUMN)\n",
    "cv.imwrite(\"colored_map15.jpg\", colormap15)\n",
    "colormap16 = cv.applyColorMap(map, cv.COLORMAP_WINTER)\n",
    "cv.imwrite(\"colored_map16.jpg\", colormap16)\n",
    "colormap17 = cv.applyColorMap(map, cv.COLORMAP_BONE)\n",
    "cv.imwrite(\"colored_map17.jpg\", colormap17)\n",
    "colormap18 = cv.applyColorMap(map, cv.COLORMAP_HSV)\n",
    "cv.imwrite(\"colored_map18.jpg\", colormap18)\n",
    "colormap19 = cv.applyColorMap(map, cv.COLORMAP_RAINBOW)\n",
    "cv.imwrite(\"colored_map19.jpg\", colormap19)\n",
    "colormap20 = cv.applyColorMap(map, cv.COLORMAP_OCEAN)\n",
    "cv.imwrite(\"colored_map20.jpg\", colormap20)\n",
    "\n",
    "for i in range(1, 21):\n",
    "    filename = f'colored_map{i}.jpg'\n",
    "    source_file = os.path.join(source_dir, filename)\n",
    "    destination_file = os.path.join(destination_dir, filename)\n",
    "    shutil.move(source_file, destination_file)\n",
    "\n",
    "\n",
    "cv.imwrite(r'C:\\Users\\shwet\\Desktop\\yesist final\\maps\\uncolored_map.jpg', map)\n",
    "edges = cv.Canny(map,50,150)\n",
    "\n",
    "contours, hierarchy = cv.findContours(edges, cv.RETR_EXTERNAL, cv.CHAIN_APPROX_SIMPLE)\n",
    "h, w, _ = map.shape\n",
    "rows = h // grid_spacing\n",
    "cols = w // grid_spacing\n",
    "for x in np.linspace(start=0, stop=w, num=cols + 1):\n",
    "    x = int(round(x))\n",
    "    cv.line(map, (x, 0), (x, h), color=grid_color, thickness=1)\n",
    "    cv.putText(map, str(x), (x, 20), cv.FONT_HERSHEY_SIMPLEX, 0.5, grid_color, 1, cv.LINE_AA)\n",
    "for y in np.linspace(start=0, stop=h, num=rows + 1):\n",
    "    y = int(round(y))\n",
    "    cv.line(map, (0, y), (w, y), color=grid_color, thickness=1)\n",
    "    cv.putText(map, str(y), (10, y), cv.FONT_HERSHEY_SIMPLEX, 0.5, grid_color, 1, cv.LINE_AA)\n",
    "cv.imwrite(r'C:\\Users\\shwet\\Desktop\\yesist final\\maps\\map_with_contours_and_grid.jpg', map)\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "import os\n",
    "\n",
    "from googleapiclient.discovery import build\n",
    "from google.oauth2 import service_account\n",
    "from googleapiclient.http import MediaFileUpload\n",
    "\n",
    "SCOPES = ['https://www.googleapis.com/auth/drive']\n",
    "SERVICE_ACCOUNT_FILE = 'service_account.json'\n",
    "\n",
    "FOLDER_MAPPING = {\n",
    "    r\"C:\\Users\\shwet\\Desktop\\yesist final\\annulus ring strips\": \"1XAztJV-V2-3ZG9OLceJm3Q53xLj22IdY\",\n",
    "    r\"C:\\Users\\shwet\\Desktop\\yesist final\\color_maps\": \"1kcNgA5bqmnlwOvYy-ljsYRdVl9MFCjJA\",\n",
    "    r\"C:\\Users\\shwet\\Desktop\\yesist final\\mapped rings\": \"1Qepap7ch6bX0RDAcl6RaoEHM66H6toXg\",\n",
    "    r\"C:\\Users\\shwet\\Desktop\\yesist final\\maps\": \"17oLJ6_A4pBiurB0jR11NZXxv9bDIwkWL\",\n",
    "    r\"C:\\Users\\shwet\\Desktop\\yesist final\\rectangular strips\": \"18TAgAJptoQNfGM3GBbO65xU6YwpWXcFH\",\n",
    "    r\"C:\\Users\\shwet\\Desktop\\yesist final\\frames\": \"1cbKT2VHtm-8VA5Vw-J9o25bTyDJAE4vL\"\n",
    "}\n",
    "\n",
    "def authenticate():\n",
    "    creds = service_account.Credentials.from_service_account_file(SERVICE_ACCOUNT_FILE, scopes=SCOPES)\n",
    "    return creds\n",
    "\n",
    "def upload_files(local_folder, drive_folder_id):\n",
    "    creds = authenticate()\n",
    "    service = build('drive', 'v3', credentials=creds)\n",
    "    \n",
    "    if os.path.isdir(local_folder):\n",
    "        for filename in os.listdir(local_folder):\n",
    "            file_full_path = os.path.join(local_folder, filename)\n",
    "            if os.path.isfile(file_full_path):\n",
    "                file_metadata = {\n",
    "                    'name': filename,\n",
    "                    'parents': [drive_folder_id]\n",
    "                }\n",
    "                media = MediaFileUpload(file_full_path)\n",
    "                file = service.files().create(\n",
    "                    body=file_metadata,\n",
    "                    media_body=media\n",
    "                ).execute()\n",
    "                print(f'Uploaded {filename}')\n",
    "    else:\n",
    "        print(f'{local_folder} is not a directory')\n",
    "\n",
    "for local_folder, drive_folder_id in FOLDER_MAPPING.items():\n",
    "    upload_files(local_folder, drive_folder_id)"
   ]
  }
 ],
 "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.11.6"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}