diff --git a/notebooks/keypoint_detection_colab_example.ipynb b/notebooks/keypoint_detection_colab_example.ipynb new file mode 100644 index 0000000..38f11a6 --- /dev/null +++ b/notebooks/keypoint_detection_colab_example.ipynb @@ -0,0 +1,1146 @@ +{ + "nbformat": 4, + "nbformat_minor": 0, + "metadata": { + "colab": { + "provenance": [], + "gpuType": "T4", + "authorship_tag": "ABX9TyMsUkvWK0X/XnbTaQ4kw2A9", + "include_colab_link": true + }, + "kernelspec": { + "name": "python3", + "display_name": "Python 3" + }, + "language_info": { + "name": "python" + }, + "accelerator": "GPU" + }, + "cells": [ + { + "cell_type": "markdown", + "metadata": { + "id": "view-in-github", + "colab_type": "text" + }, + "source": [ + "\"Open" + ] + }, + { + "cell_type": "markdown", + "source": [ + "# Keypoint detector Example\n", + "This notebook serves as example for how to set up the kepyoint detector in a notebook. Make sure to change the runtime to GPU." + ], + "metadata": { + "id": "A0QhLuQz5fEn" + } + }, + { + "cell_type": "code", + "execution_count": 1, + "metadata": { + "colab": { + "base_uri": "https://localhost:8080/" + }, + "id": "DJxKh_TZuYCg", + "outputId": "ce0375c4-285f-4b6f-d0dd-c55194936ca8" + }, + "outputs": [ + { + "output_type": "stream", + "name": "stdout", + "text": [ + "fatal: destination path 'keypoint-detection' already exists and is not an empty directory.\n", + "Obtaining file:///content/keypoint-detection\n", + " Preparing metadata (setup.py) ... \u001b[?25l\u001b[?25hdone\n", + "Requirement already satisfied: torch>=0.10 in /usr/local/lib/python3.10/dist-packages (from keypoint-detection==1.0) (2.1.0+cu118)\n", + "Requirement already satisfied: torchvision>=0.11 in /usr/local/lib/python3.10/dist-packages (from keypoint-detection==1.0) (0.16.0+cu118)\n", + "Requirement already satisfied: pytorch-lightning<=1.9.4,>=1.5.10 in /usr/local/lib/python3.10/dist-packages (from keypoint-detection==1.0) (1.9.4)\n", + "Requirement already satisfied: torchmetrics>=0.7 in /usr/local/lib/python3.10/dist-packages (from keypoint-detection==1.0) (1.2.1)\n", + "Requirement already satisfied: wandb>=0.13.7 in /usr/local/lib/python3.10/dist-packages (from keypoint-detection==1.0) (0.16.1)\n", + "Requirement already satisfied: timm>=0.6.11 in /usr/local/lib/python3.10/dist-packages (from keypoint-detection==1.0) (0.9.12)\n", + "Requirement already satisfied: tqdm in /usr/local/lib/python3.10/dist-packages (from keypoint-detection==1.0) (4.66.1)\n", + "Requirement already satisfied: pytest in /usr/local/lib/python3.10/dist-packages (from keypoint-detection==1.0) (7.4.3)\n", + "Requirement already satisfied: pre-commit in /usr/local/lib/python3.10/dist-packages (from keypoint-detection==1.0) (3.5.0)\n", + "Requirement already satisfied: scikit-image in /usr/local/lib/python3.10/dist-packages (from keypoint-detection==1.0) (0.19.3)\n", + "Requirement already satisfied: albumentations in /usr/local/lib/python3.10/dist-packages (from keypoint-detection==1.0) (1.3.1)\n", + "Requirement already satisfied: matplotlib in /usr/local/lib/python3.10/dist-packages (from keypoint-detection==1.0) (3.7.1)\n", + "Requirement already satisfied: xmltodict in /usr/local/lib/python3.10/dist-packages (from keypoint-detection==1.0) (0.13.0)\n", + "Requirement already satisfied: pydantic in /usr/local/lib/python3.10/dist-packages (from keypoint-detection==1.0) (1.10.13)\n", + "Requirement already satisfied: fiftyone in /usr/local/lib/python3.10/dist-packages (from keypoint-detection==1.0) (0.23.0)\n", + "Requirement already satisfied: numpy>=1.17.2 in /usr/local/lib/python3.10/dist-packages (from pytorch-lightning<=1.9.4,>=1.5.10->keypoint-detection==1.0) (1.23.5)\n", + "Requirement already satisfied: PyYAML>=5.4 in /usr/local/lib/python3.10/dist-packages (from pytorch-lightning<=1.9.4,>=1.5.10->keypoint-detection==1.0) (6.0.1)\n", + "Requirement already satisfied: fsspec[http]>2021.06.0 in /usr/local/lib/python3.10/dist-packages (from pytorch-lightning<=1.9.4,>=1.5.10->keypoint-detection==1.0) (2023.6.0)\n", + "Requirement already satisfied: packaging>=17.1 in /usr/local/lib/python3.10/dist-packages (from pytorch-lightning<=1.9.4,>=1.5.10->keypoint-detection==1.0) (23.2)\n", + "Requirement already satisfied: typing-extensions>=4.0.0 in /usr/local/lib/python3.10/dist-packages (from pytorch-lightning<=1.9.4,>=1.5.10->keypoint-detection==1.0) (4.5.0)\n", + "Requirement already satisfied: lightning-utilities>=0.6.0.post0 in /usr/local/lib/python3.10/dist-packages (from pytorch-lightning<=1.9.4,>=1.5.10->keypoint-detection==1.0) (0.10.0)\n", + "Requirement already satisfied: huggingface-hub in /usr/local/lib/python3.10/dist-packages (from timm>=0.6.11->keypoint-detection==1.0) (0.19.4)\n", + "Requirement already satisfied: safetensors in /usr/local/lib/python3.10/dist-packages (from timm>=0.6.11->keypoint-detection==1.0) (0.4.1)\n", + "Requirement already satisfied: filelock in /usr/local/lib/python3.10/dist-packages (from torch>=0.10->keypoint-detection==1.0) (3.13.1)\n", + "Requirement already satisfied: sympy in /usr/local/lib/python3.10/dist-packages (from torch>=0.10->keypoint-detection==1.0) (1.12)\n", + "Requirement already satisfied: networkx in /usr/local/lib/python3.10/dist-packages (from torch>=0.10->keypoint-detection==1.0) (3.2.1)\n", + "Requirement already satisfied: jinja2 in /usr/local/lib/python3.10/dist-packages (from torch>=0.10->keypoint-detection==1.0) (3.1.2)\n", + "Requirement already satisfied: triton==2.1.0 in /usr/local/lib/python3.10/dist-packages (from torch>=0.10->keypoint-detection==1.0) (2.1.0)\n", + "Requirement already satisfied: requests in /usr/local/lib/python3.10/dist-packages (from torchvision>=0.11->keypoint-detection==1.0) (2.31.0)\n", + "Requirement already satisfied: pillow!=8.3.*,>=5.3.0 in /usr/local/lib/python3.10/dist-packages (from torchvision>=0.11->keypoint-detection==1.0) (9.4.0)\n", + "Requirement already satisfied: Click!=8.0.0,>=7.1 in /usr/local/lib/python3.10/dist-packages (from wandb>=0.13.7->keypoint-detection==1.0) (8.1.7)\n", + "Requirement already satisfied: GitPython!=3.1.29,>=1.0.0 in /usr/local/lib/python3.10/dist-packages (from wandb>=0.13.7->keypoint-detection==1.0) (3.1.40)\n", + "Requirement already satisfied: psutil>=5.0.0 in /usr/local/lib/python3.10/dist-packages (from wandb>=0.13.7->keypoint-detection==1.0) (5.9.5)\n", + "Requirement already satisfied: sentry-sdk>=1.0.0 in /usr/local/lib/python3.10/dist-packages (from wandb>=0.13.7->keypoint-detection==1.0) (1.38.0)\n", + "Requirement already satisfied: docker-pycreds>=0.4.0 in /usr/local/lib/python3.10/dist-packages (from wandb>=0.13.7->keypoint-detection==1.0) (0.4.0)\n", + "Requirement already satisfied: setproctitle in /usr/local/lib/python3.10/dist-packages (from wandb>=0.13.7->keypoint-detection==1.0) (1.3.3)\n", + "Requirement already satisfied: setuptools in /usr/local/lib/python3.10/dist-packages (from wandb>=0.13.7->keypoint-detection==1.0) (67.7.2)\n", + "Requirement already satisfied: appdirs>=1.4.3 in /usr/local/lib/python3.10/dist-packages (from wandb>=0.13.7->keypoint-detection==1.0) (1.4.4)\n", + "Requirement already satisfied: protobuf!=4.21.0,<5,>=3.19.0 in /usr/local/lib/python3.10/dist-packages (from wandb>=0.13.7->keypoint-detection==1.0) (3.20.3)\n", + "Requirement already satisfied: scipy>=1.1.0 in /usr/local/lib/python3.10/dist-packages (from albumentations->keypoint-detection==1.0) (1.11.4)\n", + "Requirement already satisfied: qudida>=0.0.4 in /usr/local/lib/python3.10/dist-packages (from albumentations->keypoint-detection==1.0) (0.0.4)\n", + "Requirement already satisfied: opencv-python-headless>=4.1.1 in /usr/local/lib/python3.10/dist-packages (from albumentations->keypoint-detection==1.0) (4.8.1.78)\n", + "Requirement already satisfied: imageio>=2.4.1 in /usr/local/lib/python3.10/dist-packages (from scikit-image->keypoint-detection==1.0) (2.31.6)\n", + "Requirement already satisfied: tifffile>=2019.7.26 in /usr/local/lib/python3.10/dist-packages (from scikit-image->keypoint-detection==1.0) (2023.9.26)\n", + "Requirement already satisfied: PyWavelets>=1.1.1 in /usr/local/lib/python3.10/dist-packages (from scikit-image->keypoint-detection==1.0) (1.5.0)\n", + "Requirement already satisfied: aiofiles in /usr/local/lib/python3.10/dist-packages (from fiftyone->keypoint-detection==1.0) (23.2.1)\n", + "Requirement already satisfied: argcomplete in /usr/local/lib/python3.10/dist-packages (from fiftyone->keypoint-detection==1.0) (3.1.6)\n", + "Requirement already satisfied: beautifulsoup4 in /usr/local/lib/python3.10/dist-packages (from fiftyone->keypoint-detection==1.0) (4.11.2)\n", + "Requirement already satisfied: boto3 in /usr/local/lib/python3.10/dist-packages (from fiftyone->keypoint-detection==1.0) (1.33.8)\n", + "Requirement already satisfied: cachetools in /usr/local/lib/python3.10/dist-packages (from fiftyone->keypoint-detection==1.0) (5.3.2)\n", + "Requirement already satisfied: dacite<1.8.0,>=1.6.0 in /usr/local/lib/python3.10/dist-packages (from fiftyone->keypoint-detection==1.0) (1.7.0)\n", + "Requirement already satisfied: Deprecated in /usr/local/lib/python3.10/dist-packages (from fiftyone->keypoint-detection==1.0) (1.2.14)\n", + "Requirement already satisfied: ftfy in /usr/local/lib/python3.10/dist-packages (from fiftyone->keypoint-detection==1.0) (6.1.3)\n", + "Requirement already satisfied: humanize in /usr/local/lib/python3.10/dist-packages (from fiftyone->keypoint-detection==1.0) (4.7.0)\n", + "Requirement already satisfied: hypercorn>=0.13.2 in /usr/local/lib/python3.10/dist-packages (from fiftyone->keypoint-detection==1.0) (0.15.0)\n", + "Requirement already satisfied: kaleido!=0.2.1.post1 in /usr/local/lib/python3.10/dist-packages (from fiftyone->keypoint-detection==1.0) (0.2.1)\n", + "Requirement already satisfied: mongoengine==0.24.2 in /usr/local/lib/python3.10/dist-packages (from fiftyone->keypoint-detection==1.0) (0.24.2)\n", + "Requirement already satisfied: motor>=2.5 in /usr/local/lib/python3.10/dist-packages (from fiftyone->keypoint-detection==1.0) (3.3.2)\n", + "Requirement already satisfied: pandas in /usr/local/lib/python3.10/dist-packages (from fiftyone->keypoint-detection==1.0) (1.5.3)\n", + "Requirement already satisfied: plotly>=4.14 in /usr/local/lib/python3.10/dist-packages (from fiftyone->keypoint-detection==1.0) (5.15.0)\n", + "Requirement already satisfied: pprintpp in /usr/local/lib/python3.10/dist-packages (from fiftyone->keypoint-detection==1.0) (0.4.0)\n", + "Requirement already satisfied: pymongo>=3.12 in /usr/local/lib/python3.10/dist-packages (from fiftyone->keypoint-detection==1.0) (4.6.1)\n", + "Requirement already satisfied: pytz in /usr/local/lib/python3.10/dist-packages (from fiftyone->keypoint-detection==1.0) (2023.3.post1)\n", + "Requirement already satisfied: regex in /usr/local/lib/python3.10/dist-packages (from fiftyone->keypoint-detection==1.0) (2023.6.3)\n", + "Requirement already satisfied: retrying in /usr/local/lib/python3.10/dist-packages (from fiftyone->keypoint-detection==1.0) (1.3.4)\n", + "Requirement already satisfied: scikit-learn in /usr/local/lib/python3.10/dist-packages (from fiftyone->keypoint-detection==1.0) (1.2.2)\n", + "Requirement already satisfied: sseclient-py<2,>=1.7.2 in /usr/local/lib/python3.10/dist-packages (from fiftyone->keypoint-detection==1.0) (1.8.0)\n", + "Requirement already satisfied: sse-starlette<1,>=0.10.3 in /usr/local/lib/python3.10/dist-packages (from fiftyone->keypoint-detection==1.0) (0.10.3)\n", + "Requirement already satisfied: starlette>=0.24.0 in /usr/local/lib/python3.10/dist-packages (from fiftyone->keypoint-detection==1.0) (0.33.0)\n", + "Requirement already satisfied: strawberry-graphql==0.138.1 in /usr/local/lib/python3.10/dist-packages (from fiftyone->keypoint-detection==1.0) (0.138.1)\n", + "Requirement already satisfied: tabulate in /usr/local/lib/python3.10/dist-packages (from fiftyone->keypoint-detection==1.0) (0.9.0)\n", + "Requirement already satisfied: universal-analytics-python3<2,>=1.0.1 in /usr/local/lib/python3.10/dist-packages (from fiftyone->keypoint-detection==1.0) (1.1.1)\n", + "Requirement already satisfied: fiftyone-brain<0.15,>=0.14 in /usr/local/lib/python3.10/dist-packages (from fiftyone->keypoint-detection==1.0) (0.14.0)\n", + "Requirement already satisfied: fiftyone-db<2.0,>=0.4 in /usr/local/lib/python3.10/dist-packages (from fiftyone->keypoint-detection==1.0) (1.0)\n", + "Requirement already satisfied: voxel51-eta<0.13,>=0.12 in /usr/local/lib/python3.10/dist-packages (from fiftyone->keypoint-detection==1.0) (0.12.0)\n", + "Requirement already satisfied: graphql-core<3.3.0,>=3.2.0 in /usr/local/lib/python3.10/dist-packages (from strawberry-graphql==0.138.1->fiftyone->keypoint-detection==1.0) (3.2.3)\n", + "Requirement already satisfied: python-dateutil<3.0.0,>=2.7.0 in /usr/local/lib/python3.10/dist-packages (from strawberry-graphql==0.138.1->fiftyone->keypoint-detection==1.0) (2.8.2)\n", + "Requirement already satisfied: contourpy>=1.0.1 in /usr/local/lib/python3.10/dist-packages (from matplotlib->keypoint-detection==1.0) (1.2.0)\n", + "Requirement already satisfied: cycler>=0.10 in /usr/local/lib/python3.10/dist-packages (from matplotlib->keypoint-detection==1.0) (0.12.1)\n", + "Requirement already satisfied: fonttools>=4.22.0 in /usr/local/lib/python3.10/dist-packages (from matplotlib->keypoint-detection==1.0) (4.45.1)\n", + "Requirement already satisfied: kiwisolver>=1.0.1 in /usr/local/lib/python3.10/dist-packages (from matplotlib->keypoint-detection==1.0) (1.4.5)\n", + "Requirement already satisfied: pyparsing>=2.3.1 in /usr/local/lib/python3.10/dist-packages (from matplotlib->keypoint-detection==1.0) (3.1.1)\n", + "Requirement already satisfied: cfgv>=2.0.0 in /usr/local/lib/python3.10/dist-packages (from pre-commit->keypoint-detection==1.0) (3.4.0)\n", + "Requirement already satisfied: identify>=1.0.0 in /usr/local/lib/python3.10/dist-packages (from pre-commit->keypoint-detection==1.0) (2.5.32)\n", + "Requirement already satisfied: nodeenv>=0.11.1 in /usr/local/lib/python3.10/dist-packages (from pre-commit->keypoint-detection==1.0) (1.8.0)\n", + "Requirement already satisfied: virtualenv>=20.10.0 in /usr/local/lib/python3.10/dist-packages (from pre-commit->keypoint-detection==1.0) (20.25.0)\n", + "Requirement already satisfied: iniconfig in /usr/local/lib/python3.10/dist-packages (from pytest->keypoint-detection==1.0) (2.0.0)\n", + "Requirement already satisfied: pluggy<2.0,>=0.12 in /usr/local/lib/python3.10/dist-packages (from pytest->keypoint-detection==1.0) (1.3.0)\n", + "Requirement already satisfied: exceptiongroup>=1.0.0rc8 in /usr/local/lib/python3.10/dist-packages (from pytest->keypoint-detection==1.0) (1.2.0)\n", + "Requirement already satisfied: tomli>=1.0.0 in /usr/local/lib/python3.10/dist-packages (from pytest->keypoint-detection==1.0) (2.0.1)\n", + "Requirement already satisfied: six>=1.4.0 in /usr/local/lib/python3.10/dist-packages (from docker-pycreds>=0.4.0->wandb>=0.13.7->keypoint-detection==1.0) (1.16.0)\n", + "Requirement already satisfied: aiohttp!=4.0.0a0,!=4.0.0a1 in /usr/local/lib/python3.10/dist-packages (from fsspec[http]>2021.06.0->pytorch-lightning<=1.9.4,>=1.5.10->keypoint-detection==1.0) (3.9.1)\n", + "Requirement already satisfied: gitdb<5,>=4.0.1 in /usr/local/lib/python3.10/dist-packages (from GitPython!=3.1.29,>=1.0.0->wandb>=0.13.7->keypoint-detection==1.0) (4.0.11)\n", + "Requirement already satisfied: h11 in /usr/local/lib/python3.10/dist-packages (from hypercorn>=0.13.2->fiftyone->keypoint-detection==1.0) (0.14.0)\n", + "Requirement already satisfied: h2>=3.1.0 in /usr/local/lib/python3.10/dist-packages (from hypercorn>=0.13.2->fiftyone->keypoint-detection==1.0) (4.1.0)\n", + "Requirement already satisfied: priority in /usr/local/lib/python3.10/dist-packages (from hypercorn>=0.13.2->fiftyone->keypoint-detection==1.0) (2.0.0)\n", + "Requirement already satisfied: taskgroup in /usr/local/lib/python3.10/dist-packages (from hypercorn>=0.13.2->fiftyone->keypoint-detection==1.0) (0.0.0a4)\n", + "Requirement already satisfied: wsproto>=0.14.0 in /usr/local/lib/python3.10/dist-packages (from hypercorn>=0.13.2->fiftyone->keypoint-detection==1.0) (1.2.0)\n", + "Requirement already satisfied: MarkupSafe>=2.0 in /usr/local/lib/python3.10/dist-packages (from jinja2->torch>=0.10->keypoint-detection==1.0) (2.1.3)\n", + "Requirement already satisfied: tenacity>=6.2.0 in /usr/local/lib/python3.10/dist-packages (from plotly>=4.14->fiftyone->keypoint-detection==1.0) (8.2.3)\n", + "Requirement already satisfied: dnspython<3.0.0,>=1.16.0 in /usr/local/lib/python3.10/dist-packages (from pymongo>=3.12->fiftyone->keypoint-detection==1.0) (2.4.2)\n", + "Requirement already satisfied: charset-normalizer<4,>=2 in /usr/local/lib/python3.10/dist-packages (from requests->torchvision>=0.11->keypoint-detection==1.0) (3.3.2)\n", + "Requirement already satisfied: idna<4,>=2.5 in /usr/local/lib/python3.10/dist-packages (from requests->torchvision>=0.11->keypoint-detection==1.0) (3.6)\n", + "Requirement already satisfied: urllib3<3,>=1.21.1 in /usr/local/lib/python3.10/dist-packages (from requests->torchvision>=0.11->keypoint-detection==1.0) (2.0.7)\n", + "Requirement already satisfied: certifi>=2017.4.17 in /usr/local/lib/python3.10/dist-packages (from requests->torchvision>=0.11->keypoint-detection==1.0) (2023.11.17)\n", + "Requirement already satisfied: joblib>=1.1.1 in /usr/local/lib/python3.10/dist-packages (from scikit-learn->fiftyone->keypoint-detection==1.0) (1.3.2)\n", + "Requirement already satisfied: threadpoolctl>=2.0.0 in /usr/local/lib/python3.10/dist-packages (from scikit-learn->fiftyone->keypoint-detection==1.0) (3.2.0)\n", + "Requirement already satisfied: anyio<5,>=3.4.0 in /usr/local/lib/python3.10/dist-packages (from starlette>=0.24.0->fiftyone->keypoint-detection==1.0) (3.7.1)\n", + "Requirement already satisfied: httpx>=0.10.0 in /usr/local/lib/python3.10/dist-packages (from universal-analytics-python3<2,>=1.0.1->fiftyone->keypoint-detection==1.0) (0.25.2)\n", + "Requirement already satisfied: distlib<1,>=0.3.7 in /usr/local/lib/python3.10/dist-packages (from virtualenv>=20.10.0->pre-commit->keypoint-detection==1.0) (0.3.7)\n", + "Requirement already satisfied: platformdirs<5,>=3.9.1 in /usr/local/lib/python3.10/dist-packages (from virtualenv>=20.10.0->pre-commit->keypoint-detection==1.0) (4.0.0)\n", + "Requirement already satisfied: dill in /usr/local/lib/python3.10/dist-packages (from voxel51-eta<0.13,>=0.12->fiftyone->keypoint-detection==1.0) (0.3.7)\n", + "Requirement already satisfied: future in /usr/local/lib/python3.10/dist-packages (from voxel51-eta<0.13,>=0.12->fiftyone->keypoint-detection==1.0) (0.18.3)\n", + "Requirement already satisfied: glob2 in /usr/local/lib/python3.10/dist-packages (from voxel51-eta<0.13,>=0.12->fiftyone->keypoint-detection==1.0) (0.7)\n", + "Requirement already satisfied: jsonlines in /usr/local/lib/python3.10/dist-packages (from voxel51-eta<0.13,>=0.12->fiftyone->keypoint-detection==1.0) (4.0.0)\n", + "Requirement already satisfied: py7zr in /usr/local/lib/python3.10/dist-packages (from voxel51-eta<0.13,>=0.12->fiftyone->keypoint-detection==1.0) (0.20.8)\n", + "Requirement already satisfied: rarfile in /usr/local/lib/python3.10/dist-packages (from voxel51-eta<0.13,>=0.12->fiftyone->keypoint-detection==1.0) (4.1)\n", + "Requirement already satisfied: sortedcontainers in /usr/local/lib/python3.10/dist-packages (from voxel51-eta<0.13,>=0.12->fiftyone->keypoint-detection==1.0) (2.4.0)\n", + "Requirement already satisfied: tzlocal in /usr/local/lib/python3.10/dist-packages (from voxel51-eta<0.13,>=0.12->fiftyone->keypoint-detection==1.0) (5.2)\n", + "Requirement already satisfied: soupsieve>1.2 in /usr/local/lib/python3.10/dist-packages (from beautifulsoup4->fiftyone->keypoint-detection==1.0) (2.5)\n", + "Requirement already satisfied: botocore<1.34.0,>=1.33.8 in /usr/local/lib/python3.10/dist-packages (from boto3->fiftyone->keypoint-detection==1.0) (1.33.8)\n", + "Requirement already satisfied: jmespath<2.0.0,>=0.7.1 in /usr/local/lib/python3.10/dist-packages (from boto3->fiftyone->keypoint-detection==1.0) (1.0.1)\n", + "Requirement already satisfied: s3transfer<0.9.0,>=0.8.2 in /usr/local/lib/python3.10/dist-packages (from boto3->fiftyone->keypoint-detection==1.0) (0.8.2)\n", + "Requirement already satisfied: wrapt<2,>=1.10 in /usr/local/lib/python3.10/dist-packages (from Deprecated->fiftyone->keypoint-detection==1.0) (1.14.1)\n", + "Requirement already satisfied: wcwidth<0.3.0,>=0.2.12 in /usr/local/lib/python3.10/dist-packages (from ftfy->fiftyone->keypoint-detection==1.0) (0.2.12)\n", + "Requirement already satisfied: mpmath>=0.19 in /usr/local/lib/python3.10/dist-packages (from sympy->torch>=0.10->keypoint-detection==1.0) (1.3.0)\n", + "Requirement already satisfied: attrs>=17.3.0 in /usr/local/lib/python3.10/dist-packages (from aiohttp!=4.0.0a0,!=4.0.0a1->fsspec[http]>2021.06.0->pytorch-lightning<=1.9.4,>=1.5.10->keypoint-detection==1.0) (23.1.0)\n", + "Requirement already satisfied: multidict<7.0,>=4.5 in /usr/local/lib/python3.10/dist-packages (from aiohttp!=4.0.0a0,!=4.0.0a1->fsspec[http]>2021.06.0->pytorch-lightning<=1.9.4,>=1.5.10->keypoint-detection==1.0) (6.0.4)\n", + "Requirement already satisfied: yarl<2.0,>=1.0 in /usr/local/lib/python3.10/dist-packages (from aiohttp!=4.0.0a0,!=4.0.0a1->fsspec[http]>2021.06.0->pytorch-lightning<=1.9.4,>=1.5.10->keypoint-detection==1.0) (1.9.3)\n", + "Requirement already satisfied: frozenlist>=1.1.1 in /usr/local/lib/python3.10/dist-packages (from aiohttp!=4.0.0a0,!=4.0.0a1->fsspec[http]>2021.06.0->pytorch-lightning<=1.9.4,>=1.5.10->keypoint-detection==1.0) (1.4.0)\n", + "Requirement already satisfied: aiosignal>=1.1.2 in /usr/local/lib/python3.10/dist-packages (from aiohttp!=4.0.0a0,!=4.0.0a1->fsspec[http]>2021.06.0->pytorch-lightning<=1.9.4,>=1.5.10->keypoint-detection==1.0) (1.3.1)\n", + "Requirement already satisfied: async-timeout<5.0,>=4.0 in /usr/local/lib/python3.10/dist-packages (from aiohttp!=4.0.0a0,!=4.0.0a1->fsspec[http]>2021.06.0->pytorch-lightning<=1.9.4,>=1.5.10->keypoint-detection==1.0) (4.0.3)\n", + "Requirement already satisfied: sniffio>=1.1 in /usr/local/lib/python3.10/dist-packages (from anyio<5,>=3.4.0->starlette>=0.24.0->fiftyone->keypoint-detection==1.0) (1.3.0)\n", + "Requirement already satisfied: smmap<6,>=3.0.1 in /usr/local/lib/python3.10/dist-packages (from gitdb<5,>=4.0.1->GitPython!=3.1.29,>=1.0.0->wandb>=0.13.7->keypoint-detection==1.0) (5.0.1)\n", + "Requirement already satisfied: hyperframe<7,>=6.0 in /usr/local/lib/python3.10/dist-packages (from h2>=3.1.0->hypercorn>=0.13.2->fiftyone->keypoint-detection==1.0) (6.0.1)\n", + "Requirement already satisfied: hpack<5,>=4.0 in /usr/local/lib/python3.10/dist-packages (from h2>=3.1.0->hypercorn>=0.13.2->fiftyone->keypoint-detection==1.0) (4.0.0)\n", + "Requirement already satisfied: httpcore==1.* in /usr/local/lib/python3.10/dist-packages (from httpx>=0.10.0->universal-analytics-python3<2,>=1.0.1->fiftyone->keypoint-detection==1.0) (1.0.2)\n", + "Requirement already satisfied: texttable in /usr/local/lib/python3.10/dist-packages (from py7zr->voxel51-eta<0.13,>=0.12->fiftyone->keypoint-detection==1.0) (1.7.0)\n", + "Requirement already satisfied: pycryptodomex>=3.16.0 in /usr/local/lib/python3.10/dist-packages (from py7zr->voxel51-eta<0.13,>=0.12->fiftyone->keypoint-detection==1.0) (3.19.0)\n", + "Requirement already satisfied: pyzstd>=0.15.9 in /usr/local/lib/python3.10/dist-packages (from py7zr->voxel51-eta<0.13,>=0.12->fiftyone->keypoint-detection==1.0) (0.15.9)\n", + "Requirement already satisfied: pyppmd<1.2.0,>=1.1.0 in /usr/local/lib/python3.10/dist-packages (from py7zr->voxel51-eta<0.13,>=0.12->fiftyone->keypoint-detection==1.0) (1.1.0)\n", + "Requirement already satisfied: pybcj<1.1.0,>=1.0.0 in /usr/local/lib/python3.10/dist-packages (from py7zr->voxel51-eta<0.13,>=0.12->fiftyone->keypoint-detection==1.0) (1.0.2)\n", + "Requirement already satisfied: multivolumefile>=0.2.3 in /usr/local/lib/python3.10/dist-packages (from py7zr->voxel51-eta<0.13,>=0.12->fiftyone->keypoint-detection==1.0) (0.2.3)\n", + "Requirement already satisfied: inflate64<1.1.0,>=1.0.0 in /usr/local/lib/python3.10/dist-packages (from py7zr->voxel51-eta<0.13,>=0.12->fiftyone->keypoint-detection==1.0) (1.0.0)\n", + "Requirement already satisfied: brotli>=1.1.0 in /usr/local/lib/python3.10/dist-packages (from py7zr->voxel51-eta<0.13,>=0.12->fiftyone->keypoint-detection==1.0) (1.1.0)\n", + "Installing collected packages: keypoint-detection\n", + " Attempting uninstall: keypoint-detection\n", + " Found existing installation: keypoint-detection 1.0\n", + " Uninstalling keypoint-detection-1.0:\n", + " Successfully uninstalled keypoint-detection-1.0\n", + " Running setup.py develop for keypoint-detection\n", + "Successfully installed keypoint-detection-1.0\n" + ] + } + ], + "source": [ + "!git clone https://github.com/tlpss/keypoint-detection.git\n", + "!pip install -e keypoint-detection" + ] + }, + { + "cell_type": "code", + "source": [ + "# install kp detector. Need to navigate to folder first to avoid strange 'could not import' error\n", + "!cd keypoint-detection && pip install ." + ], + "metadata": { + "colab": { + "base_uri": "https://localhost:8080/" + }, + "id": "eVPfwqpz1AkS", + "outputId": "9ee517fb-4201-4a16-917b-2481c148e326" + }, + "execution_count": 2, + "outputs": [ + { + "output_type": "stream", + "name": "stdout", + "text": [ + "Processing /content/keypoint-detection\n", + " Preparing metadata (setup.py) ... \u001b[?25l\u001b[?25hdone\n", + "Requirement already satisfied: torch>=0.10 in /usr/local/lib/python3.10/dist-packages (from keypoint-detection==1.0) (2.1.0+cu118)\n", + "Requirement already satisfied: torchvision>=0.11 in /usr/local/lib/python3.10/dist-packages (from keypoint-detection==1.0) (0.16.0+cu118)\n", + "Requirement already satisfied: pytorch-lightning<=1.9.4,>=1.5.10 in /usr/local/lib/python3.10/dist-packages (from keypoint-detection==1.0) (1.9.4)\n", + "Requirement already satisfied: torchmetrics>=0.7 in /usr/local/lib/python3.10/dist-packages (from keypoint-detection==1.0) (1.2.1)\n", + "Requirement already satisfied: wandb>=0.13.7 in /usr/local/lib/python3.10/dist-packages (from keypoint-detection==1.0) (0.16.1)\n", + "Requirement already satisfied: timm>=0.6.11 in /usr/local/lib/python3.10/dist-packages (from keypoint-detection==1.0) (0.9.12)\n", + "Requirement already satisfied: tqdm in /usr/local/lib/python3.10/dist-packages (from keypoint-detection==1.0) (4.66.1)\n", + "Requirement already satisfied: pytest in /usr/local/lib/python3.10/dist-packages (from keypoint-detection==1.0) (7.4.3)\n", + "Requirement already satisfied: pre-commit in /usr/local/lib/python3.10/dist-packages (from keypoint-detection==1.0) (3.5.0)\n", + "Requirement already satisfied: scikit-image in /usr/local/lib/python3.10/dist-packages (from keypoint-detection==1.0) (0.19.3)\n", + "Requirement already satisfied: albumentations in /usr/local/lib/python3.10/dist-packages (from keypoint-detection==1.0) (1.3.1)\n", + "Requirement already satisfied: matplotlib in /usr/local/lib/python3.10/dist-packages (from keypoint-detection==1.0) (3.7.1)\n", + "Requirement already satisfied: xmltodict in /usr/local/lib/python3.10/dist-packages (from keypoint-detection==1.0) (0.13.0)\n", + "Requirement already satisfied: pydantic in /usr/local/lib/python3.10/dist-packages (from keypoint-detection==1.0) (1.10.13)\n", + "Requirement already satisfied: fiftyone in /usr/local/lib/python3.10/dist-packages (from keypoint-detection==1.0) (0.23.0)\n", + "Requirement already satisfied: numpy>=1.17.2 in /usr/local/lib/python3.10/dist-packages (from pytorch-lightning<=1.9.4,>=1.5.10->keypoint-detection==1.0) (1.23.5)\n", + "Requirement already satisfied: PyYAML>=5.4 in /usr/local/lib/python3.10/dist-packages (from pytorch-lightning<=1.9.4,>=1.5.10->keypoint-detection==1.0) (6.0.1)\n", + "Requirement already satisfied: fsspec[http]>2021.06.0 in /usr/local/lib/python3.10/dist-packages (from pytorch-lightning<=1.9.4,>=1.5.10->keypoint-detection==1.0) (2023.6.0)\n", + "Requirement already satisfied: packaging>=17.1 in /usr/local/lib/python3.10/dist-packages (from pytorch-lightning<=1.9.4,>=1.5.10->keypoint-detection==1.0) (23.2)\n", + "Requirement already satisfied: typing-extensions>=4.0.0 in /usr/local/lib/python3.10/dist-packages (from pytorch-lightning<=1.9.4,>=1.5.10->keypoint-detection==1.0) (4.5.0)\n", + "Requirement already satisfied: lightning-utilities>=0.6.0.post0 in /usr/local/lib/python3.10/dist-packages (from pytorch-lightning<=1.9.4,>=1.5.10->keypoint-detection==1.0) (0.10.0)\n", + "Requirement already satisfied: huggingface-hub in /usr/local/lib/python3.10/dist-packages (from timm>=0.6.11->keypoint-detection==1.0) (0.19.4)\n", + "Requirement already satisfied: safetensors in /usr/local/lib/python3.10/dist-packages (from timm>=0.6.11->keypoint-detection==1.0) (0.4.1)\n", + "Requirement already satisfied: filelock in /usr/local/lib/python3.10/dist-packages (from torch>=0.10->keypoint-detection==1.0) (3.13.1)\n", + "Requirement already satisfied: sympy in /usr/local/lib/python3.10/dist-packages (from torch>=0.10->keypoint-detection==1.0) (1.12)\n", + "Requirement already satisfied: networkx in /usr/local/lib/python3.10/dist-packages (from torch>=0.10->keypoint-detection==1.0) (3.2.1)\n", + "Requirement already satisfied: jinja2 in /usr/local/lib/python3.10/dist-packages (from torch>=0.10->keypoint-detection==1.0) (3.1.2)\n", + "Requirement already satisfied: triton==2.1.0 in /usr/local/lib/python3.10/dist-packages (from torch>=0.10->keypoint-detection==1.0) (2.1.0)\n", + "Requirement already satisfied: requests in /usr/local/lib/python3.10/dist-packages (from torchvision>=0.11->keypoint-detection==1.0) (2.31.0)\n", + "Requirement already satisfied: pillow!=8.3.*,>=5.3.0 in /usr/local/lib/python3.10/dist-packages (from torchvision>=0.11->keypoint-detection==1.0) (9.4.0)\n", + "Requirement already satisfied: Click!=8.0.0,>=7.1 in /usr/local/lib/python3.10/dist-packages (from wandb>=0.13.7->keypoint-detection==1.0) (8.1.7)\n", + "Requirement already satisfied: GitPython!=3.1.29,>=1.0.0 in /usr/local/lib/python3.10/dist-packages (from wandb>=0.13.7->keypoint-detection==1.0) (3.1.40)\n", + "Requirement already satisfied: psutil>=5.0.0 in /usr/local/lib/python3.10/dist-packages (from wandb>=0.13.7->keypoint-detection==1.0) (5.9.5)\n", + "Requirement already satisfied: sentry-sdk>=1.0.0 in /usr/local/lib/python3.10/dist-packages (from wandb>=0.13.7->keypoint-detection==1.0) (1.38.0)\n", + "Requirement already satisfied: docker-pycreds>=0.4.0 in /usr/local/lib/python3.10/dist-packages (from wandb>=0.13.7->keypoint-detection==1.0) (0.4.0)\n", + "Requirement already satisfied: setproctitle in /usr/local/lib/python3.10/dist-packages (from wandb>=0.13.7->keypoint-detection==1.0) (1.3.3)\n", + "Requirement already satisfied: setuptools in /usr/local/lib/python3.10/dist-packages (from wandb>=0.13.7->keypoint-detection==1.0) (67.7.2)\n", + "Requirement already satisfied: appdirs>=1.4.3 in /usr/local/lib/python3.10/dist-packages (from wandb>=0.13.7->keypoint-detection==1.0) (1.4.4)\n", + "Requirement already satisfied: protobuf!=4.21.0,<5,>=3.19.0 in /usr/local/lib/python3.10/dist-packages (from wandb>=0.13.7->keypoint-detection==1.0) (3.20.3)\n", + "Requirement already satisfied: scipy>=1.1.0 in /usr/local/lib/python3.10/dist-packages (from albumentations->keypoint-detection==1.0) (1.11.4)\n", + "Requirement already satisfied: qudida>=0.0.4 in /usr/local/lib/python3.10/dist-packages (from albumentations->keypoint-detection==1.0) (0.0.4)\n", + "Requirement already satisfied: opencv-python-headless>=4.1.1 in /usr/local/lib/python3.10/dist-packages (from albumentations->keypoint-detection==1.0) (4.8.1.78)\n", + "Requirement already satisfied: imageio>=2.4.1 in /usr/local/lib/python3.10/dist-packages (from scikit-image->keypoint-detection==1.0) (2.31.6)\n", + "Requirement already satisfied: tifffile>=2019.7.26 in /usr/local/lib/python3.10/dist-packages (from scikit-image->keypoint-detection==1.0) (2023.9.26)\n", + "Requirement already satisfied: PyWavelets>=1.1.1 in /usr/local/lib/python3.10/dist-packages (from scikit-image->keypoint-detection==1.0) (1.5.0)\n", + "Requirement already satisfied: aiofiles in /usr/local/lib/python3.10/dist-packages (from fiftyone->keypoint-detection==1.0) (23.2.1)\n", + "Requirement already satisfied: argcomplete in /usr/local/lib/python3.10/dist-packages (from fiftyone->keypoint-detection==1.0) (3.1.6)\n", + "Requirement already satisfied: beautifulsoup4 in /usr/local/lib/python3.10/dist-packages (from fiftyone->keypoint-detection==1.0) (4.11.2)\n", + "Requirement already satisfied: boto3 in /usr/local/lib/python3.10/dist-packages (from fiftyone->keypoint-detection==1.0) (1.33.8)\n", + "Requirement already satisfied: cachetools in /usr/local/lib/python3.10/dist-packages (from fiftyone->keypoint-detection==1.0) (5.3.2)\n", + "Requirement already satisfied: dacite<1.8.0,>=1.6.0 in /usr/local/lib/python3.10/dist-packages (from fiftyone->keypoint-detection==1.0) (1.7.0)\n", + "Requirement already satisfied: Deprecated in /usr/local/lib/python3.10/dist-packages (from fiftyone->keypoint-detection==1.0) (1.2.14)\n", + "Requirement already satisfied: ftfy in /usr/local/lib/python3.10/dist-packages (from fiftyone->keypoint-detection==1.0) (6.1.3)\n", + "Requirement already satisfied: humanize in /usr/local/lib/python3.10/dist-packages (from fiftyone->keypoint-detection==1.0) (4.7.0)\n", + "Requirement already satisfied: hypercorn>=0.13.2 in /usr/local/lib/python3.10/dist-packages (from fiftyone->keypoint-detection==1.0) (0.15.0)\n", + "Requirement already satisfied: kaleido!=0.2.1.post1 in /usr/local/lib/python3.10/dist-packages (from fiftyone->keypoint-detection==1.0) (0.2.1)\n", + "Requirement already satisfied: mongoengine==0.24.2 in /usr/local/lib/python3.10/dist-packages (from fiftyone->keypoint-detection==1.0) (0.24.2)\n", + "Requirement already satisfied: motor>=2.5 in /usr/local/lib/python3.10/dist-packages (from fiftyone->keypoint-detection==1.0) (3.3.2)\n", + "Requirement already satisfied: pandas in /usr/local/lib/python3.10/dist-packages (from fiftyone->keypoint-detection==1.0) (1.5.3)\n", + "Requirement already satisfied: plotly>=4.14 in /usr/local/lib/python3.10/dist-packages (from fiftyone->keypoint-detection==1.0) (5.15.0)\n", + "Requirement already satisfied: pprintpp in /usr/local/lib/python3.10/dist-packages (from fiftyone->keypoint-detection==1.0) (0.4.0)\n", + "Requirement already satisfied: pymongo>=3.12 in /usr/local/lib/python3.10/dist-packages (from fiftyone->keypoint-detection==1.0) (4.6.1)\n", + "Requirement already satisfied: pytz in /usr/local/lib/python3.10/dist-packages (from fiftyone->keypoint-detection==1.0) (2023.3.post1)\n", + "Requirement already satisfied: regex in /usr/local/lib/python3.10/dist-packages (from fiftyone->keypoint-detection==1.0) (2023.6.3)\n", + "Requirement already satisfied: retrying in /usr/local/lib/python3.10/dist-packages (from fiftyone->keypoint-detection==1.0) (1.3.4)\n", + "Requirement already satisfied: scikit-learn in /usr/local/lib/python3.10/dist-packages (from fiftyone->keypoint-detection==1.0) (1.2.2)\n", + "Requirement already satisfied: sseclient-py<2,>=1.7.2 in /usr/local/lib/python3.10/dist-packages (from fiftyone->keypoint-detection==1.0) (1.8.0)\n", + "Requirement already satisfied: sse-starlette<1,>=0.10.3 in /usr/local/lib/python3.10/dist-packages (from fiftyone->keypoint-detection==1.0) (0.10.3)\n", + "Requirement already satisfied: starlette>=0.24.0 in /usr/local/lib/python3.10/dist-packages (from fiftyone->keypoint-detection==1.0) (0.33.0)\n", + "Requirement already satisfied: strawberry-graphql==0.138.1 in /usr/local/lib/python3.10/dist-packages (from fiftyone->keypoint-detection==1.0) (0.138.1)\n", + "Requirement already satisfied: tabulate in /usr/local/lib/python3.10/dist-packages (from fiftyone->keypoint-detection==1.0) (0.9.0)\n", + "Requirement already satisfied: universal-analytics-python3<2,>=1.0.1 in /usr/local/lib/python3.10/dist-packages (from fiftyone->keypoint-detection==1.0) (1.1.1)\n", + "Requirement already satisfied: fiftyone-brain<0.15,>=0.14 in /usr/local/lib/python3.10/dist-packages (from fiftyone->keypoint-detection==1.0) (0.14.0)\n", + "Requirement already satisfied: fiftyone-db<2.0,>=0.4 in /usr/local/lib/python3.10/dist-packages (from fiftyone->keypoint-detection==1.0) (1.0)\n", + "Requirement already satisfied: voxel51-eta<0.13,>=0.12 in /usr/local/lib/python3.10/dist-packages (from fiftyone->keypoint-detection==1.0) (0.12.0)\n", + "Requirement already satisfied: graphql-core<3.3.0,>=3.2.0 in /usr/local/lib/python3.10/dist-packages (from strawberry-graphql==0.138.1->fiftyone->keypoint-detection==1.0) (3.2.3)\n", + "Requirement already satisfied: python-dateutil<3.0.0,>=2.7.0 in /usr/local/lib/python3.10/dist-packages (from strawberry-graphql==0.138.1->fiftyone->keypoint-detection==1.0) (2.8.2)\n", + "Requirement already satisfied: contourpy>=1.0.1 in /usr/local/lib/python3.10/dist-packages (from matplotlib->keypoint-detection==1.0) (1.2.0)\n", + "Requirement already satisfied: cycler>=0.10 in /usr/local/lib/python3.10/dist-packages (from matplotlib->keypoint-detection==1.0) (0.12.1)\n", + "Requirement already satisfied: fonttools>=4.22.0 in /usr/local/lib/python3.10/dist-packages (from matplotlib->keypoint-detection==1.0) (4.45.1)\n", + "Requirement already satisfied: kiwisolver>=1.0.1 in /usr/local/lib/python3.10/dist-packages (from matplotlib->keypoint-detection==1.0) (1.4.5)\n", + "Requirement already satisfied: pyparsing>=2.3.1 in /usr/local/lib/python3.10/dist-packages (from matplotlib->keypoint-detection==1.0) (3.1.1)\n", + "Requirement already satisfied: cfgv>=2.0.0 in /usr/local/lib/python3.10/dist-packages (from pre-commit->keypoint-detection==1.0) (3.4.0)\n", + "Requirement already satisfied: identify>=1.0.0 in /usr/local/lib/python3.10/dist-packages (from pre-commit->keypoint-detection==1.0) (2.5.32)\n", + "Requirement already satisfied: nodeenv>=0.11.1 in /usr/local/lib/python3.10/dist-packages (from pre-commit->keypoint-detection==1.0) (1.8.0)\n", + "Requirement already satisfied: virtualenv>=20.10.0 in /usr/local/lib/python3.10/dist-packages (from pre-commit->keypoint-detection==1.0) (20.25.0)\n", + "Requirement already satisfied: iniconfig in /usr/local/lib/python3.10/dist-packages (from pytest->keypoint-detection==1.0) (2.0.0)\n", + "Requirement already satisfied: pluggy<2.0,>=0.12 in /usr/local/lib/python3.10/dist-packages (from pytest->keypoint-detection==1.0) (1.3.0)\n", + "Requirement already satisfied: exceptiongroup>=1.0.0rc8 in /usr/local/lib/python3.10/dist-packages (from pytest->keypoint-detection==1.0) (1.2.0)\n", + "Requirement already satisfied: tomli>=1.0.0 in /usr/local/lib/python3.10/dist-packages (from pytest->keypoint-detection==1.0) (2.0.1)\n", + "Requirement already satisfied: six>=1.4.0 in /usr/local/lib/python3.10/dist-packages (from docker-pycreds>=0.4.0->wandb>=0.13.7->keypoint-detection==1.0) (1.16.0)\n", + "Requirement already satisfied: aiohttp!=4.0.0a0,!=4.0.0a1 in /usr/local/lib/python3.10/dist-packages (from fsspec[http]>2021.06.0->pytorch-lightning<=1.9.4,>=1.5.10->keypoint-detection==1.0) (3.9.1)\n", + "Requirement already satisfied: gitdb<5,>=4.0.1 in /usr/local/lib/python3.10/dist-packages (from GitPython!=3.1.29,>=1.0.0->wandb>=0.13.7->keypoint-detection==1.0) (4.0.11)\n", + "Requirement already satisfied: h11 in /usr/local/lib/python3.10/dist-packages (from hypercorn>=0.13.2->fiftyone->keypoint-detection==1.0) (0.14.0)\n", + "Requirement already satisfied: h2>=3.1.0 in /usr/local/lib/python3.10/dist-packages (from hypercorn>=0.13.2->fiftyone->keypoint-detection==1.0) (4.1.0)\n", + "Requirement already satisfied: priority in /usr/local/lib/python3.10/dist-packages (from hypercorn>=0.13.2->fiftyone->keypoint-detection==1.0) (2.0.0)\n", + "Requirement already satisfied: taskgroup in /usr/local/lib/python3.10/dist-packages (from hypercorn>=0.13.2->fiftyone->keypoint-detection==1.0) (0.0.0a4)\n", + "Requirement already satisfied: wsproto>=0.14.0 in /usr/local/lib/python3.10/dist-packages (from hypercorn>=0.13.2->fiftyone->keypoint-detection==1.0) (1.2.0)\n", + "Requirement already satisfied: MarkupSafe>=2.0 in /usr/local/lib/python3.10/dist-packages (from jinja2->torch>=0.10->keypoint-detection==1.0) (2.1.3)\n", + "Requirement already satisfied: tenacity>=6.2.0 in /usr/local/lib/python3.10/dist-packages (from plotly>=4.14->fiftyone->keypoint-detection==1.0) (8.2.3)\n", + "Requirement already satisfied: dnspython<3.0.0,>=1.16.0 in /usr/local/lib/python3.10/dist-packages (from pymongo>=3.12->fiftyone->keypoint-detection==1.0) (2.4.2)\n", + "Requirement already satisfied: charset-normalizer<4,>=2 in /usr/local/lib/python3.10/dist-packages (from requests->torchvision>=0.11->keypoint-detection==1.0) (3.3.2)\n", + "Requirement already satisfied: idna<4,>=2.5 in /usr/local/lib/python3.10/dist-packages (from requests->torchvision>=0.11->keypoint-detection==1.0) (3.6)\n", + "Requirement already satisfied: urllib3<3,>=1.21.1 in /usr/local/lib/python3.10/dist-packages (from requests->torchvision>=0.11->keypoint-detection==1.0) (2.0.7)\n", + "Requirement already satisfied: certifi>=2017.4.17 in /usr/local/lib/python3.10/dist-packages (from requests->torchvision>=0.11->keypoint-detection==1.0) (2023.11.17)\n", + "Requirement already satisfied: joblib>=1.1.1 in /usr/local/lib/python3.10/dist-packages (from scikit-learn->fiftyone->keypoint-detection==1.0) (1.3.2)\n", + "Requirement already satisfied: threadpoolctl>=2.0.0 in /usr/local/lib/python3.10/dist-packages (from scikit-learn->fiftyone->keypoint-detection==1.0) (3.2.0)\n", + "Requirement already satisfied: anyio<5,>=3.4.0 in /usr/local/lib/python3.10/dist-packages (from starlette>=0.24.0->fiftyone->keypoint-detection==1.0) (3.7.1)\n", + "Requirement already satisfied: httpx>=0.10.0 in /usr/local/lib/python3.10/dist-packages (from universal-analytics-python3<2,>=1.0.1->fiftyone->keypoint-detection==1.0) (0.25.2)\n", + "Requirement already satisfied: distlib<1,>=0.3.7 in /usr/local/lib/python3.10/dist-packages (from virtualenv>=20.10.0->pre-commit->keypoint-detection==1.0) (0.3.7)\n", + "Requirement already satisfied: platformdirs<5,>=3.9.1 in /usr/local/lib/python3.10/dist-packages (from virtualenv>=20.10.0->pre-commit->keypoint-detection==1.0) (4.0.0)\n", + "Requirement already satisfied: dill in /usr/local/lib/python3.10/dist-packages (from voxel51-eta<0.13,>=0.12->fiftyone->keypoint-detection==1.0) (0.3.7)\n", + "Requirement already satisfied: future in /usr/local/lib/python3.10/dist-packages (from voxel51-eta<0.13,>=0.12->fiftyone->keypoint-detection==1.0) (0.18.3)\n", + "Requirement already satisfied: glob2 in /usr/local/lib/python3.10/dist-packages (from voxel51-eta<0.13,>=0.12->fiftyone->keypoint-detection==1.0) (0.7)\n", + "Requirement already satisfied: jsonlines in /usr/local/lib/python3.10/dist-packages (from voxel51-eta<0.13,>=0.12->fiftyone->keypoint-detection==1.0) (4.0.0)\n", + "Requirement already satisfied: py7zr in /usr/local/lib/python3.10/dist-packages (from voxel51-eta<0.13,>=0.12->fiftyone->keypoint-detection==1.0) (0.20.8)\n", + "Requirement already satisfied: rarfile in /usr/local/lib/python3.10/dist-packages (from voxel51-eta<0.13,>=0.12->fiftyone->keypoint-detection==1.0) (4.1)\n", + "Requirement already satisfied: sortedcontainers in /usr/local/lib/python3.10/dist-packages (from voxel51-eta<0.13,>=0.12->fiftyone->keypoint-detection==1.0) (2.4.0)\n", + "Requirement already satisfied: tzlocal in /usr/local/lib/python3.10/dist-packages (from voxel51-eta<0.13,>=0.12->fiftyone->keypoint-detection==1.0) (5.2)\n", + "Requirement already satisfied: soupsieve>1.2 in /usr/local/lib/python3.10/dist-packages (from beautifulsoup4->fiftyone->keypoint-detection==1.0) (2.5)\n", + "Requirement already satisfied: botocore<1.34.0,>=1.33.8 in /usr/local/lib/python3.10/dist-packages (from boto3->fiftyone->keypoint-detection==1.0) (1.33.8)\n", + "Requirement already satisfied: jmespath<2.0.0,>=0.7.1 in /usr/local/lib/python3.10/dist-packages (from boto3->fiftyone->keypoint-detection==1.0) (1.0.1)\n", + "Requirement already satisfied: s3transfer<0.9.0,>=0.8.2 in /usr/local/lib/python3.10/dist-packages (from boto3->fiftyone->keypoint-detection==1.0) (0.8.2)\n", + "Requirement already satisfied: wrapt<2,>=1.10 in /usr/local/lib/python3.10/dist-packages (from Deprecated->fiftyone->keypoint-detection==1.0) (1.14.1)\n", + "Requirement already satisfied: wcwidth<0.3.0,>=0.2.12 in /usr/local/lib/python3.10/dist-packages (from ftfy->fiftyone->keypoint-detection==1.0) (0.2.12)\n", + "Requirement already satisfied: mpmath>=0.19 in /usr/local/lib/python3.10/dist-packages (from sympy->torch>=0.10->keypoint-detection==1.0) (1.3.0)\n", + "Requirement already satisfied: attrs>=17.3.0 in /usr/local/lib/python3.10/dist-packages (from aiohttp!=4.0.0a0,!=4.0.0a1->fsspec[http]>2021.06.0->pytorch-lightning<=1.9.4,>=1.5.10->keypoint-detection==1.0) (23.1.0)\n", + "Requirement already satisfied: multidict<7.0,>=4.5 in /usr/local/lib/python3.10/dist-packages (from aiohttp!=4.0.0a0,!=4.0.0a1->fsspec[http]>2021.06.0->pytorch-lightning<=1.9.4,>=1.5.10->keypoint-detection==1.0) (6.0.4)\n", + "Requirement already satisfied: yarl<2.0,>=1.0 in /usr/local/lib/python3.10/dist-packages (from aiohttp!=4.0.0a0,!=4.0.0a1->fsspec[http]>2021.06.0->pytorch-lightning<=1.9.4,>=1.5.10->keypoint-detection==1.0) (1.9.3)\n", + "Requirement already satisfied: frozenlist>=1.1.1 in /usr/local/lib/python3.10/dist-packages (from aiohttp!=4.0.0a0,!=4.0.0a1->fsspec[http]>2021.06.0->pytorch-lightning<=1.9.4,>=1.5.10->keypoint-detection==1.0) (1.4.0)\n", + "Requirement already satisfied: aiosignal>=1.1.2 in /usr/local/lib/python3.10/dist-packages (from aiohttp!=4.0.0a0,!=4.0.0a1->fsspec[http]>2021.06.0->pytorch-lightning<=1.9.4,>=1.5.10->keypoint-detection==1.0) (1.3.1)\n", + "Requirement already satisfied: async-timeout<5.0,>=4.0 in /usr/local/lib/python3.10/dist-packages (from aiohttp!=4.0.0a0,!=4.0.0a1->fsspec[http]>2021.06.0->pytorch-lightning<=1.9.4,>=1.5.10->keypoint-detection==1.0) (4.0.3)\n", + "Requirement already satisfied: sniffio>=1.1 in /usr/local/lib/python3.10/dist-packages (from anyio<5,>=3.4.0->starlette>=0.24.0->fiftyone->keypoint-detection==1.0) (1.3.0)\n", + "Requirement already satisfied: smmap<6,>=3.0.1 in /usr/local/lib/python3.10/dist-packages (from gitdb<5,>=4.0.1->GitPython!=3.1.29,>=1.0.0->wandb>=0.13.7->keypoint-detection==1.0) (5.0.1)\n", + "Requirement already satisfied: hyperframe<7,>=6.0 in /usr/local/lib/python3.10/dist-packages (from h2>=3.1.0->hypercorn>=0.13.2->fiftyone->keypoint-detection==1.0) (6.0.1)\n", + "Requirement already satisfied: hpack<5,>=4.0 in /usr/local/lib/python3.10/dist-packages (from h2>=3.1.0->hypercorn>=0.13.2->fiftyone->keypoint-detection==1.0) (4.0.0)\n", + "Requirement already satisfied: httpcore==1.* in /usr/local/lib/python3.10/dist-packages (from httpx>=0.10.0->universal-analytics-python3<2,>=1.0.1->fiftyone->keypoint-detection==1.0) (1.0.2)\n", + "Requirement already satisfied: texttable in /usr/local/lib/python3.10/dist-packages (from py7zr->voxel51-eta<0.13,>=0.12->fiftyone->keypoint-detection==1.0) (1.7.0)\n", + "Requirement already satisfied: pycryptodomex>=3.16.0 in /usr/local/lib/python3.10/dist-packages (from py7zr->voxel51-eta<0.13,>=0.12->fiftyone->keypoint-detection==1.0) (3.19.0)\n", + "Requirement already satisfied: pyzstd>=0.15.9 in /usr/local/lib/python3.10/dist-packages (from py7zr->voxel51-eta<0.13,>=0.12->fiftyone->keypoint-detection==1.0) (0.15.9)\n", + "Requirement already satisfied: pyppmd<1.2.0,>=1.1.0 in /usr/local/lib/python3.10/dist-packages (from py7zr->voxel51-eta<0.13,>=0.12->fiftyone->keypoint-detection==1.0) (1.1.0)\n", + "Requirement already satisfied: pybcj<1.1.0,>=1.0.0 in /usr/local/lib/python3.10/dist-packages (from py7zr->voxel51-eta<0.13,>=0.12->fiftyone->keypoint-detection==1.0) (1.0.2)\n", + "Requirement already satisfied: multivolumefile>=0.2.3 in /usr/local/lib/python3.10/dist-packages (from py7zr->voxel51-eta<0.13,>=0.12->fiftyone->keypoint-detection==1.0) (0.2.3)\n", + "Requirement already satisfied: inflate64<1.1.0,>=1.0.0 in /usr/local/lib/python3.10/dist-packages (from py7zr->voxel51-eta<0.13,>=0.12->fiftyone->keypoint-detection==1.0) (1.0.0)\n", + "Requirement already satisfied: brotli>=1.1.0 in /usr/local/lib/python3.10/dist-packages (from py7zr->voxel51-eta<0.13,>=0.12->fiftyone->keypoint-detection==1.0) (1.1.0)\n", + "Building wheels for collected packages: keypoint-detection\n", + " Building wheel for keypoint-detection (setup.py) ... \u001b[?25l\u001b[?25hdone\n", + " Created wheel for keypoint-detection: filename=keypoint_detection-1.0-py3-none-any.whl size=45730 sha256=80a52445b07e08fbb61ece2f10d99b6be8ae31c22f2ffe1422a4b45da11b2d40\n", + " Stored in directory: /root/.cache/pip/wheels/39/15/25/bb2380f6740e5bd39be30c0479272fa5daf04770870f14f3ff\n", + "Successfully built keypoint-detection\n", + "Installing collected packages: keypoint-detection\n", + " Attempting uninstall: keypoint-detection\n", + " Found existing installation: keypoint-detection 1.0\n", + " Uninstalling keypoint-detection-1.0:\n", + " Successfully uninstalled keypoint-detection-1.0\n", + "Successfully installed keypoint-detection-1.0\n" + ] + } + ] + }, + { + "cell_type": "code", + "source": [ + "# install font for visualization\n", + "!apt install fonts-freefont-ttf\n", + "\n", + "# install the torch-xla package for TPU acceleration\n", + "# instructions from https://lightning.ai/docs/pytorch/stable/accelerators/tpu_basic.html\n", + "!pip install cloud-tpu-client==0.10 https://storage.googleapis.com/tpu-pytorch/wheels/colab/torch_xla-2.0-cp310-cp310-linux_x86_64.whl\n", + "\n" + ], + "metadata": { + "colab": { + "base_uri": "https://localhost:8080/" + }, + "id": "at5pIAaU5KRW", + "outputId": "711c9f30-7628-4d21-a28e-c97d8c215643" + }, + "execution_count": 9, + "outputs": [ + { + "output_type": "stream", + "name": "stdout", + "text": [ + "Reading package lists... Done\n", + "Building dependency tree... Done\n", + "Reading state information... Done\n", + "fonts-freefont-ttf is already the newest version (20120503-10build1).\n", + "0 upgraded, 0 newly installed, 0 to remove and 15 not upgraded.\n", + "Collecting torch-xla==2.0\n", + " Downloading https://storage.googleapis.com/tpu-pytorch/wheels/colab/torch_xla-2.0-cp310-cp310-linux_x86_64.whl (162.9 MB)\n", + "\u001b[2K \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m162.9/162.9 MB\u001b[0m \u001b[31m7.1 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n", + "\u001b[?25hRequirement already satisfied: cloud-tpu-client==0.10 in /usr/local/lib/python3.10/dist-packages (0.10)\n", + "Requirement already satisfied: google-api-python-client==1.8.0 in /usr/local/lib/python3.10/dist-packages (from cloud-tpu-client==0.10) (1.8.0)\n", + "Requirement already satisfied: oauth2client in /usr/local/lib/python3.10/dist-packages (from cloud-tpu-client==0.10) (4.1.3)\n", + "Requirement already satisfied: httplib2<1dev,>=0.9.2 in /usr/local/lib/python3.10/dist-packages (from google-api-python-client==1.8.0->cloud-tpu-client==0.10) (0.22.0)\n", + "Requirement already satisfied: google-auth>=1.4.1 in /usr/local/lib/python3.10/dist-packages (from google-api-python-client==1.8.0->cloud-tpu-client==0.10) (2.17.3)\n", + "Requirement already satisfied: google-auth-httplib2>=0.0.3 in /usr/local/lib/python3.10/dist-packages (from google-api-python-client==1.8.0->cloud-tpu-client==0.10) (0.1.1)\n", + "Requirement already satisfied: google-api-core<2dev,>=1.13.0 in /usr/local/lib/python3.10/dist-packages (from google-api-python-client==1.8.0->cloud-tpu-client==0.10) (1.34.0)\n", + "Requirement already satisfied: six<2dev,>=1.6.1 in /usr/local/lib/python3.10/dist-packages (from google-api-python-client==1.8.0->cloud-tpu-client==0.10) (1.16.0)\n", + "Requirement already satisfied: uritemplate<4dev,>=3.0.0 in /usr/local/lib/python3.10/dist-packages (from google-api-python-client==1.8.0->cloud-tpu-client==0.10) (3.0.1)\n", + "Requirement already satisfied: absl-py>=1.0.0 in /usr/local/lib/python3.10/dist-packages (from torch-xla==2.0) (1.4.0)\n", + "Requirement already satisfied: pyasn1>=0.1.7 in /usr/local/lib/python3.10/dist-packages (from oauth2client->cloud-tpu-client==0.10) (0.5.1)\n", + "Requirement already satisfied: pyasn1-modules>=0.0.5 in /usr/local/lib/python3.10/dist-packages (from oauth2client->cloud-tpu-client==0.10) (0.3.0)\n", + "Requirement already satisfied: rsa>=3.1.4 in /usr/local/lib/python3.10/dist-packages (from oauth2client->cloud-tpu-client==0.10) (4.9)\n", + "Requirement already satisfied: googleapis-common-protos<2.0dev,>=1.56.2 in /usr/local/lib/python3.10/dist-packages (from google-api-core<2dev,>=1.13.0->google-api-python-client==1.8.0->cloud-tpu-client==0.10) (1.61.0)\n", + "Requirement already satisfied: protobuf!=3.20.0,!=3.20.1,!=4.21.0,!=4.21.1,!=4.21.2,!=4.21.3,!=4.21.4,!=4.21.5,<4.0.0dev,>=3.19.5 in /usr/local/lib/python3.10/dist-packages (from google-api-core<2dev,>=1.13.0->google-api-python-client==1.8.0->cloud-tpu-client==0.10) (3.20.3)\n", + "Requirement already satisfied: requests<3.0.0dev,>=2.18.0 in /usr/local/lib/python3.10/dist-packages (from google-api-core<2dev,>=1.13.0->google-api-python-client==1.8.0->cloud-tpu-client==0.10) (2.31.0)\n", + "Requirement already satisfied: cachetools<6.0,>=2.0.0 in /usr/local/lib/python3.10/dist-packages (from google-auth>=1.4.1->google-api-python-client==1.8.0->cloud-tpu-client==0.10) (5.3.2)\n", + "Requirement already satisfied: pyparsing!=3.0.0,!=3.0.1,!=3.0.2,!=3.0.3,<4,>=2.4.2 in /usr/local/lib/python3.10/dist-packages (from httplib2<1dev,>=0.9.2->google-api-python-client==1.8.0->cloud-tpu-client==0.10) (3.1.1)\n", + "Requirement already satisfied: charset-normalizer<4,>=2 in /usr/local/lib/python3.10/dist-packages (from requests<3.0.0dev,>=2.18.0->google-api-core<2dev,>=1.13.0->google-api-python-client==1.8.0->cloud-tpu-client==0.10) (3.3.2)\n", + "Requirement already satisfied: idna<4,>=2.5 in /usr/local/lib/python3.10/dist-packages (from requests<3.0.0dev,>=2.18.0->google-api-core<2dev,>=1.13.0->google-api-python-client==1.8.0->cloud-tpu-client==0.10) (3.6)\n", + "Requirement already satisfied: urllib3<3,>=1.21.1 in /usr/local/lib/python3.10/dist-packages (from requests<3.0.0dev,>=2.18.0->google-api-core<2dev,>=1.13.0->google-api-python-client==1.8.0->cloud-tpu-client==0.10) (2.0.7)\n", + "Requirement already satisfied: certifi>=2017.4.17 in /usr/local/lib/python3.10/dist-packages (from requests<3.0.0dev,>=2.18.0->google-api-core<2dev,>=1.13.0->google-api-python-client==1.8.0->cloud-tpu-client==0.10) (2023.11.17)\n", + "Installing collected packages: torch-xla\n", + " Attempting uninstall: torch-xla\n", + " Found existing installation: torch-xla 2.1.0\n", + " Uninstalling torch-xla-2.1.0:\n", + " Successfully uninstalled torch-xla-2.1.0\n", + "Successfully installed torch-xla-2.0.0.dev20230516+colab\n" + ] + } + ] + }, + { + "cell_type": "code", + "source": [ + "!wandb login" + ], + "metadata": { + "colab": { + "base_uri": "https://localhost:8080/" + }, + "id": "JR8p1gMW1hZ5", + "outputId": "658dfb4e-8303-4d67-c1af-3e6c0b142bc1" + }, + "execution_count": 6, + "outputs": [ + { + "output_type": "stream", + "name": "stdout", + "text": [ + "\u001b[34m\u001b[1mwandb\u001b[0m: Currently logged in as: \u001b[33mtlips\u001b[0m. Use \u001b[1m`wandb login --relogin`\u001b[0m to force relogin\n" + ] + } + ] + }, + { + "cell_type": "code", + "source": [ + "!keypoint-detection train --keypoint_channel_configuration \"box_corner0= box_corner1 = box_corner2= box_corner3: flap_corner0:flap_corner2\" \\\n", + "--json_dataset_path \"keypoint-detection/test/test_dataset/coco_dataset.json\" --json_validation_dataset_path \"keypoint-detection/test/test_dataset/coco_dataset.json\" --batch_size 2 --wandb_project \"keypoint-detector-integration-test\" \\\n", + "--max_epochs 50 --early_stopping_relative_threshold -1.0 --log_every_n_steps 1 --accelerator=\"auto\" --devices 1 --precision 16 --augment_train\n" + ], + "metadata": { + "colab": { + "base_uri": "https://localhost:8080/" + }, + "id": "7lU6vCZ91yOx", + "outputId": "1833df50-4020-4533-da29-7d5fb0314b0e" + }, + "execution_count": 10, + "outputs": [ + { + "output_type": "stream", + "name": "stdout", + "text": [ + "['/usr/local/bin/keypoint-detection', 'train', '--keypoint_channel_configuration', 'box_corner0= box_corner1 = box_corner2= box_corner3: flap_corner0:flap_corner2', '--json_dataset_path', 'keypoint-detection/test/test_dataset/coco_dataset.json', '--json_validation_dataset_path', 'keypoint-detection/test/test_dataset/coco_dataset.json', '--batch_size', '2', '--wandb_project', 'keypoint-detector-integration-test', '--max_epochs', '50', '--early_stopping_relative_threshold', '-1.0', '--log_every_n_steps', '1', '--accelerator=auto', '--devices', '1', '--precision', '16', '--augment_train']\n", + " argparse arguments ={'seed': 2022, 'wandb_project': 'keypoint-detector-integration-test', 'wandb_entity': None, 'wandb_name': None, 'keypoint_channel_configuration': [['box_corner0', 'box_corner1', 'box_corner2', 'box_corner3'], ['flap_corner0'], ['flap_corner2']], 'early_stopping_relative_threshold': -1.0, 'deterministic': True, 'wandb_checkpoint_artifact': None, 'heatmap_sigma': 2, 'minimal_keypoint_extraction_pixel_distance': 1, 'maximal_gt_keypoint_pixel_distances': '2 4', 'learning_rate': 0.0003, 'ap_epoch_start': 1, 'ap_epoch_freq': 2, 'lr_scheduler_relative_threshold': 0.0, 'max_keypoints': 20, 'logger': True, 'enable_checkpointing': True, 'default_root_dir': None, 'gradient_clip_val': None, 'gradient_clip_algorithm': None, 'num_nodes': 1, 'num_processes': None, 'devices': '1', 'gpus': None, 'auto_select_gpus': None, 'tpu_cores': None, 'ipus': None, 'enable_progress_bar': True, 'overfit_batches': 0.0, 'track_grad_norm': -1, 'check_val_every_n_epoch': 1, 'fast_dev_run': False, 'accumulate_grad_batches': None, 'max_epochs': 50, 'min_epochs': None, 'max_steps': -1, 'min_steps': None, 'max_time': None, 'limit_train_batches': None, 'limit_val_batches': None, 'limit_test_batches': None, 'limit_predict_batches': None, 'val_check_interval': None, 'log_every_n_steps': 1, 'accelerator': 'auto', 'strategy': None, 'sync_batchnorm': False, 'precision': 16, 'enable_model_summary': True, 'num_sanity_val_steps': 2, 'resume_from_checkpoint': None, 'profiler': None, 'benchmark': None, 'reload_dataloaders_every_n_epochs': 0, 'auto_lr_find': False, 'replace_sampler_ddp': True, 'detect_anomaly': False, 'auto_scale_batch_size': False, 'plugins': None, 'amp_backend': None, 'amp_level': None, 'move_metrics_to_cpu': False, 'multiple_trainloader_mode': 'max_size_cycle', 'inference_mode': True, 'batch_size': 2, 'validation_split_ratio': 0.25, 'num_workers': 4, 'json_dataset_path': 'keypoint-detection/test/test_dataset/coco_dataset.json', 'json_validation_dataset_path': 'keypoint-detection/test/test_dataset/coco_dataset.json', 'json_test_dataset_path': None, 'augment_train': True, 'detect_only_visible_keypoints': False, 'backbone_type': 'Unet', 'n_channels_in': 3, 'n_channels': 32, 'n_resnet_blocks': 3, 'n_downsampling_layers': 2, 'kernel_size': 3}\n", + "\u001b[34m\u001b[1mwandb\u001b[0m: Currently logged in as: \u001b[33mtlips\u001b[0m. Use \u001b[1m`wandb login --relogin`\u001b[0m to force relogin\n", + "\u001b[34m\u001b[1mwandb\u001b[0m: Tracking run with wandb version 0.16.1\n", + "\u001b[34m\u001b[1mwandb\u001b[0m: Run data is saved locally in \u001b[35m\u001b[1m/usr/local/lib/python3.10/dist-packages/logging/wandb/wandb/run-20231206_114137-zq6ry3jx\u001b[0m\n", + "\u001b[34m\u001b[1mwandb\u001b[0m: Run \u001b[1m`wandb offline`\u001b[0m to turn off syncing.\n", + "\u001b[34m\u001b[1mwandb\u001b[0m: Syncing run \u001b[33mbumbling-river-7\u001b[0m\n", + "\u001b[34m\u001b[1mwandb\u001b[0m: ⭐️ View project at \u001b[34m\u001b[4mhttps://wandb.ai/tlips/keypoint-detector-integration-test\u001b[0m\n", + "\u001b[34m\u001b[1mwandb\u001b[0m: 🚀 View run at \u001b[34m\u001b[4mhttps://wandb.ai/tlips/keypoint-detector-integration-test/runs/zq6ry3jx\u001b[0m\n", + " config after wandb init: {'seed': 2022, 'wandb_project': 'keypoint-detector-integration-test', 'wandb_entity': None, 'wandb_name': None, 'keypoint_channel_configuration': [['box_corner0', 'box_corner1', 'box_corner2', 'box_corner3'], ['flap_corner0'], ['flap_corner2']], 'early_stopping_relative_threshold': -1.0, 'deterministic': True, 'wandb_checkpoint_artifact': None, 'heatmap_sigma': 2, 'minimal_keypoint_extraction_pixel_distance': 1, 'maximal_gt_keypoint_pixel_distances': '2 4', 'learning_rate': 0.0003, 'ap_epoch_start': 1, 'ap_epoch_freq': 2, 'lr_scheduler_relative_threshold': 0.0, 'max_keypoints': 20, 'logger': True, 'enable_checkpointing': True, 'default_root_dir': None, 'gradient_clip_val': None, 'gradient_clip_algorithm': None, 'num_nodes': 1, 'num_processes': None, 'devices': '1', 'gpus': None, 'auto_select_gpus': None, 'tpu_cores': None, 'ipus': None, 'enable_progress_bar': True, 'overfit_batches': 0.0, 'track_grad_norm': -1, 'check_val_every_n_epoch': 1, 'fast_dev_run': False, 'accumulate_grad_batches': None, 'max_epochs': 50, 'min_epochs': None, 'max_steps': -1, 'min_steps': None, 'max_time': None, 'limit_train_batches': None, 'limit_val_batches': None, 'limit_test_batches': None, 'limit_predict_batches': None, 'val_check_interval': None, 'log_every_n_steps': 1, 'accelerator': 'auto', 'strategy': None, 'sync_batchnorm': False, 'precision': 16, 'enable_model_summary': True, 'num_sanity_val_steps': 2, 'resume_from_checkpoint': None, 'profiler': None, 'benchmark': None, 'reload_dataloaders_every_n_epochs': 0, 'auto_lr_find': False, 'replace_sampler_ddp': True, 'detect_anomaly': False, 'auto_scale_batch_size': False, 'plugins': None, 'amp_backend': None, 'amp_level': None, 'move_metrics_to_cpu': False, 'multiple_trainloader_mode': 'max_size_cycle', 'inference_mode': True, 'batch_size': 2, 'validation_split_ratio': 0.25, 'num_workers': 4, 'json_dataset_path': 'keypoint-detection/test/test_dataset/coco_dataset.json', 'json_validation_dataset_path': 'keypoint-detection/test/test_dataset/coco_dataset.json', 'json_test_dataset_path': None, 'augment_train': True, 'detect_only_visible_keypoints': False, 'backbone_type': 'Unet', 'n_channels_in': 3, 'n_channels': 32, 'n_resnet_blocks': 3, 'n_downsampling_layers': 2, 'kernel_size': 3}\n", + "starting training\n", + "Global seed set to 2022\n", + "detect_only_visible_keypoints=False\n", + "detect_only_visible_keypoints=False\n", + "Augmenting the training dataset!\n", + "/usr/local/lib/python3.10/dist-packages/pytorch_lightning/loggers/wandb.py:395: UserWarning: There is a wandb run already in progress and newly created instances of `WandbLogger` will reuse this run. If this is not desired, call `wandb.finish()` before instantiating `WandbLogger`.\n", + " rank_zero_warn(\n", + "Using 16bit None Automatic Mixed Precision (AMP)\n", + "GPU available: True (cuda), used: True\n", + "TPU available: False, using: 0 TPU cores\n", + "IPU available: False, using: 0 IPUs\n", + "HPU available: False, using: 0 HPUs\n", + "LOCAL_RANK: 0 - CUDA_VISIBLE_DEVICES: [0]\n", + "\n", + " | Name | Type | Params\n", + "--------------------------------------------------\n", + "0 | unnormalized_model | Sequential | 112 K \n", + "--------------------------------------------------\n", + "112 K Trainable params\n", + "0 Non-trainable params\n", + "112 K Total params\n", + "0.226 Total estimated model params size (MB)\n", + "Sanity Checking: 0it [00:00, ?it/s]/usr/local/lib/python3.10/dist-packages/torch/utils/data/dataloader.py:557: UserWarning: This DataLoader will create 4 worker processes in total. Our suggested max number of worker in current system is 2, which is smaller than what this DataLoader is going to create. Please be aware that excessive worker creation might get DataLoader running slow or even freeze, lower the worker number to avoid potential slowness/freeze if necessary.\n", + " warnings.warn(_create_warning_msg(\n", + "Sanity Checking DataLoader 0: 0% 0/2 [00:00= min_delta = 1.0. New best score: 0.2022384\n", + "Epoch 2: 0% 0/4 [00:00= min_delta = 1.0. New best score: 0.2017633\n", + "Epoch 3: 0% 0/4 [00:00= min_delta = 1.0. New best score: 0.2009415\n", + "Epoch 4: 0% 0/4 [00:00= min_delta = 1.0. New best score: 0.2001730\n", + "Epoch 5: 0% 0/4 [00:00= min_delta = 1.0. New best score: 0.1996601\n", + "Epoch 6: 0% 0/4 [00:00= min_delta = 1.0. New best score: 0.1995964\n", + "Epoch 7: 0% 0/4 [00:00= min_delta = 1.0. New best score: 0.1997790\n", + "Epoch 8: 0% 0/4 [00:00= min_delta = 1.0. New best score: 0.1999645\n", + "Epoch 9: 0% 0/4 [00:00= min_delta = 1.0. New best score: 0.2003065\n", + "Epoch 10: 0% 0/4 [00:00= min_delta = 1.0. New best score: 0.2007432\n", + "Epoch 11: 0% 0/4 [00:00= min_delta = 1.0. New best score: 0.2007637\n", + "Epoch 12: 0% 0/4 [00:00= min_delta = 1.0. New best score: 0.2005924\n", + "Epoch 13: 0% 0/4 [00:00= min_delta = 1.0. New best score: 0.2002886\n", + "Epoch 14: 0% 0/4 [00:00= min_delta = 1.0. New best score: 0.1994187\n", + "Epoch 15: 0% 0/4 [00:00= min_delta = 1.0. New best score: 0.1980296\n", + "Epoch 16: 0% 0/4 [00:00= min_delta = 1.0. New best score: 0.1956944\n", + "Epoch 17: 0% 0/4 [00:00= min_delta = 1.0. New best score: 0.1926106\n", + "Epoch 18: 0% 0/4 [00:00= min_delta = 1.0. New best score: 0.1885449\n", + "Epoch 19: 0% 0/4 [00:00= min_delta = 1.0. New best score: 0.1796938\n", + "Epoch 20: 0% 0/4 [00:00= min_delta = 1.0. New best score: 0.1687419\n", + "Epoch 21: 0% 0/4 [00:00= min_delta = 1.0. New best score: 0.1624074\n", + "Epoch 22: 0% 0/4 [00:00= min_delta = 1.0. New best score: 0.1578251\n", + "Epoch 23: 0% 0/4 [00:00= min_delta = 1.0. New best score: 0.1550909\n", + "Epoch 24: 0% 0/4 [00:00= min_delta = 1.0. New best score: 0.1496606\n", + "Epoch 25: 0% 0/4 [00:00= min_delta = 1.0. New best score: 0.1439323\n", + "Epoch 26: 0% 0/4 [00:00= min_delta = 1.0. New best score: 0.1406528\n", + "Epoch 27: 0% 0/4 [00:00= min_delta = 1.0. New best score: 0.1377558\n", + "Epoch 28: 0% 0/4 [00:00= min_delta = 1.0. New best score: 0.1348399\n", + "Epoch 29: 0% 0/4 [00:00= min_delta = 1.0. New best score: 0.1307023\n", + "Epoch 30: 0% 0/4 [00:00= min_delta = 1.0. New best score: 0.1277269\n", + "Epoch 31: 0% 0/4 [00:00= min_delta = 1.0. New best score: 0.1247422\n", + "Epoch 32: 0% 0/4 [00:00= min_delta = 1.0. New best score: 0.1203263\n", + "Epoch 33: 0% 0/4 [00:00= min_delta = 1.0. New best score: 0.1174109\n", + "Epoch 34: 0% 0/4 [00:00= min_delta = 1.0. New best score: 0.1146553\n", + "Epoch 35: 0% 0/4 [00:00= min_delta = 1.0. New best score: 0.1124170\n", + "Epoch 36: 0% 0/4 [00:00= min_delta = 1.0. New best score: 0.1102822\n", + "Epoch 37: 0% 0/4 [00:00= min_delta = 1.0. New best score: 0.1088773\n", + "Epoch 38: 0% 0/4 [00:00= min_delta = 1.0. New best score: 0.1086690\n", + "Epoch 39: 0% 0/4 [00:00= min_delta = 1.0. New best score: 0.1074851\n", + "Epoch 40: 0% 0/4 [00:00= min_delta = 1.0. New best score: 0.1057437\n", + "Epoch 41: 0% 0/4 [00:00= min_delta = 1.0. New best score: 0.1071071\n", + "Epoch 42: 0% 0/4 [00:00= min_delta = 1.0. New best score: 0.1082156\n", + "Epoch 43: 0% 0/4 [00:00= min_delta = 1.0. New best score: 0.1075207\n", + "Epoch 44: 0% 0/4 [00:00= min_delta = 1.0. New best score: 0.1102015\n", + "Epoch 45: 0% 0/4 [00:00= min_delta = 1.0. New best score: 0.1141067\n", + "Epoch 46: 0% 0/4 [00:00= min_delta = 1.0. New best score: 0.1144184\n", + "Epoch 47: 0% 0/4 [00:00= min_delta = 1.0. New best score: 0.1094995\n", + "Epoch 48: 0% 0/4 [00:00= min_delta = 1.0. New best score: 0.1064956\n", + "Epoch 49: 0% 0/4 [00:00= min_delta = 1.0. New best score: 0.1036285\n", + "Epoch 49: 100% 4/4 [00:00<00:00, 6.93it/s, loss=0.109, v_num=y3jx]`Trainer.fit` stopped: `max_epochs=50` reached.\n", + "Epoch 49: 100% 4/4 [00:00<00:00, 6.76it/s, loss=0.109, v_num=y3jx]\n", + "Restoring states from the checkpoint path at /usr/local/lib/python3.10/dist-packages/logging/wandb/lightning_logs/zq6ry3jx/checkpoints/epoch=49-step=100.ckpt\n", + "LOCAL_RANK: 0 - CUDA_VISIBLE_DEVICES: [0]\n", + "Loaded model weights from checkpoint at /usr/local/lib/python3.10/dist-packages/logging/wandb/lightning_logs/zq6ry3jx/checkpoints/epoch=49-step=100.ckpt\n", + "/usr/local/lib/python3.10/dist-packages/pytorch_lightning/trainer/connectors/data_connector.py:366: UserWarning: One of given dataloaders is None and it will be skipped.\n", + " rank_zero_warn(\"One of given dataloaders is None and it will be skipped.\")\n", + "\u001b[34m\u001b[1mwandb\u001b[0m: \n", + "\u001b[34m\u001b[1mwandb\u001b[0m: Run history:\n", + "\u001b[34m\u001b[1mwandb\u001b[0m: epoch ▁▁▁▁▂▂▂▂▂▃▃▃▃▃▃▄▄▄▄▄▅▅▅▅▅▅▆▆▆▆▆▇▇▇▇▇▇███\n", + "\u001b[34m\u001b[1mwandb\u001b[0m: train/['box_corner0', 'box_corner1', 'box_corner2', 'box_corner3'] █▅▅▄▄▄▃▃▃▃▃▃▄▃▃▄▃▂▃▂▃▂▂▂▂▂▂▂▁▂▂▁▁▁▁▁▁▁▁▁\n", + "\u001b[34m\u001b[1mwandb\u001b[0m: train/['box_corner0', 'box_corner1', 'box_corner2', 'box_corner3']_ap/d=2.0 ▁▁▁▃▂▄▂▂▅▃▆▅▇▆▅▇▇▇▆▆████▇\n", + "\u001b[34m\u001b[1mwandb\u001b[0m: train/['box_corner0', 'box_corner1', 'box_corner2', 'box_corner3']_ap/d=4.0 ▁▂▃▃▄▄▃▅▅▅▆▆▇▆▅▆▇▇█████▇▇\n", + "\u001b[34m\u001b[1mwandb\u001b[0m: train/['box_corner0', 'box_corner1', 'box_corner2', 'box_corner3']_ap/meanAP ▁▁▂▂▃▄▃▃▅▄▆▅▇▆▅▇▇▇▇▇████▇\n", + "\u001b[34m\u001b[1mwandb\u001b[0m: train/['flap_corner0'] █▇▇▅▆▅▄▄▄▄▃▃▄▃▄▄▄▃▃▃▃▂▂▂▃▂▃▃▂▃▂▃▂▁▁▁▁▁▁▁\n", + "\u001b[34m\u001b[1mwandb\u001b[0m: train/['flap_corner0']_ap/d=2.0 ▁▂▃▂▅▅▅▅▄▅▅▄▆▅▄████████▇▆\n", + "\u001b[34m\u001b[1mwandb\u001b[0m: train/['flap_corner0']_ap/d=4.0 ▁▂▁▅▃▅▅▆▄▄▆▇▇▆▄████████▇█\n", + "\u001b[34m\u001b[1mwandb\u001b[0m: train/['flap_corner0']_ap/meanAP ▁▂▂▃▄▅▅▅▄▄▆▅▇▅▄████████▇▇\n", + "\u001b[34m\u001b[1mwandb\u001b[0m: train/['flap_corner2'] █▇▆▅▅▅▄▄▄▄▄▄▄▄▅▄▄▃▄▃▃▃▃▃▃▂▂▂▂▂▂▂▂▁▂▂▁▁▁▁\n", + "\u001b[34m\u001b[1mwandb\u001b[0m: train/['flap_corner2']_ap/d=2.0 ▁▂▁▃▂▂▂▂▇▄▅▃▅▅▆▇█▆██▆████\n", + "\u001b[34m\u001b[1mwandb\u001b[0m: train/['flap_corner2']_ap/d=4.0 ▂▁▃▆▅▆▇▅▇▅▅█▅▇▆▆█▇██▇████\n", + "\u001b[34m\u001b[1mwandb\u001b[0m: train/['flap_corner2']_ap/meanAP ▁▁▂▄▃▃▄▃▇▄▅▅▅▆▆▆█▆██▆████\n", + "\u001b[34m\u001b[1mwandb\u001b[0m: train/gt_loss ▇▂▁█▄██▄▂▇▁█▇▇▂▁▄█▄█▂█▇▂▃█▂▃█▂▁▄▁█▄█▁▂██\n", + "\u001b[34m\u001b[1mwandb\u001b[0m: train/loss_epoch █▆▆▅▅▄▄▄▄▄▃▃▃▃▃▃▃▃▃▂▃▂▂▂▂▂▂▂▂▂▁▁▁▁▁▂▁▁▁▁\n", + "\u001b[34m\u001b[1mwandb\u001b[0m: train/loss_step █▆▅▅▅▄▄▄▃▄▃▃▄▃▄▄▃▃▃▂▃▂▂▂▂▂▂▂▁▂▂▂▂▁▁▁▁▁▁▁\n", + "\u001b[34m\u001b[1mwandb\u001b[0m: train/meanAP ▁▁▂▃▃▄▄▄▅▄▅▅▆▆▅▇█▇██▇███▇\n", + "\u001b[34m\u001b[1mwandb\u001b[0m: train/meanAP/d=2.0 ▁▂▂▂▃▃▃▃▅▄▅▄▆▅▅▇█▇▇▇▇███▇\n", + "\u001b[34m\u001b[1mwandb\u001b[0m: train/meanAP/d=4.0 ▁▁▂▅▄▅▅▅▅▄▆▇▆▆▅▇▇▇█████▇█\n", + "\u001b[34m\u001b[1mwandb\u001b[0m: train/meanAP/meanAP ▁▁▂▃▃▄▄▄▅▄▅▅▆▆▅▇█▇██▇███▇\n", + "\u001b[34m\u001b[1mwandb\u001b[0m: trainer/global_step ▁▁▁▁▂▂▂▂▂▃▃▃▃▃▃▄▄▄▄▄▅▅▅▅▅▅▆▆▆▆▆▇▇▇▇▇▇███\n", + "\u001b[34m\u001b[1mwandb\u001b[0m: validation/['box_corner0', 'box_corner1', 'box_corner2', 'box_corner3']_ap/d=2.0 ▁▁▁▁▁▁▁▁▁▃▃▅▇▆▇▇▇▇█▇█████\n", + "\u001b[34m\u001b[1mwandb\u001b[0m: validation/['box_corner0', 'box_corner1', 'box_corner2', 'box_corner3']_ap/d=4.0 ▁▂▁▁▂▁▁▂▃▄▆▆▇▇▇█▇██▇█████\n", + "\u001b[34m\u001b[1mwandb\u001b[0m: validation/['box_corner0', 'box_corner1', 'box_corner2', 'box_corner3']_ap/meanAP ▁▁▁▁▁▁▁▂▂▃▄▆▇▇▇▇▇▇█▇█████\n", + "\u001b[34m\u001b[1mwandb\u001b[0m: validation/['flap_corner0']_ap/d=2.0 ▁▁▁▁▁▁▃▃▁▂▅▆▆▅▆▆█▇▇██▆▇██\n", + "\u001b[34m\u001b[1mwandb\u001b[0m: validation/['flap_corner0']_ap/d=4.0 ▁▂▃▄▃▄▃▅▅▅▆▇▆▇▆▆█▇▇██▆▇██\n", + "\u001b[34m\u001b[1mwandb\u001b[0m: validation/['flap_corner0']_ap/meanAP ▁▁▂▂▂▂▃▄▃▃▅▇▆▆▆▆█▇▇██▆▇██\n", + "\u001b[34m\u001b[1mwandb\u001b[0m: validation/['flap_corner2']_ap/d=2.0 ▁▁▁▁▁▁▁▁▃▁▅██▆███████████\n", + "\u001b[34m\u001b[1mwandb\u001b[0m: validation/['flap_corner2']_ap/d=4.0 ▁▁▁▁▁▁▁▁▄▄▇██████████████\n", + "\u001b[34m\u001b[1mwandb\u001b[0m: validation/['flap_corner2']_ap/meanAP ▁▁▁▁▁▁▁▁▃▂▆██▇███████████\n", + "\u001b[34m\u001b[1mwandb\u001b[0m: validation/epoch_loss ██████████████▇▇▆▅▅▅▄▄▃▃▃▂▂▂▂▁▁▁▁▁▁▁▂▂▁▁\n", + "\u001b[34m\u001b[1mwandb\u001b[0m: validation/gt_loss ▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁\n", + "\u001b[34m\u001b[1mwandb\u001b[0m: validation/meanAP ▁▁▁▁▁▁▂▂▃▃▅▇▇▇▇▇█████▇███\n", + "\u001b[34m\u001b[1mwandb\u001b[0m: validation/meanAP/d=2.0 ▁▁▁▁▁▁▂▂▂▂▄▆▇▆▇▇█▇███▇███\n", + "\u001b[34m\u001b[1mwandb\u001b[0m: validation/meanAP/d=4.0 ▁▂▂▂▂▂▂▃▄▄▆▇▇▇▇▇█████▇███\n", + "\u001b[34m\u001b[1mwandb\u001b[0m: validation/meanAP/meanAP ▁▁▁▁▁▁▂▂▃▃▅▇▇▇▇▇█████▇███\n", + "\u001b[34m\u001b[1mwandb\u001b[0m: \n", + "\u001b[34m\u001b[1mwandb\u001b[0m: Run summary:\n", + "\u001b[34m\u001b[1mwandb\u001b[0m: epoch 49\n", + "\u001b[34m\u001b[1mwandb\u001b[0m: train/['box_corner0', 'box_corner1', 'box_corner2', 'box_corner3'] 0.05829\n", + "\u001b[34m\u001b[1mwandb\u001b[0m: train/['box_corner0', 'box_corner1', 'box_corner2', 'box_corner3']_ap/d=2.0 0.76818\n", + "\u001b[34m\u001b[1mwandb\u001b[0m: train/['box_corner0', 'box_corner1', 'box_corner2', 'box_corner3']_ap/d=4.0 0.88098\n", + "\u001b[34m\u001b[1mwandb\u001b[0m: train/['box_corner0', 'box_corner1', 'box_corner2', 'box_corner3']_ap/meanAP 0.82458\n", + "\u001b[34m\u001b[1mwandb\u001b[0m: train/['flap_corner0'] 0.0251\n", + "\u001b[34m\u001b[1mwandb\u001b[0m: train/['flap_corner0']_ap/d=2.0 0.6875\n", + "\u001b[34m\u001b[1mwandb\u001b[0m: train/['flap_corner0']_ap/d=4.0 1.0\n", + "\u001b[34m\u001b[1mwandb\u001b[0m: train/['flap_corner0']_ap/meanAP 0.84375\n", + "\u001b[34m\u001b[1mwandb\u001b[0m: train/['flap_corner2'] 0.0242\n", + "\u001b[34m\u001b[1mwandb\u001b[0m: train/['flap_corner2']_ap/d=2.0 1.0\n", + "\u001b[34m\u001b[1mwandb\u001b[0m: train/['flap_corner2']_ap/d=4.0 1.0\n", + "\u001b[34m\u001b[1mwandb\u001b[0m: train/['flap_corner2']_ap/meanAP 1.0\n", + "\u001b[34m\u001b[1mwandb\u001b[0m: train/gt_loss 0.06045\n", + "\u001b[34m\u001b[1mwandb\u001b[0m: train/loss_epoch 0.10923\n", + "\u001b[34m\u001b[1mwandb\u001b[0m: train/loss_step 0.10758\n", + "\u001b[34m\u001b[1mwandb\u001b[0m: train/meanAP 0.88944\n", + "\u001b[34m\u001b[1mwandb\u001b[0m: train/meanAP/d=2.0 0.81856\n", + "\u001b[34m\u001b[1mwandb\u001b[0m: train/meanAP/d=4.0 0.96033\n", + "\u001b[34m\u001b[1mwandb\u001b[0m: train/meanAP/meanAP 0.88944\n", + "\u001b[34m\u001b[1mwandb\u001b[0m: trainer/global_step 99\n", + "\u001b[34m\u001b[1mwandb\u001b[0m: validation/['box_corner0', 'box_corner1', 'box_corner2', 'box_corner3']_ap/d=2.0 0.95665\n", + "\u001b[34m\u001b[1mwandb\u001b[0m: validation/['box_corner0', 'box_corner1', 'box_corner2', 'box_corner3']_ap/d=4.0 0.95983\n", + "\u001b[34m\u001b[1mwandb\u001b[0m: validation/['box_corner0', 'box_corner1', 'box_corner2', 'box_corner3']_ap/meanAP 0.95824\n", + "\u001b[34m\u001b[1mwandb\u001b[0m: validation/['flap_corner0']_ap/d=2.0 1.0\n", + "\u001b[34m\u001b[1mwandb\u001b[0m: validation/['flap_corner0']_ap/d=4.0 1.0\n", + "\u001b[34m\u001b[1mwandb\u001b[0m: validation/['flap_corner0']_ap/meanAP 1.0\n", + "\u001b[34m\u001b[1mwandb\u001b[0m: validation/['flap_corner2']_ap/d=2.0 1.0\n", + "\u001b[34m\u001b[1mwandb\u001b[0m: validation/['flap_corner2']_ap/d=4.0 1.0\n", + "\u001b[34m\u001b[1mwandb\u001b[0m: validation/['flap_corner2']_ap/meanAP 1.0\n", + "\u001b[34m\u001b[1mwandb\u001b[0m: validation/epoch_loss 0.10363\n", + "\u001b[34m\u001b[1mwandb\u001b[0m: validation/gt_loss 0.05993\n", + "\u001b[34m\u001b[1mwandb\u001b[0m: validation/meanAP 0.98608\n", + "\u001b[34m\u001b[1mwandb\u001b[0m: validation/meanAP/d=2.0 0.98555\n", + "\u001b[34m\u001b[1mwandb\u001b[0m: validation/meanAP/d=4.0 0.98661\n", + "\u001b[34m\u001b[1mwandb\u001b[0m: validation/meanAP/meanAP 0.98608\n", + "\u001b[34m\u001b[1mwandb\u001b[0m: \n", + "\u001b[34m\u001b[1mwandb\u001b[0m: 🚀 View run \u001b[33mbumbling-river-7\u001b[0m at: \u001b[34m\u001b[4mhttps://wandb.ai/tlips/keypoint-detector-integration-test/runs/zq6ry3jx\u001b[0m\n", + "\u001b[34m\u001b[1mwandb\u001b[0m: Synced 5 W&B file(s), 175 media file(s), 1 artifact file(s) and 0 other file(s)\n", + "\u001b[34m\u001b[1mwandb\u001b[0m: Find logs at: \u001b[35m\u001b[1m/usr/local/lib/python3.10/dist-packages/logging/wandb/wandb/run-20231206_114137-zq6ry3jx/logs\u001b[0m\n" + ] + } + ] + }, + { + "cell_type": "code", + "source": [], + "metadata": { + "id": "sz0TbOIj1HDB" + }, + "execution_count": 5, + "outputs": [] + } + ] +} \ No newline at end of file