notebooks/Voc2012 to VL Bounding Box input format/voc2012 to VL Annoation.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "cd132616",
   "metadata": {},
   "outputs": [],
   "source": [
    "\"\"\"\n",
    "### Jupyter Notebook Documentation\n",
    "\n",
    "This script processes an XML annotation file and converts it into a structured CSV format for Visual Layer. \n",
    "\n",
    "#### Steps:\n",
    "1. Read the XML file and parse the elements.\n",
    "2. Extract relevant fields such as filename, bounding box coordinates, and object labels.\n",
    "3. Include the full XML annotation as a field in the CSV for reference.\n",
    "4. Save the extracted data in `output.csv`.\n",
    "\n",
    "#### Output CSV Format:\n",
    "| filename  | col_x | row_y | width | height | label  | xml_annotation |\n",
    "|-----------|-------|-------|-------|--------|--------|----------------|\n",
    "| 00149.jpg | 1718  | 1043  | 27    | 38     | person | (full XML data) |\n",
    "\n",
    "#### How to Run:\n",
    "Run this script in a Jupyter Notebook cell to process the XML file and generate the CSV.\n",
    "\"\"\"\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "3d422076",
   "metadata": {},
   "outputs": [],
   "source": [
    "import xml.etree.ElementTree as ET\n",
    "import csv\n",
    "\n",
    "# Fixed XML string\n",
    "XML_STRING = \"\"\"\n",
    "<annotation>\n",
    "    <filename>00149.jpg</filename>\n",
    "    <object>\n",
    "        <class>person</class>\n",
    "        <bndbox>\n",
    "            <xmin>1718</xmin>\n",
    "            <ymin>1043</ymin>\n",
    "            <xmax>1745</xmax>\n",
    "            <ymax>1081</ymax>\n",
    "        </bndbox>\n",
    "    </object>\n",
    "    <object>\n",
    "        <class>car</class>\n",
    "        <bndbox>\n",
    "            <xmin>1296</xmin>\n",
    "            <ymin>1036</ymin>\n",
    "            <xmax>1335</xmax>\n",
    "            <ymax>1074</ymax>\n",
    "        </bndbox>\n",
    "    </object>\n",
    "</annotation>\n",
    "\"\"\"\n",
    "\n",
    "# Function to parse XML and convert to CSV\n",
    "def convert_xml_to_csv(csv_file):\n",
    "    root = ET.fromstring(XML_STRING)\n",
    "    \n",
    "    # Open CSV file for writing\n",
    "    with open(csv_file, mode='w', newline='') as file:\n",
    "        writer = csv.writer(file)\n",
    "        writer.writerow([\"filename\", \"col_x\", \"row_y\", \"width\", \"height\", \"label\", \"xml_annotation\"])\n",
    "        \n",
    "        filename = root.find('filename').text\n",
    "        xml_annotation = XML_STRING\n",
    "        \n",
    "        for obj in root.findall('object'):\n",
    "            label = obj.find('class').text\n",
    "            bndbox = obj.find('bndbox')\n",
    "            xmin = int(bndbox.find('xmin').text)\n",
    "            ymin = int(bndbox.find('ymin').text)\n",
    "            xmax = int(bndbox.find('xmax').text)\n",
    "            ymax = int(bndbox.find('ymax').text)\n",
    "            \n",
    "            width = xmax - xmin\n",
    "            height = ymax - ymin\n",
    "            \n",
    "            writer.writerow([filename, xmin, ymin, width, height, label, xml_annotation])\n",
    "\n",
    "# Example usage\n",
    "csv_file = \"object_annoations.csv\"  # Replace with desired CSV output file\n",
    "convert_xml_to_csv(csv_file)\n",
    "print(f\"Conversion complete. CSV saved to {csv_file}\")"
   ]
  }
 ],
 "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.10.12"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 5
}