PNG to JPG and other changes in resize

Author: anushkaray

notebooks/resize.ipynb

@@ -2,7 +2,7 @@
  "cells": [
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 1,
    "id": "ab143ef4-f298-46ed-b4ba-4ae1a8b15154",
    "metadata": {},
    "outputs": [],
@@ -193,20 +193,90 @@
     "    resize_images_uav(input_paths[i], output_paths[i])"
    ]
   },
+  {
+   "cell_type": "markdown",
+   "id": "05832748-102f-4742-ab5c-4ef9ed455b5b",
+   "metadata": {},
+   "source": [
+    "Convert Nearmap Images from PNG to JPG"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 2,
+   "id": "fed8594e-3475-4f09-a92d-fd9d3f771703",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "data_root = \"/home/gridsan/aray/meng/mmsegmentation/data/dataset_mmseg\" #ANUSHKA"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 11,
+   "id": "ff8b40ac-de73-4a7e-8427-a9a037c4ef55",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "nearmap_img = os.path.join(data_root, \"img_dir_nearmap/\")\n",
+    "output_img = os.path.join(data_root, \"img_dir_nearmap_jpg/\")\n",
+    "nearmap_ann = os.path.join(data_root, \"ann_dir_nearmap/\")\n",
+    "output_ann = os.path.join(data_root, \"ann_dir_nearmap_jpg/\")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 3,
+   "id": "8fc18838-7b7a-489f-a13c-b95e165dfe06",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "#UAV Directories \n",
+    "uav_img = os.path.join(data_root, \"img_dir_uav/\")\n",
+    "output_img = os.path.join(data_root, \"img_dir_uav_jpg/\")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 4,
+   "id": "61e4bdda-e3c5-4c7e-92d9-0df4986c0c81",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "def change_jpg(path, output):\n",
+    "    sub_dirs = ['train/', 'test/', 'val/']\n",
+    "    for sub_dir in sub_dirs:\n",
+    "        dirs = os.listdir(path+sub_dir)\n",
+    "        for item in dirs:\n",
+    "            if os.path.isfile(path+sub_dir+item):\n",
+    "                im = Image.open(path+sub_dir+item)\n",
+    "                im = im.convert('RGB')\n",
+    "                filename = item.split('.')[0]\n",
+    "                im.save(output+sub_dir+filename+'.jpg')\n",
+    "\n",
+    "# dirs = [uav_img, nearmap_ann]\n",
+    "# outputs = [output_img, output_ann]\n",
+    "# for i in range(len(dirs)):\n",
+    "#     change_jpg(dirs[i], outputs[i])\n",
+    "    \n",
+    "change_jpg(uav_img, output_img)\n",
+    "\n"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "666617ee-782a-4bf2-b49c-53bb83ee3830",
+   "id": "1de781eb-7d84-4a17-94d9-23f067537023",
    "metadata": {},
    "outputs": [],
    "source": []
   }
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": "open-mmlab",
+   "display_name": "Python 3",
    "language": "python",
-   "name": "open-mmlab"
+   "name": "python3"
   },
   "language_info": {
    "codemirror_mode": {