notebooks/Autolabel-ZST.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [],
   "source": [
    "#!pip install opencv-python zstandard python_on_whales scikit-image"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "metadata": {},
   "outputs": [],
   "source": [
    "import os\n",
    "\n",
    "import data\n",
    "import auto_label"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "metadata": {},
   "outputs": [],
   "source": [
    "\n",
    "\n",
    "def prepare_and_label(file:str, model:str):\n",
    "\n",
    "    try: \n",
    "        sig_data = data.Data(file)\n",
    "    except ValueError:\n",
    "        print(f\"Warning: Skipping {file}\")\n",
    "        return None\n",
    "    \n",
    "    # # print(f\"Sample Rate: {sig_data.sample_rate}\")\n",
    "    # # print(f\"Center Frequency: {sig_data.center_freq}\")\n",
    "    # # print(f\"Duration: {sig_data.duration}\")\n",
    "    # print(\"Exporting SIGMF data...\")\n",
    "\n",
    "    try:\n",
    "        sig_data.export_sigmf_data(overwrite=True)\n",
    "    except:\n",
    "        return None\n",
    "    # sig_data.zst_to_sigmf_data()\n",
    "    # n_fft = 1024\n",
    "    # time_dim = 512 # equivalent to tune_step_fft parameter in GamutRF\n",
    "    # n_samples = n_fft * time_dim\n",
    "    # print(f\"Generating spectrograms with n_samples = {n_samples} and n_fft = {n_fft}...\")\n",
    "    # sig_data.generate_spectrograms(n_samples, n_fft)\n",
    "    # print(\"Auto-labeling spectrograms...\")\n",
    "    # sig_data.auto_label_spectrograms(model)\n",
    "    # # print(\"Exporting annotation IQ data...\")\n",
    "    # # sig_data.export_annotation_iq()\n",
    "\n",
    "def auto_label_ZST_directory(path:str, model:str):\n",
    "    if os.path.isdir(path):\n",
    "        for file in os.listdir(path):\n",
    "            if file.endswith('.zst'):\n",
    "                # Process the file here\n",
    "                full_path = os.path.join(path, file)\n",
    "                print(full_path)\n",
    "                prepare_and_label(full_path, model)\n",
    "    else:\n",
    "        print(f\"{path} is not a directory.\")\n",
    "    "
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "dev_data/torchsig_test/fhss_css/1690987701_2450000000Hz_20480000sps.ci16.zst\n"
     ]
    }
   ],
   "source": [
    "path = \"dev_data/torchsig_test/fhss_css\"\n",
    "auto_label_ZST_directory(path, model=\"fhss_css\")"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "metadata": {},
   "outputs": [],
   "source": [
    "n_annotations = 0\n",
    "for file in os.listdir(path):\n",
    "    if file.endswith('.zst'):\n",
    "        # Process the file here\n",
    "        full_path = os.path.join(path, file)\n",
    "        d = data.Data(full_path)\n",
    "        n_annotations += len(d.metadata[\"annotations\"])"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 6,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "296"
      ]
     },
     "execution_count": 6,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "n_annotations"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 7,
   "metadata": {},
   "outputs": [],
   "source": [
    "\n",
    "\n",
    "def auto_label_directory(path:str, force_model:str = None):\n",
    "\n",
    "\n",
    "    \n",
    "    if not os.path.isdir(path):\n",
    "        print(f\"Error: {path} is not a directory\")\n",
    "    \n",
    "    sample_files = [os.path.join(dirpath, f)  for dirpath, dirnames, filenames in os.walk(path) for f in filenames if f.endswith((\".zst\", \".raw\")) ]\n",
    "    for f in sample_files:\n",
    "\n",
    "        model = force_model if force_model else get_model(f)\n",
    "\n",
    "        if model:    \n",
    "            print(f\"Info: Trying to label {f} with model {model}\")\n",
    "            prepare_and_label(f, model)\n",
    "\n",
    "def get_model(filepath:str):\n",
    "\n",
    "    model_path_filters = {\n",
    "        #TODO: add Immersion_Ghost/open_air_collect/fpv_medium\n",
    "        #TODO: debug Immersion_Ghost/open_air_collect/fpv_small\n",
    "        \"wifi\": {\n",
    "            \"include\": [\"wifi\"],\n",
    "            \"exclude\": [],\n",
    "        },\n",
    "        \"tbs_crossfire\": {\n",
    "            \"include\": [\"TBS_CRSF\"],\n",
    "            \"exclude\": [],\n",
    "        },\n",
    "        \"fhss_css\": {\n",
    "            \"include\": [\"Immersion_Ghost/open_air_collect/fpv_large\", \"Immersion_Ghost/faraday_cage_collect\"],\n",
    "            \"exclude\": [\"msk\", \"fpv_drone_only\"],\n",
    "        },\n",
    "        \"msk\": {\n",
    "            \"include\": [\"msk\"],\n",
    "            \"exclude\": [],\n",
    "        },\n",
    "        \"mini2\": {\n",
    "            \"include\": [\"mini2\", \"pdx_field_day\", \"leesburg_field_day\"],\n",
    "            \"exclude\": [],\n",
    "        },\n",
    "        \"mavic3\": {\n",
    "            \"include\": [\"mavic3\"],\n",
    "            \"exclude\": [],\n",
    "        },        \n",
    "    }\n",
    "\n",
    "    models = []\n",
    "    for model_name in model_path_filters:\n",
    "        match = False\n",
    "        for include in model_path_filters[model_name][\"include\"]:\n",
    "            if include in filepath:\n",
    "                match = True\n",
    "                break\n",
    "        if match:\n",
    "            for exclude in model_path_filters[model_name][\"exclude\"]:\n",
    "                if exclude in filepath:\n",
    "                    match = False\n",
    "                    break\n",
    "        if match:\n",
    "            models.append(model_name)\n",
    "    if len(models) == 0: \n",
    "        print(f\"Warning: Could not find a matching model for {filepath}\")\n",
    "        return None\n",
    "    elif len(models) > 1: \n",
    "        print(f\"Warning: Found multiple matching models [{models}] for {filepath}\")\n",
    "        return None\n",
    "    else:\n",
    "        return models[0]\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 10,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Info: Trying to label data/gamutrf/gamutrf-fpv/Immersion_Ghost/open_air_collect/fpv_large/fpv_pairing/2.45e9_20480000_10_10s_1690987570.ci16.zst with model fhss_css\n",
      "Info: Trying to label data/gamutrf/gamutrf-fpv/Immersion_Ghost/open_air_collect/fpv_large/fpv_pairing/2.45e9_20480000_10_10s_1690987644.ci16.zst with model fhss_css\n",
      "Info: Trying to label data/gamutrf/gamutrf-fpv/Immersion_Ghost/open_air_collect/fpv_large/fpv_pairing/2.45e9_20480000_10_10s_1690987608.ci16.zst with model fhss_css\n",
      "Warning: Could not find a matching model for data/gamutrf/gamutrf-fpv/Immersion_Ghost/open_air_collect/fpv_large/fpv_drone_only/5.917e9_20480000_0_10s_1690987398.ci16.zst\n",
      "Warning: Could not find a matching model for data/gamutrf/gamutrf-fpv/Immersion_Ghost/open_air_collect/fpv_large/fpv_drone_only/5.917e9_20480000_0_10s_1690987434.ci16.zst\n",
      "Warning: Could not find a matching model for data/gamutrf/gamutrf-fpv/Immersion_Ghost/open_air_collect/fpv_large/fpv_drone_only/5.917e9_20480000_10_10s_1690987261.ci16.zst\n",
      "Warning: Could not find a matching model for data/gamutrf/gamutrf-fpv/Immersion_Ghost/open_air_collect/fpv_large/fpv_drone_only/2.45e9_20480000_10_5s_1690987520.ci16.zst\n",
      "Info: Trying to label data/gamutrf/gamutrf-fpv/Immersion_Ghost/open_air_collect/fpv_large/fpv_no_telemetry/2.45e9_20480000_0_5s_1690988185.ci16.zst with model fhss_css\n",
      "Info: Trying to label data/gamutrf/gamutrf-fpv/Immersion_Ghost/open_air_collect/fpv_large/fpv_no_telemetry/2.46e9_20480000_10_5s_1690988089.ci16.zst with model fhss_css\n",
      "Info: Trying to label data/gamutrf/gamutrf-fpv/Immersion_Ghost/open_air_collect/fpv_large/fpv_no_telemetry/2.45e9_20480000_10_5s_1690988055.ci16.zst with model fhss_css\n",
      "Info: Trying to label data/gamutrf/gamutrf-fpv/Immersion_Ghost/open_air_collect/fpv_large/fpv_no_telemetry/2.44e9_20480000_10_5s_1690988118.ci16.zst with model fhss_css\n",
      "Info: Trying to label data/gamutrf/gamutrf-fpv/Immersion_Ghost/open_air_collect/fpv_large/fpv_msk/2.45e9_20480000_0_10s_1690988293.ci16.zst with model msk\n",
      "Info: Trying to label data/gamutrf/gamutrf-fpv/Immersion_Ghost/open_air_collect/fpv_large/fpv_msk/2.406e9_20480000_10_5s_1690988548.ci16.zst with model msk\n",
      "Info: Trying to label data/gamutrf/gamutrf-fpv/Immersion_Ghost/open_air_collect/fpv_large/fpv_msk/2.45e9_20480000_10_10s_1690988581.ci16.zst with model msk\n",
      "Info: Trying to label data/gamutrf/gamutrf-fpv/Immersion_Ghost/open_air_collect/fpv_large/fpv_msk/2.45e9_20480000_10_10s_1690988329.ci16.zst with model msk\n",
      "Info: Trying to label data/gamutrf/gamutrf-fpv/Immersion_Ghost/open_air_collect/fpv_large/fpv_msk/2.45e9_20480000_10_10s_1690988379.ci16.zst with model msk\n",
      "Info: Trying to label data/gamutrf/gamutrf-fpv/Immersion_Ghost/open_air_collect/fpv_large/fpv_flying/2.4e9_20480000_10_5s_1690987805.ci16.zst with model fhss_css\n",
      "Info: Trying to label data/gamutrf/gamutrf-fpv/Immersion_Ghost/open_air_collect/fpv_large/fpv_flying/2.45e9_20480000_10_5s_1690987701.ci16.zst with model fhss_css\n",
      "Info: Trying to label data/gamutrf/gamutrf-fpv/Immersion_Ghost/open_air_collect/fpv_large/fpv_flying/2.46e9_20480000_10_5s_1690987772.ci16.zst with model fhss_css\n",
      "Info: Trying to label data/gamutrf/gamutrf-fpv/Immersion_Ghost/open_air_collect/fpv_large/fpv_flying/2.45e9_20480000_10_5s_1690987918.ci16.zst with model fhss_css\n",
      "Warning: Could not find a matching model for data/gamutrf/gamutrf-fpv/Immersion_Ghost/open_air_collect/fpv_small/fpv_pairing/2.45e9_20480000_20_20s_1690986817.ci16.zst\n",
      "Warning: Could not find a matching model for data/gamutrf/gamutrf-fpv/Immersion_Ghost/open_air_collect/fpv_small/fpv_pairing/2.45e9_20480000_20_10s_1690986768.ci16.zst\n",
      "Warning: Could not find a matching model for data/gamutrf/gamutrf-fpv/Immersion_Ghost/open_air_collect/fpv_small/fpv_pairing/2.45e9_20480000_20_5s_1690986866.ci16.zst\n",
      "Warning: Could not find a matching model for data/gamutrf/gamutrf-fpv/Immersion_Ghost/open_air_collect/fpv_small/fpv_controller_only/2.4e9_20480000_20_10s_1690986229.ci16.zst\n",
      "Warning: Could not find a matching model for data/gamutrf/gamutrf-fpv/Immersion_Ghost/open_air_collect/fpv_small/fpv_controller_only/2.45e9_20480000_20_10s_1690986319.ci16.zst\n",
      "Warning: Could not find a matching model for data/gamutrf/gamutrf-fpv/Immersion_Ghost/open_air_collect/fpv_small/fpv_controller_only/2.45e9_20480000_20_10s_1690986280.ci16.zst\n",
      "Warning: Could not find a matching model for data/gamutrf/gamutrf-fpv/Immersion_Ghost/open_air_collect/fpv_small/fpv_drone_only/2.45e9_20480000_20_10s_1690986466.ci16.zst\n",
      "Warning: Could not find a matching model for data/gamutrf/gamutrf-fpv/Immersion_Ghost/open_air_collect/fpv_small/fpv_drone_only/5.917e9_20480000_20_10s_1690986591.ci16.zst\n",
      "Warning: Could not find a matching model for data/gamutrf/gamutrf-fpv/Immersion_Ghost/open_air_collect/fpv_small/fpv_drone_only/5.917e9_20480000_10_10s_1690986661.ci16.zst\n",
      "Warning: Could not find a matching model for data/gamutrf/gamutrf-fpv/Immersion_Ghost/open_air_collect/fpv_small/fpv_drone_only/5.917e9_20480000_20_10s_1690986561.ci16.zst\n",
      "Warning: Could not find a matching model for data/gamutrf/gamutrf-fpv/Immersion_Ghost/open_air_collect/fpv_small/fpv_drone_only/2.45e9_20480000_20_10s_1690986418.ci16.zst\n",
      "Warning: Could not find a matching model for data/gamutrf/gamutrf-fpv/Immersion_Ghost/open_air_collect/fpv_small/fpv_drone_only/5.917e9_20480000_10_10s_1690986697.ci16.zst\n",
      "Warning: Could not find a matching model for data/gamutrf/gamutrf-fpv/Immersion_Ghost/open_air_collect/fpv_small/fpv_drone_only/5.917e9_20480000_20_10s_1690986624.ci16.zst\n",
      "Warning: Could not find a matching model for data/gamutrf/gamutrf-fpv/Immersion_Ghost/open_air_collect/fpv_small/fpv_flying/2.45e9_20480000_20_10s_1690987025.ci16.zst\n",
      "Warning: Could not find a matching model for data/gamutrf/gamutrf-fpv/Immersion_Ghost/open_air_collect/fpv_small/fpv_flying/2.46e9_20480000_20_10s_1690986978.ci16.zst\n",
      "Warning: Could not find a matching model for data/gamutrf/gamutrf-fpv/Immersion_Ghost/open_air_collect/fpv_small/fpv_flying/2.45e9_20480000_20_5s_1690986903.ci16.zst\n",
      "Warning: Could not find a matching model for data/gamutrf/gamutrf-fpv/Immersion_Ghost/open_air_collect/fpv_small/fpv_flying/2.46e9_20480000_20_5s_1690986959.ci16.zst\n",
      "Warning: Could not find a matching model for data/gamutrf/gamutrf-fpv/Immersion_Ghost/open_air_collect/fpv_small/fpv_flying/5.917e9_20480000_10_10s_1690987052.ci16.zst\n",
      "Warning: Could not find a matching model for data/gamutrf/gamutrf-fpv/Immersion_Ghost/open_air_collect/fpv_small/fpv_flying/5.917e9_20480000_10_10s_1690987111.ci16.zst\n",
      "Warning: Could not find a matching model for data/gamutrf/gamutrf-fpv/Immersion_Ghost/open_air_collect/fpv_medium/fpv_goggles_only/5.839e9_20480000_20_5s_1690990003.ci16.zst\n",
      "Warning: Could not find a matching model for data/gamutrf/gamutrf-fpv/Immersion_Ghost/open_air_collect/fpv_medium/fpv_goggles_only/5.839e9_20480000_10_5s_1690989963.ci16.zst\n",
      "Warning: Could not find a matching model for data/gamutrf/gamutrf-fpv/Immersion_Ghost/open_air_collect/fpv_medium/fpv_goggles_high_bw/5.878e9_56000000_00_5s_1690990556.ci16.zst\n",
      "Warning: Could not find a matching model for data/gamutrf/gamutrf-fpv/Immersion_Ghost/open_air_collect/fpv_medium/fpv_goggles_high_bw/5.878e9_56000000_10_5s_1690990517.ci16.zst\n",
      "Warning: Could not find a matching model for data/gamutrf/gamutrf-fpv/Immersion_Ghost/open_air_collect/fpv_medium/fpv_goggles_high_bw/5.878e9_56000000_10_5s_1690990471.ci16.zst\n",
      "Warning: Could not find a matching model for data/gamutrf/gamutrf-fpv/Immersion_Ghost/open_air_collect/fpv_medium/fpv_goggles_low_bw/5.91e9_20480000_10_5s_1690990195.ci16.zst\n",
      "Warning: Could not find a matching model for data/gamutrf/gamutrf-fpv/Immersion_Ghost/open_air_collect/fpv_medium/fpv_goggles_low_bw/5.914e9_20480000_10_5s_1690990162.ci16.zst\n",
      "Warning: Could not find a matching model for data/gamutrf/gamutrf-fpv/Immersion_Ghost/open_air_collect/fpv_medium/fpv_goggles_low_bw/5.914e9_20480000_10_5s_1690990261.ci16.zst\n",
      "Warning: Could not find a matching model for data/gamutrf/gamutrf-fpv/Immersion_Ghost/open_air_collect/fpv_medium/fpv_goggles_low_bw/5.92e9_20480000_10_5s_1690990229.ci16.zst\n",
      "Warning: Could not find a matching model for data/gamutrf/gamutrf-fpv/Immersion_Ghost/open_air_collect/fpv_medium/fpv_goggles_low_bw/5.917e9_20480000_10_5s_1690990058.ci16.zst\n",
      "Warning: Could not find a matching model for data/gamutrf/gamutrf-fpv/Immersion_Ghost/open_air_collect/fpv_medium/fpv_goggles_low_bw/5.914e9_20480000_10_5s_1690990094.ci16.zst\n",
      "Info: Trying to label data/gamutrf/gamutrf-fpv/Immersion_Ghost/faraday_cage_collect/msk_500Hz_drone_only/2.46e9_56000000_-20_5s_1692801742.ci16.zst with model msk\n",
      "Info: Trying to label data/gamutrf/gamutrf-fpv/Immersion_Ghost/faraday_cage_collect/msk_500Hz_drone_only/2.46e9_56000000_0_5s_1692801873.ci16.zst with model msk\n",
      "Info: Trying to label data/gamutrf/gamutrf-fpv/Immersion_Ghost/faraday_cage_collect/msk_500Hz_drone_only/2.42e9_56000000_0_5s_1692801781.ci16.zst with model msk\n",
      "Info: Trying to label data/gamutrf/gamutrf-fpv/Immersion_Ghost/faraday_cage_collect/msk_500Hz_drone_only/2.42e9_56000000_20_5s_1692801830.ci16.zst with model msk\n",
      "Info: Trying to label data/gamutrf/gamutrf-fpv/Immersion_Ghost/faraday_cage_collect/166Hz_drone_only/2.46e9_56000000_0_5s_1692800459.ci16.zst with model fhss_css\n",
      "Info: Trying to label data/gamutrf/gamutrf-fpv/Immersion_Ghost/faraday_cage_collect/166Hz_drone_only/2.44e9_56000000_-20_5s_1692800161.ci16.zst with model fhss_css\n",
      "Info: Trying to label data/gamutrf/gamutrf-fpv/Immersion_Ghost/faraday_cage_collect/166Hz_drone_only/2.42e9_56000000_-20_5s_1692800612.ci16.zst with model fhss_css\n",
      "Info: Trying to label data/gamutrf/gamutrf-fpv/Immersion_Ghost/faraday_cage_collect/166Hz_drone_only/2.46e9_56000000_0_5s_1692800346.ci16.zst with model fhss_css\n",
      "Info: Trying to label data/gamutrf/gamutrf-fpv/Immersion_Ghost/faraday_cage_collect/166Hz_drone_only/2.44e9_56000000_-20_5s_1692800232.ci16.zst with model fhss_css\n",
      "Info: Trying to label data/gamutrf/gamutrf-fpv/Immersion_Ghost/faraday_cage_collect/166Hz_drone_only/2.5e9_56000000_0_5s_1692800400.ci16.zst with model fhss_css\n",
      "Info: Trying to label data/gamutrf/gamutrf-fpv/Immersion_Ghost/faraday_cage_collect/166Hz_drone_only/2.42e9_56000000_0_5s_1692800502.ci16.zst with model fhss_css\n",
      "Info: Trying to label data/gamutrf/gamutrf-fpv/Immersion_Ghost/faraday_cage_collect/166Hz_drone_only/2.46e9_56000000_-20_5s_1692800279.ci16.zst with model fhss_css\n",
      "Info: Trying to label data/gamutrf/gamutrf-fpv/Immersion_Ghost/faraday_cage_collect/166Hz_drone_only/2.42e9_56000000_-20_5s_1692800088.ci16.zst with model fhss_css\n",
      "Info: Trying to label data/gamutrf/gamutrf-fpv/Immersion_Ghost/faraday_cage_collect/166Hz_drone_only/baseline/2.42e9_56000000_-20_5s_1692799926.ci16.zst with model fhss_css\n",
      "Info: Trying to label data/gamutrf/gamutrf-fpv/Immersion_Ghost/faraday_cage_collect/166Hz/2.46e9_56000000_-20_5s_1692801237.ci16.zst with model fhss_css\n",
      "Info: Trying to label data/gamutrf/gamutrf-fpv/Immersion_Ghost/faraday_cage_collect/166Hz/2.42e9_56000000_0_5s_1692801180.ci16.zst with model fhss_css\n",
      "Info: Trying to label data/gamutrf/gamutrf-fpv/Immersion_Ghost/faraday_cage_collect/166Hz/2.42e9_56000000_-20_5s_1692801265.ci16.zst with model fhss_css\n",
      "Info: Trying to label data/gamutrf/gamutrf-fpv/Immersion_Ghost/faraday_cage_collect/166Hz/2.46e9_56000000_0_5s_1692801207.ci16.zst with model fhss_css\n",
      "Info: Trying to label data/gamutrf/gamutrf-fpv/Immersion_Ghost/faraday_cage_collect/no_telemetry_drone_only/2.42e9_56000000_0_5s_1692800923.ci16.zst with model fhss_css\n",
      "Info: Trying to label data/gamutrf/gamutrf-fpv/Immersion_Ghost/faraday_cage_collect/no_telemetry_drone_only/2.46e9_56000000_0_5s_1692800773.ci16.zst with model fhss_css\n",
      "Info: Trying to label data/gamutrf/gamutrf-fpv/Immersion_Ghost/faraday_cage_collect/no_telemetry_drone_only/2.42e9_56000000_-20_5s_1692800730.ci16.zst with model fhss_css\n",
      "Info: Trying to label data/gamutrf/gamutrf-fpv/Immersion_Ghost/faraday_cage_collect/no_telemetry/2.46e9_56000000_0_5s_1692801088.ci16.zst with model fhss_css\n",
      "Info: Trying to label data/gamutrf/gamutrf-fpv/Immersion_Ghost/faraday_cage_collect/no_telemetry/2.42e9_56000000_0_5s_1692801021.ci16.zst with model fhss_css\n",
      "Info: Trying to label data/gamutrf/gamutrf-fpv/Immersion_Ghost/faraday_cage_collect/msk_500Hz/2.46e9_56000000_0_5s_1692802061.ci16.zst with model msk\n",
      "Info: Trying to label data/gamutrf/gamutrf-fpv/Immersion_Ghost/faraday_cage_collect/msk_500Hz/2.42e9_56000000_0_5s_1692802012.ci16.zst with model msk\n",
      "Info: Trying to label data/gamutrf/gamutrf-fpv/Immersion_Ghost/faraday_cage_collect/msk_500Hz/2.46e9_56000000_0_5s_1692801937.ci16.zst with model msk\n",
      "Info: Trying to label data/gamutrf/gamutrf-fpv/Immersion_Ghost/faraday_cage_collect/msk_drone_only/2.46e9_56000000_0_5s_1692801554.ci16.zst with model msk\n",
      "Info: Trying to label data/gamutrf/gamutrf-fpv/Immersion_Ghost/faraday_cage_collect/msk_drone_only/2.42e9_56000000_0_5s_1692801602.ci16.zst with model msk\n",
      "Info: Trying to label data/gamutrf/gamutrf-fpv/Immersion_Ghost/faraday_cage_collect/msk_drone_only/2.46e9_56000000_-20_5s_1692801645.ci16.zst with model msk\n",
      "Info: Trying to label data/gamutrf/gamutrf-fpv/Immersion_Ghost/faraday_cage_collect/msk/2.46e9_56000000_-20_5s_1692801411.ci16.zst with model msk\n",
      "Info: Trying to label data/gamutrf/gamutrf-fpv/Immersion_Ghost/faraday_cage_collect/msk/2.42e9_56000000_-20_5s_1692801383.ci16.zst with model msk\n",
      "Info: Trying to label data/gamutrf/gamutrf-fpv/Immersion_Ghost/faraday_cage_collect/msk/2.46e9_56000000_0_5s_1692801498.ci16.zst with model msk\n",
      "Warning: Could not find a matching model for data/gamutrf/gamutrf-fpv/ELRS/2.42e9_56000000_0_5s_1692803442.ci16.zst\n",
      "Warning: Could not find a matching model for data/gamutrf/gamutrf-fpv/ELRS/2.42e9_56000000_-20_5s_1692735799.ci16.zst\n",
      "Warning: Could not find a matching model for data/gamutrf/gamutrf-fpv/ELRS/2.42e9_56000000_10_5s_1692735689.ci16.zst\n",
      "Warning: Could not find a matching model for data/gamutrf/gamutrf-fpv/ELRS/2.42e9_56000000_0_5s_1692735761.ci16.zst\n",
      "Warning: Could not find a matching model for data/gamutrf/gamutrf-fpv/ELRS/2.42e9_56000000_50_5s_1692735515.ci16.zst\n",
      "Info: Trying to label data/gamutrf/gamutrf-fpv/TBS_CRSF/915/50Hz/915e6_56000000_30_5s_1692635831.ci16.zst with model tbs_crossfire\n",
      "Info: Trying to label data/gamutrf/gamutrf-fpv/TBS_CRSF/915/50Hz/915e6_56000000_30_5s_1692635867.ci16.zst with model tbs_crossfire\n",
      "Info: Trying to label data/gamutrf/gamutrf-fpv/TBS_CRSF/915/narrow_record/910e6_2560000_30_5s_1692635724.ci16.zst with model tbs_crossfire\n",
      "Info: Trying to label data/gamutrf/gamutrf-fpv/TBS_CRSF/915/narrow_record/910e6_5120000_30_5s_1692635689.ci16.zst with model tbs_crossfire\n",
      "Info: Trying to label data/gamutrf/gamutrf-fpv/TBS_CRSF/915/controller_only/915e6_56000000_30_5s_1692639060.ci16.zst with model tbs_crossfire\n",
      "Info: Trying to label data/gamutrf/gamutrf-fpv/TBS_CRSF/915/controller_only/915e6_56000000_30_5s_1692638924.ci16.zst with model tbs_crossfire\n",
      "Info: Trying to label data/gamutrf/gamutrf-fpv/TBS_CRSF/915/controller_only/915e6_56000000_30_5s_1692639110.ci16.zst with model tbs_crossfire\n",
      "Info: Trying to label data/gamutrf/gamutrf-fpv/TBS_CRSF/915/150Hz/915e6_56000000_30_5s_1692636033.ci16.zst with model tbs_crossfire\n",
      "Info: Trying to label data/gamutrf/gamutrf-fpv/TBS_CRSF/915/150Hz/915e6_56000000_30_5s_1692635956.ci16.zst with model tbs_crossfire\n",
      "Info: Trying to label data/gamutrf/gamutrf-fpv/TBS_CRSF/915/no_telemetry/915e6_56000000_30_5s_1692635598.ci16.zst with model tbs_crossfire\n",
      "Info: Trying to label data/gamutrf/gamutrf-fpv/TBS_CRSF/915/no_telemetry/915e6_56000000_30_5s_1692635566.ci16.zst with model tbs_crossfire\n",
      "Info: Trying to label data/gamutrf/gamutrf-fpv/TBS_CRSF/915/factory_settings/915e6_20480000_10_5s_1692635111.ci16.zst with model tbs_crossfire\n",
      "Info: Trying to label data/gamutrf/gamutrf-fpv/TBS_CRSF/915/factory_settings/915e6_56000000_30_5s_1692635319.ci16.zst with model tbs_crossfire\n",
      "Info: Trying to label data/gamutrf/gamutrf-fpv/TBS_CRSF/915/factory_settings/915e6_56000000_30_5s_1692635401.ci16.zst with model tbs_crossfire\n",
      "Info: Trying to label data/gamutrf/gamutrf-fpv/TBS_CRSF/915/factory_settings/915e6_56000000_10_5s_1692635148.ci16.zst with model tbs_crossfire\n",
      "Info: Trying to label data/gamutrf/gamutrf-fpv/TBS_CRSF/915/factory_settings/915e6_56000000_10_5s_1692635279.ci16.zst with model tbs_crossfire\n",
      "Info: Trying to label data/gamutrf/gamutrf-fpv/TBS_CRSF/868/50Hz/868e6_56000000_30_5s_1692638010.ci16.zst with model tbs_crossfire\n",
      "Info: Trying to label data/gamutrf/gamutrf-fpv/TBS_CRSF/868/50Hz/868e6_56000000_30_5s_1692637975.ci16.zst with model tbs_crossfire\n",
      "Info: Trying to label data/gamutrf/gamutrf-fpv/TBS_CRSF/868/150Hz/868e6_56000000_30_5s_1692638556.ci16.zst with model tbs_crossfire\n",
      "Info: Trying to label data/gamutrf/gamutrf-fpv/TBS_CRSF/868/150Hz/868e6_56000000_30_5s_1692638355.ci16.zst with model tbs_crossfire\n",
      "Info: Trying to label data/gamutrf/gamutrf-fpv/TBS_CRSF/868/no_telemetry/868e6_56000000_30_5s_1692638192.ci16.zst with model tbs_crossfire\n",
      "Info: Trying to label data/gamutrf/gamutrf-fpv/TBS_CRSF/868/no_telemetry/868e6_56000000_30_5s_1692638102.ci16.zst with model tbs_crossfire\n"
     ]
    }
   ],
   "source": [
    "path = \"data/gamutrf/gamutrf-fpv\"\n",
    "\n",
    "auto_label_directory(path)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": []
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "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.9"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 4
}