2_ObtencionNegativos.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 11,
   "metadata": {},
   "outputs": [],
   "source": [
    "import utils.yolo_fun as yolo_fun\n",
    "import utils.img_fun as img_fun\n",
    "import os\n",
    "import pandas as pd\n",
    "from tqdm import tqdm  \n",
    "import rasterio\n",
    "from rasterio.windows import Window\n",
    "from rasterio.errors import RasterioIOError\n",
    "import numpy as np\n",
    "import shutil\n",
    "\n",
    "#valid_tiles = [13, 14, 15, 22, 23, 24, 35, 44, 45, 51, 52, 53, 54, 61, 62, 63, 70, 71, 72, 73, 81, 82]  \n",
    "\n",
    "orthomosiac_coords = os.path.join('coords', 'yolo_coords.csv')\n",
    "coords_dir_negatives = os.path.join('coords', 'negatives')\n",
    "\n",
    "os.makedirs(coords_dir_negatives, exist_ok=True)\n",
    "os.makedirs(coords_dir_normalized, exist_ok=True)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 7,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "\n",
      "\n",
      "Recortando imagen 24...\n",
      "_________________________________________________________\n",
      "Metadata:\n",
      "---------\n",
      "driver: GTiff\n",
      "dtype: uint8\n",
      "nodata: None\n",
      "width: 10195\n",
      "height: 11420\n",
      "count: 4\n",
      "crs: EPSG:4326\n",
      "transform: | 0.00, 0.00,-59.23|\n",
      "| 0.00,-0.00,-62.29|\n",
      "| 0.00, 0.00, 1.00|\n",
      "blockxsize: 10195\n",
      "blockysize: 1\n",
      "tiled: False\n",
      "interleave: pixel\n",
      "\n",
      "Coordenadas de las esquinas de la imagen:\n",
      "TOP LEFT: (-59.22587875249678, -62.290013442834734)\n",
      "BOTTOM RIGHT: (-59.22062349506678, -62.29275364899473)\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n",
      "Coordenadas vacías... guardando\n"
     ]
    }
   ],
   "source": [
    "\n",
    "# Tile 24 tiene focas y pinguinos descansando\n",
    "\n",
    "for n_tile in range(24, 25):\n",
    "    try:\n",
    "        # PARTE 1: Cargar la imagen y recortarla en imágenes más pequeñas de aproximadamente 500x500 píxeles\n",
    "        print(f\"\\n\\nRecortando imagen {n_tile}...\")\n",
    "        print('_________________________________________________________')\n",
    "        NUM_TILE = n_tile\n",
    "        path_doctorado = 'G:\\\\.shortcut-targets-by-id\\\\1pYgV5EIk4-LapLNhlCwpQaDAzuqNffXG\\\\doctorado_albert\\\\conteo_pinguinos'\n",
    "        subrecortes_dir = os.path.join('cut_tiles', 'negatives')\n",
    "        os.makedirs(subrecortes_dir, exist_ok=True)\n",
    "\n",
    "        image_name = f\"recortes/recorte_{NUM_TILE}.tif\"\n",
    "        tiff_file = os.path.join(path_doctorado, image_name)\n",
    "\n",
    "\n",
    "        # Sacamos un diccionario con toda la información de la imagen\n",
    "        img_info = img_fun.get_img_info(tiff_file)\n",
    "        WIDTH = img_info[\"width\"]\n",
    "        HEIGHT = img_info[\"height\"]\n",
    "        TOP_LEFT = img_info[\"top_left\"]\n",
    "        BOTTOM_RIGHT = img_info[\"bottom_right\"]\n",
    "        min_x, max_y = img_info['top_left']\n",
    "        max_x, min_y = img_info['bottom_right']\n",
    "\n",
    "        img_fun.crop_tile_into_subrecortes(\n",
    "            tiff_file = tiff_file, \n",
    "            output_dir = subrecortes_dir, \n",
    "            num_tile = NUM_TILE,\n",
    "            is_negative = True\n",
    "        )\n",
    "\n",
    "\n",
    "\n",
    "\n",
    "    except RasterioIOError as e:\n",
    "        print(f\"Error al cargar la imagen con rasterio: {e}\")\n",
    "        continue\n",
    "    except FileNotFoundError as e:\n",
    "        print(f\"Error: {e}\")\n",
    "        continue\n",
    "    except Exception as e:\n",
    "        print(f\"Error: {e}\")\n",
    "        continue"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "### Creación de las etiquetas txt\n",
    "\n",
    "Después de seleccionar exactamente los negativos que queremos, ahora sí que podemos generar los txt. Puesto que los txt están vacíos, bastará con pegarlos en nuestra carpeta de datasets y ya estarán listos para ser usados.\n",
    "\n",
    "Para tenerlos mejor identificados, los 20 primeros (o 30 o 40, si tenemos más) -con cierta variedad- serán para el conjunto de validación. El resto, para train."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 12,
   "metadata": {},
   "outputs": [],
   "source": [
    " # PARTE 3: ASIGNACIÓN DE LABELS EN TXT A CADA SUBRECORTE\n",
    "for img in os.listdir(subrecortes_dir):\n",
    "    if not img.lower().endswith(('.tif', '.tiff', '.jpg', '.png')):\n",
    "        continue\n",
    "\n",
    "    txt_name = img.split('.')[0] + '.txt'\n",
    "    txt_path = os.path.join(coords_dir_negatives, txt_name)\n",
    "    with open(txt_path, 'w') as f:\n",
    "        f.write(\"\")  "
   ]
  }
 ],
 "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.12.3"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}