awoo

dataset_tools/Create Empty caption Files for Images.ipynb

@@ -2,7 +2,7 @@
  "cells": [
   {
    "cell_type": "code",
-   "execution_count": 2,
+   "execution_count": null,
    "metadata": {},
    "outputs": [
     {
@@ -25,20 +25,17 @@
     }
    ],
    "source": [
-    "import os\n",
+    "from pathlib import Path\n",
     "\n",
     "def create_caption_file(directory):\n",
-    "    for root, dirs, files in os.walk(directory):\n",
-    "        for file in files:\n",
-    "            if file.lower().endswith(('.jpg', '.jpeg', '.png')):\n",
-    "                image_path = os.path.join(root, file)\n",
-    "                caption_file = os.path.splitext(image_path)[0] + \".caption\"\n",
-    "                if not os.path.exists(caption_file):\n",
-    "                    with open(caption_file, 'w'):\n",
-    "                        pass\n",
+    "    for image_path in Path(directory).rglob('*'):\n",
+    "        if image_path.suffix.lower() in ['.jpg', '.jpeg', '.png']:\n",
+    "            caption_file = image_path.with_suffix('.caption')\n",
+    "            if not caption_file.exists():\n",
+    "                caption_file.touch()\n",
     "\n",
     "if __name__ == \"__main__\":\n",
-    "    directory = r'C:\\Users\\kade\\Desktop\\training_dir_staging'\n",
+    "    directory = r'E:\\training_dir_staging'\n",
     "    create_caption_file(directory)\n",
     "    print(\"Caption files created successfully.\")"
    ]
@@ -60,7 +57,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.12.2"
+   "version": "3.12.3"
   }
  },
  "nbformat": 4,