From 619b98b121ffed529ff3e731fba2a76918fbfaa2 Mon Sep 17 00:00:00 2001
From: Neelesh Karthikeyan <121501350+nee1k@users.noreply.github.com>
Date: Sat, 1 Feb 2025 12:48:44 -0500
Subject: [PATCH] Delete examples/notebooks/megaDet.ipynb
---
examples/notebooks/megaDet.ipynb | 1189 ------------------------------
1 file changed, 1189 deletions(-)
delete mode 100644 examples/notebooks/megaDet.ipynb
diff --git a/examples/notebooks/megaDet.ipynb b/examples/notebooks/megaDet.ipynb
deleted file mode 100644
index 6e85096..0000000
--- a/examples/notebooks/megaDet.ipynb
+++ /dev/null
@@ -1,1189 +0,0 @@
-{
- "cells": [
- {
- "cell_type": "markdown",
- "metadata": {
- "id": "view-in-github",
- "colab_type": "text"
- },
- "source": [
- ""
- ]
- },
- {
- "cell_type": "markdown",
- "id": "b739c0e2",
- "metadata": {
- "id": "b739c0e2"
- },
- "source": [
- "# Image Detection Demo with PytorchWildlife"
- ]
- },
- {
- "cell_type": "code",
- "execution_count": null,
- "id": "hmVk-Zi1IhGi",
- "metadata": {
- "colab": {
- "base_uri": "https://localhost:8080/"
- },
- "id": "hmVk-Zi1IhGi",
- "outputId": "30588612-f9c6-4804-b2fb-df78799b55fb"
- },
- "outputs": [
- {
- "output_type": "stream",
- "name": "stdout",
- "text": [
- "Tue Jan 28 14:42:34 2025 \n",
- "+---------------------------------------------------------------------------------------+\n",
- "| NVIDIA-SMI 535.104.05 Driver Version: 535.104.05 CUDA Version: 12.2 |\n",
- "|-----------------------------------------+----------------------+----------------------+\n",
- "| GPU Name Persistence-M | Bus-Id Disp.A | Volatile Uncorr. ECC |\n",
- "| Fan Temp Perf Pwr:Usage/Cap | Memory-Usage | GPU-Util Compute M. |\n",
- "| | | MIG M. |\n",
- "|=========================================+======================+======================|\n",
- "| 0 Tesla T4 Off | 00000000:00:04.0 Off | 0 |\n",
- "| N/A 51C P8 12W / 70W | 0MiB / 15360MiB | 0% Default |\n",
- "| | | N/A |\n",
- "+-----------------------------------------+----------------------+----------------------+\n",
- " \n",
- "+---------------------------------------------------------------------------------------+\n",
- "| Processes: |\n",
- "| GPU GI CI PID Type Process name GPU Memory |\n",
- "| ID ID Usage |\n",
- "|=======================================================================================|\n",
- "| No running processes found |\n",
- "+---------------------------------------------------------------------------------------+\n"
- ]
- }
- ],
- "source": [
- "!nvidia-smi"
- ]
- },
- {
- "cell_type": "code",
- "execution_count": null,
- "id": "wNOxLWOPWBEU",
- "metadata": {
- "colab": {
- "base_uri": "https://localhost:8080/"
- },
- "collapsed": true,
- "id": "wNOxLWOPWBEU",
- "outputId": "bc077412-05fe-4b63-a247-c8bfc83bb3ee"
- },
- "outputs": [
- {
- "output_type": "stream",
- "name": "stdout",
- "text": [
- "\r0% [Working]\r \rGet:1 https://cloud.r-project.org/bin/linux/ubuntu jammy-cran40/ InRelease [3,626 B]\n",
- "Get:2 https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64 InRelease [1,581 B]\n",
- "Get:3 https://cloud.r-project.org/bin/linux/ubuntu jammy-cran40/ Packages [62.7 kB]\n",
- "Get:4 http://security.ubuntu.com/ubuntu jammy-security InRelease [129 kB]\n",
- "Get:5 https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64 Packages [1,303 kB]\n",
- "Hit:6 http://archive.ubuntu.com/ubuntu jammy InRelease\n",
- "Get:7 http://archive.ubuntu.com/ubuntu jammy-updates InRelease [128 kB]\n",
- "Get:8 https://r2u.stat.illinois.edu/ubuntu jammy InRelease [6,555 B]\n",
- "Hit:9 https://ppa.launchpadcontent.net/deadsnakes/ppa/ubuntu jammy InRelease\n",
- "Hit:10 https://ppa.launchpadcontent.net/graphics-drivers/ppa/ubuntu jammy InRelease\n",
- "Get:11 http://security.ubuntu.com/ubuntu jammy-security/restricted amd64 Packages [3,601 kB]\n",
- "Hit:12 https://ppa.launchpadcontent.net/ubuntugis/ppa/ubuntu jammy InRelease\n",
- "Get:13 https://r2u.stat.illinois.edu/ubuntu jammy/main amd64 Packages [2,646 kB]\n",
- "Get:14 http://archive.ubuntu.com/ubuntu jammy-backports InRelease [127 kB]\n",
- "Get:15 http://security.ubuntu.com/ubuntu jammy-security/universe amd64 Packages [1,229 kB]\n",
- "Get:16 http://archive.ubuntu.com/ubuntu jammy-updates/restricted amd64 Packages [3,742 kB]\n",
- "Get:17 http://security.ubuntu.com/ubuntu jammy-security/main amd64 Packages [2,601 kB]\n",
- "Get:18 https://r2u.stat.illinois.edu/ubuntu jammy/main all Packages [8,631 kB]\n",
- "Get:19 http://archive.ubuntu.com/ubuntu jammy-updates/universe amd64 Packages [1,521 kB]\n",
- "Get:20 http://archive.ubuntu.com/ubuntu jammy-updates/main amd64 Packages [2,904 kB]\n",
- "Fetched 28.6 MB in 8s (3,627 kB/s)\n",
- "Reading package lists... Done\n",
- "W: Skipping acquire of configured file 'main/source/Sources' as repository 'https://r2u.stat.illinois.edu/ubuntu jammy InRelease' does not seem to provide it (sources.list entry misspelt?)\n",
- "Reading package lists... Done\n",
- "Building dependency tree... Done\n",
- "Reading state information... Done\n",
- "python3 is already the newest version (3.10.6-1~22.04.1).\n",
- "python3 set to manually installed.\n",
- "0 upgraded, 0 newly installed, 0 to remove and 54 not upgraded.\n",
- "update-alternatives: error: alternative path /usr/bin/python3.8 doesn't exist\n",
- "There are 2 choices for the alternative python3 (providing /usr/bin/python3).\n",
- "\n",
- " Selection Path Priority Status\n",
- "------------------------------------------------------------\n",
- "* 0 /usr/bin/python3.11 2 auto mode\n",
- " 1 /usr/bin/python3.10 1 manual mode\n",
- " 2 /usr/bin/python3.11 2 manual mode\n",
- "\n",
- "Press to keep the current choice[*], or type selection number: \n"
- ]
- }
- ],
- "source": [
- "!sudo apt-get update -y\n",
- "!sudo apt-get install python3\n",
- "!sudo update-alternatives --install /usr/bin/python3 python3 /usr/bin/python3.8 1\n",
- "!sudo update-alternatives --config python3"
- ]
- },
- {
- "cell_type": "code",
- "execution_count": null,
- "id": "SYbWG-YE7A4_",
- "metadata": {
- "id": "SYbWG-YE7A4_",
- "colab": {
- "base_uri": "https://localhost:8080/"
- },
- "outputId": "53e0b55b-8309-4927-84af-010b0fe515d7"
- },
- "outputs": [
- {
- "output_type": "stream",
- "name": "stdout",
- "text": [
- "3.11.11 (main, Dec 4 2024, 08:55:07) [GCC 11.4.0]\n"
- ]
- }
- ],
- "source": [
- "import sys\n",
- "print(sys.version)"
- ]
- },
- {
- "cell_type": "code",
- "execution_count": null,
- "id": "ogxTrwjxaILl",
- "metadata": {
- "collapsed": true,
- "id": "ogxTrwjxaILl",
- "colab": {
- "base_uri": "https://localhost:8080/"
- },
- "outputId": "1849f24a-8d07-48a9-a6a2-fd2f5b6e1cd1"
- },
- "outputs": [
- {
- "output_type": "stream",
- "name": "stdout",
- "text": [
- "Reading package lists... Done\n",
- "Building dependency tree... Done\n",
- "Reading state information... Done\n",
- "The following additional packages will be installed:\n",
- " python3-setuptools python3-wheel\n",
- "Suggested packages:\n",
- " python-setuptools-doc\n",
- "The following NEW packages will be installed:\n",
- " python3-pip python3-setuptools python3-wheel\n",
- "0 upgraded, 3 newly installed, 0 to remove and 54 not upgraded.\n",
- "Need to get 1,677 kB of archives.\n",
- "After this operation, 8,968 kB of additional disk space will be used.\n",
- "Get:1 http://archive.ubuntu.com/ubuntu jammy-updates/main amd64 python3-setuptools all 59.6.0-1.2ubuntu0.22.04.2 [340 kB]\n",
- "Get:2 http://archive.ubuntu.com/ubuntu jammy-updates/universe amd64 python3-wheel all 0.37.1-2ubuntu0.22.04.1 [32.0 kB]\n",
- "Get:3 http://archive.ubuntu.com/ubuntu jammy-updates/universe amd64 python3-pip all 22.0.2+dfsg-1ubuntu0.5 [1,306 kB]\n",
- "Fetched 1,677 kB in 2s (674 kB/s)\n",
- "debconf: unable to initialize frontend: Dialog\n",
- "debconf: (No usable dialog-like program is installed, so the dialog based frontend cannot be used. at /usr/share/perl5/Debconf/FrontEnd/Dialog.pm line 78, <> line 3.)\n",
- "debconf: falling back to frontend: Readline\n",
- "debconf: unable to initialize frontend: Readline\n",
- "debconf: (This frontend requires a controlling tty.)\n",
- "debconf: falling back to frontend: Teletype\n",
- "dpkg-preconfigure: unable to re-open stdin: \n",
- "Selecting previously unselected package python3-setuptools.\n",
- "(Reading database ... 124788 files and directories currently installed.)\n",
- "Preparing to unpack .../python3-setuptools_59.6.0-1.2ubuntu0.22.04.2_all.deb ...\n",
- "Unpacking python3-setuptools (59.6.0-1.2ubuntu0.22.04.2) ...\n",
- "Selecting previously unselected package python3-wheel.\n",
- "Preparing to unpack .../python3-wheel_0.37.1-2ubuntu0.22.04.1_all.deb ...\n",
- "Unpacking python3-wheel (0.37.1-2ubuntu0.22.04.1) ...\n",
- "Selecting previously unselected package python3-pip.\n",
- "Preparing to unpack .../python3-pip_22.0.2+dfsg-1ubuntu0.5_all.deb ...\n",
- "Unpacking python3-pip (22.0.2+dfsg-1ubuntu0.5) ...\n",
- "Setting up python3-setuptools (59.6.0-1.2ubuntu0.22.04.2) ...\n",
- "Setting up python3-wheel (0.37.1-2ubuntu0.22.04.1) ...\n",
- "Setting up python3-pip (22.0.2+dfsg-1ubuntu0.5) ...\n",
- "Processing triggers for man-db (2.10.2-1) ...\n"
- ]
- }
- ],
- "source": [
- "!sudo apt install python3-pip"
- ]
- },
- {
- "cell_type": "code",
- "execution_count": null,
- "id": "Q0FvMS2hfGNr",
- "metadata": {
- "id": "Q0FvMS2hfGNr",
- "colab": {
- "base_uri": "https://localhost:8080/"
- },
- "outputId": "7cce4b26-e080-48f7-8c65-f5ca2625269c"
- },
- "outputs": [
- {
- "output_type": "stream",
- "name": "stdout",
- "text": [
- "Name: torch\n",
- "Version: 2.5.1+cu121\n",
- "Summary: Tensors and Dynamic neural networks in Python with strong GPU acceleration\n",
- "Home-page: https://pytorch.org/\n",
- "Author: PyTorch Team\n",
- "Author-email: packages@pytorch.org\n",
- "License: BSD-3-Clause\n",
- "Location: /usr/local/lib/python3.11/dist-packages\n",
- "Requires: filelock, fsspec, jinja2, networkx, nvidia-cublas-cu12, nvidia-cuda-cupti-cu12, nvidia-cuda-nvrtc-cu12, nvidia-cuda-runtime-cu12, nvidia-cudnn-cu12, nvidia-cufft-cu12, nvidia-curand-cu12, nvidia-cusolver-cu12, nvidia-cusparse-cu12, nvidia-nccl-cu12, nvidia-nvtx-cu12, sympy, triton, typing-extensions\n",
- "Required-by: accelerate, fastai, peft, sentence-transformers, timm, torchaudio, torchvision\n"
- ]
- }
- ],
- "source": [
- "!pip show torch"
- ]
- },
- {
- "cell_type": "code",
- "execution_count": null,
- "id": "WGoB82wyt3oY",
- "metadata": {
- "id": "WGoB82wyt3oY",
- "colab": {
- "base_uri": "https://localhost:8080/"
- },
- "outputId": "cfd0c1e5-5c5e-48bf-9994-6c8c451cfe2b"
- },
- "outputs": [
- {
- "output_type": "stream",
- "name": "stdout",
- "text": [
- "2.5.1+cu121\n",
- "0.20.1+cu121\n",
- "2.5.1+cu121\n"
- ]
- }
- ],
- "source": [
- "import torch, torchvision, torchaudio\n",
- "print(torch.__version__)\n",
- "print(torchvision.__version__)\n",
- "print(torchaudio.__version__)"
- ]
- },
- {
- "cell_type": "code",
- "execution_count": null,
- "id": "wu5UIVFVvbHQ",
- "metadata": {
- "id": "wu5UIVFVvbHQ",
- "colab": {
- "base_uri": "https://localhost:8080/",
- "height": 1000
- },
- "outputId": "b82fe1a2-2cb1-4cba-9b86-6eecf4e25320"
- },
- "outputs": [
- {
- "output_type": "stream",
- "name": "stdout",
- "text": [
- "Collecting torch==2.2.1\n",
- " Downloading torch-2.2.1-cp311-cp311-manylinux1_x86_64.whl.metadata (26 kB)\n",
- "Collecting torchvision==0.17.1\n",
- " Downloading torchvision-0.17.1-cp311-cp311-manylinux1_x86_64.whl.metadata (6.6 kB)\n",
- "Collecting torchaudio==2.2.1\n",
- " Downloading torchaudio-2.2.1-cp311-cp311-manylinux1_x86_64.whl.metadata (6.4 kB)\n",
- "Requirement already satisfied: filelock in /usr/local/lib/python3.11/dist-packages (from torch==2.2.1) (3.17.0)\n",
- "Requirement already satisfied: typing-extensions>=4.8.0 in /usr/local/lib/python3.11/dist-packages (from torch==2.2.1) (4.12.2)\n",
- "Requirement already satisfied: sympy in /usr/local/lib/python3.11/dist-packages (from torch==2.2.1) (1.13.1)\n",
- "Requirement already satisfied: networkx in /usr/local/lib/python3.11/dist-packages (from torch==2.2.1) (3.4.2)\n",
- "Requirement already satisfied: jinja2 in /usr/local/lib/python3.11/dist-packages (from torch==2.2.1) (3.1.5)\n",
- "Requirement already satisfied: fsspec in /usr/local/lib/python3.11/dist-packages (from torch==2.2.1) (2024.10.0)\n",
- "Requirement already satisfied: nvidia-cuda-nvrtc-cu12==12.1.105 in /usr/local/lib/python3.11/dist-packages (from torch==2.2.1) (12.1.105)\n",
- "Requirement already satisfied: nvidia-cuda-runtime-cu12==12.1.105 in /usr/local/lib/python3.11/dist-packages (from torch==2.2.1) (12.1.105)\n",
- "Requirement already satisfied: nvidia-cuda-cupti-cu12==12.1.105 in /usr/local/lib/python3.11/dist-packages (from torch==2.2.1) (12.1.105)\n",
- "Collecting nvidia-cudnn-cu12==8.9.2.26 (from torch==2.2.1)\n",
- " Downloading nvidia_cudnn_cu12-8.9.2.26-py3-none-manylinux1_x86_64.whl.metadata (1.6 kB)\n",
- "Requirement already satisfied: nvidia-cublas-cu12==12.1.3.1 in /usr/local/lib/python3.11/dist-packages (from torch==2.2.1) (12.1.3.1)\n",
- "Requirement already satisfied: nvidia-cufft-cu12==11.0.2.54 in /usr/local/lib/python3.11/dist-packages (from torch==2.2.1) (11.0.2.54)\n",
- "Requirement already satisfied: nvidia-curand-cu12==10.3.2.106 in /usr/local/lib/python3.11/dist-packages (from torch==2.2.1) (10.3.2.106)\n",
- "Requirement already satisfied: nvidia-cusolver-cu12==11.4.5.107 in /usr/local/lib/python3.11/dist-packages (from torch==2.2.1) (11.4.5.107)\n",
- "Requirement already satisfied: nvidia-cusparse-cu12==12.1.0.106 in /usr/local/lib/python3.11/dist-packages (from torch==2.2.1) (12.1.0.106)\n",
- "Collecting nvidia-nccl-cu12==2.19.3 (from torch==2.2.1)\n",
- " Downloading nvidia_nccl_cu12-2.19.3-py3-none-manylinux1_x86_64.whl.metadata (1.8 kB)\n",
- "Requirement already satisfied: nvidia-nvtx-cu12==12.1.105 in /usr/local/lib/python3.11/dist-packages (from torch==2.2.1) (12.1.105)\n",
- "Collecting triton==2.2.0 (from torch==2.2.1)\n",
- " Downloading triton-2.2.0-cp311-cp311-manylinux_2_17_x86_64.manylinux2014_x86_64.whl.metadata (1.4 kB)\n",
- "Requirement already satisfied: numpy in /usr/local/lib/python3.11/dist-packages (from torchvision==0.17.1) (1.26.4)\n",
- "Requirement already satisfied: pillow!=8.3.*,>=5.3.0 in /usr/local/lib/python3.11/dist-packages (from torchvision==0.17.1) (11.1.0)\n",
- "Requirement already satisfied: nvidia-nvjitlink-cu12 in /usr/local/lib/python3.11/dist-packages (from nvidia-cusolver-cu12==11.4.5.107->torch==2.2.1) (12.6.85)\n",
- "Requirement already satisfied: MarkupSafe>=2.0 in /usr/local/lib/python3.11/dist-packages (from jinja2->torch==2.2.1) (3.0.2)\n",
- "Requirement already satisfied: mpmath<1.4,>=1.1.0 in /usr/local/lib/python3.11/dist-packages (from sympy->torch==2.2.1) (1.3.0)\n",
- "Downloading torch-2.2.1-cp311-cp311-manylinux1_x86_64.whl (755.6 MB)\n",
- "\u001b[2K \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m755.6/755.6 MB\u001b[0m \u001b[31m2.2 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
- "\u001b[?25hDownloading torchvision-0.17.1-cp311-cp311-manylinux1_x86_64.whl (6.9 MB)\n",
- "\u001b[2K \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m6.9/6.9 MB\u001b[0m \u001b[31m104.2 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
- "\u001b[?25hDownloading torchaudio-2.2.1-cp311-cp311-manylinux1_x86_64.whl (3.3 MB)\n",
- "\u001b[2K \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m3.3/3.3 MB\u001b[0m \u001b[31m3.3 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
- "\u001b[?25hDownloading nvidia_cudnn_cu12-8.9.2.26-py3-none-manylinux1_x86_64.whl (731.7 MB)\n",
- "\u001b[2K \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m731.7/731.7 MB\u001b[0m \u001b[31m2.8 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
- "\u001b[?25hDownloading nvidia_nccl_cu12-2.19.3-py3-none-manylinux1_x86_64.whl (166.0 MB)\n",
- "\u001b[2K \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m166.0/166.0 MB\u001b[0m \u001b[31m6.9 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
- "\u001b[?25hDownloading triton-2.2.0-cp311-cp311-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (167.9 MB)\n",
- "\u001b[2K \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m167.9/167.9 MB\u001b[0m \u001b[31m6.5 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
- "\u001b[?25hInstalling collected packages: triton, nvidia-nccl-cu12, nvidia-cudnn-cu12, torch, torchvision, torchaudio\n",
- " Attempting uninstall: triton\n",
- " Found existing installation: triton 3.1.0\n",
- " Uninstalling triton-3.1.0:\n",
- " Successfully uninstalled triton-3.1.0\n",
- " Attempting uninstall: nvidia-nccl-cu12\n",
- " Found existing installation: nvidia-nccl-cu12 2.21.5\n",
- " Uninstalling nvidia-nccl-cu12-2.21.5:\n",
- " Successfully uninstalled nvidia-nccl-cu12-2.21.5\n",
- " Attempting uninstall: nvidia-cudnn-cu12\n",
- " Found existing installation: nvidia-cudnn-cu12 9.1.0.70\n",
- " Uninstalling nvidia-cudnn-cu12-9.1.0.70:\n",
- " Successfully uninstalled nvidia-cudnn-cu12-9.1.0.70\n",
- " Attempting uninstall: torch\n",
- " Found existing installation: torch 2.5.1+cu121\n",
- " Uninstalling torch-2.5.1+cu121:\n",
- " Successfully uninstalled torch-2.5.1+cu121\n",
- " Attempting uninstall: torchvision\n",
- " Found existing installation: torchvision 0.20.1+cu121\n",
- " Uninstalling torchvision-0.20.1+cu121:\n",
- " Successfully uninstalled torchvision-0.20.1+cu121\n",
- " Attempting uninstall: torchaudio\n",
- " Found existing installation: torchaudio 2.5.1+cu121\n",
- " Uninstalling torchaudio-2.5.1+cu121:\n",
- " Successfully uninstalled torchaudio-2.5.1+cu121\n",
- "Successfully installed nvidia-cudnn-cu12-8.9.2.26 nvidia-nccl-cu12-2.19.3 torch-2.2.1 torchaudio-2.2.1 torchvision-0.17.1 triton-2.2.0\n"
- ]
- },
- {
- "output_type": "display_data",
- "data": {
- "application/vnd.colab-display-data+json": {
- "pip_warning": {
- "packages": [
- "torch",
- "torchaudio",
- "torchgen",
- "torchvision",
- "torio",
- "triton"
- ]
- },
- "id": "c4bc2dd0f38e4626bd1dae8bac2ae342"
- }
- },
- "metadata": {}
- },
- {
- "output_type": "stream",
- "name": "stdout",
- "text": [
- "2.5.1+cu121\n"
- ]
- }
- ],
- "source": [
- " !pip install torch==2.2.1 torchvision==0.17.1 torchaudio==2.2.1\n",
- " print(torch.__version__)"
- ]
- },
- {
- "cell_type": "code",
- "execution_count": null,
- "id": "YUziw_XOfU0Z",
- "metadata": {
- "id": "YUziw_XOfU0Z",
- "colab": {
- "base_uri": "https://localhost:8080/"
- },
- "outputId": "a2d8c3b4-6784-476b-ac27-c7e8244fc44e"
- },
- "outputs": [
- {
- "output_type": "stream",
- "name": "stdout",
- "text": [
- "\u001b[33mWARNING: Package(s) not found: PytorchWildlife\u001b[0m\u001b[33m\n",
- "\u001b[0m"
- ]
- }
- ],
- "source": [
- "!pip show PytorchWildlife"
- ]
- },
- {
- "cell_type": "code",
- "execution_count": null,
- "id": "VTB_vmHhaaGj",
- "metadata": {
- "collapsed": true,
- "id": "VTB_vmHhaaGj",
- "colab": {
- "base_uri": "https://localhost:8080/",
- "height": 1000
- },
- "outputId": "9ab593f1-51d0-4c00-f5e1-d2954529858b"
- },
- "outputs": [
- {
- "output_type": "stream",
- "name": "stdout",
- "text": [
- "Reading package lists... Done\n",
- "Building dependency tree... Done\n",
- "Reading state information... Done\n",
- "python3-distutils is already the newest version (3.10.8-1~22.04).\n",
- "0 upgraded, 0 newly installed, 0 to remove and 54 not upgraded.\n",
- "Collecting PytorchWildlife\n",
- " Downloading PytorchWildlife-1.2.0.tar.gz (62 kB)\n",
- "\u001b[2K \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m62.6/62.6 kB\u001b[0m \u001b[31m5.0 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
- "\u001b[?25h Preparing metadata (setup.py) ... \u001b[?25l\u001b[?25hdone\n",
- "Requirement already satisfied: torch in /usr/local/lib/python3.11/dist-packages (from PytorchWildlife) (2.2.1)\n",
- "Requirement already satisfied: torchvision in /usr/local/lib/python3.11/dist-packages (from PytorchWildlife) (0.17.1)\n",
- "Requirement already satisfied: torchaudio in /usr/local/lib/python3.11/dist-packages (from PytorchWildlife) (2.2.1)\n",
- "Requirement already satisfied: tqdm in /usr/local/lib/python3.11/dist-packages (from PytorchWildlife) (4.67.1)\n",
- "Requirement already satisfied: Pillow in /usr/local/lib/python3.11/dist-packages (from PytorchWildlife) (11.1.0)\n",
- "Collecting supervision==0.23.0 (from PytorchWildlife)\n",
- " Downloading supervision-0.23.0-py3-none-any.whl.metadata (14 kB)\n",
- "Collecting gradio (from PytorchWildlife)\n",
- " Downloading gradio-5.13.1-py3-none-any.whl.metadata (16 kB)\n",
- "Collecting ultralytics-yolov5 (from PytorchWildlife)\n",
- " Downloading ultralytics-yolov5-0.1.1.tar.gz (820 kB)\n",
- "\u001b[2K \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m820.5/820.5 kB\u001b[0m \u001b[31m45.0 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
- "\u001b[?25h Installing build dependencies ... \u001b[?25l\u001b[?25hdone\n",
- " Getting requirements to build wheel ... \u001b[?25l\u001b[?25hdone\n",
- " Preparing metadata (pyproject.toml) ... \u001b[?25l\u001b[?25hdone\n",
- "Collecting ultralytics (from PytorchWildlife)\n",
- " Downloading ultralytics-8.3.68-py3-none-any.whl.metadata (35 kB)\n",
- "Requirement already satisfied: chardet in /usr/local/lib/python3.11/dist-packages (from PytorchWildlife) (5.2.0)\n",
- "Collecting wget (from PytorchWildlife)\n",
- " Downloading wget-3.2.zip (10 kB)\n",
- " Preparing metadata (setup.py) ... \u001b[?25l\u001b[?25hdone\n",
- "Requirement already satisfied: setuptools in /usr/local/lib/python3.11/dist-packages (from PytorchWildlife) (75.1.0)\n",
- "Requirement already satisfied: scikit-learn in /usr/local/lib/python3.11/dist-packages (from PytorchWildlife) (1.6.1)\n",
- "Requirement already satisfied: defusedxml<0.8.0,>=0.7.1 in /usr/local/lib/python3.11/dist-packages (from supervision==0.23.0->PytorchWildlife) (0.7.1)\n",
- "Requirement already satisfied: matplotlib>=3.6.0 in /usr/local/lib/python3.11/dist-packages (from supervision==0.23.0->PytorchWildlife) (3.10.0)\n",
- "Requirement already satisfied: numpy>=1.23.3 in /usr/local/lib/python3.11/dist-packages (from supervision==0.23.0->PytorchWildlife) (1.26.4)\n",
- "Requirement already satisfied: opencv-python-headless>=4.5.5.64 in /usr/local/lib/python3.11/dist-packages (from supervision==0.23.0->PytorchWildlife) (4.11.0.86)\n",
- "Requirement already satisfied: pyyaml>=5.3 in /usr/local/lib/python3.11/dist-packages (from supervision==0.23.0->PytorchWildlife) (6.0.2)\n",
- "Requirement already satisfied: scipy<2.0.0,>=1.10.0 in /usr/local/lib/python3.11/dist-packages (from supervision==0.23.0->PytorchWildlife) (1.13.1)\n",
- "Collecting aiofiles<24.0,>=22.0 (from gradio->PytorchWildlife)\n",
- " Downloading aiofiles-23.2.1-py3-none-any.whl.metadata (9.7 kB)\n",
- "Requirement already satisfied: anyio<5.0,>=3.0 in /usr/local/lib/python3.11/dist-packages (from gradio->PytorchWildlife) (3.7.1)\n",
- "Collecting fastapi<1.0,>=0.115.2 (from gradio->PytorchWildlife)\n",
- " Downloading fastapi-0.115.7-py3-none-any.whl.metadata (27 kB)\n",
- "Collecting ffmpy (from gradio->PytorchWildlife)\n",
- " Downloading ffmpy-0.5.0-py3-none-any.whl.metadata (3.0 kB)\n",
- "Collecting gradio-client==1.6.0 (from gradio->PytorchWildlife)\n",
- " Downloading gradio_client-1.6.0-py3-none-any.whl.metadata (7.1 kB)\n",
- "Requirement already satisfied: httpx>=0.24.1 in /usr/local/lib/python3.11/dist-packages (from gradio->PytorchWildlife) (0.28.1)\n",
- "Requirement already satisfied: huggingface-hub>=0.25.1 in /usr/local/lib/python3.11/dist-packages (from gradio->PytorchWildlife) (0.27.1)\n",
- "Requirement already satisfied: jinja2<4.0 in /usr/local/lib/python3.11/dist-packages (from gradio->PytorchWildlife) (3.1.5)\n",
- "Collecting markupsafe~=2.0 (from gradio->PytorchWildlife)\n",
- " Downloading MarkupSafe-2.1.5-cp311-cp311-manylinux_2_17_x86_64.manylinux2014_x86_64.whl.metadata (3.0 kB)\n",
- "Requirement already satisfied: orjson~=3.0 in /usr/local/lib/python3.11/dist-packages (from gradio->PytorchWildlife) (3.10.15)\n",
- "Requirement already satisfied: packaging in /usr/local/lib/python3.11/dist-packages (from gradio->PytorchWildlife) (24.2)\n",
- "Requirement already satisfied: pandas<3.0,>=1.0 in /usr/local/lib/python3.11/dist-packages (from gradio->PytorchWildlife) (2.2.2)\n",
- "Requirement already satisfied: pydantic>=2.0 in /usr/local/lib/python3.11/dist-packages (from gradio->PytorchWildlife) (2.10.5)\n",
- "Collecting pydub (from gradio->PytorchWildlife)\n",
- " Downloading pydub-0.25.1-py2.py3-none-any.whl.metadata (1.4 kB)\n",
- "Collecting python-multipart>=0.0.18 (from gradio->PytorchWildlife)\n",
- " Downloading python_multipart-0.0.20-py3-none-any.whl.metadata (1.8 kB)\n",
- "Collecting ruff>=0.2.2 (from gradio->PytorchWildlife)\n",
- " Downloading ruff-0.9.3-py3-none-manylinux_2_17_x86_64.manylinux2014_x86_64.whl.metadata (25 kB)\n",
- "Collecting safehttpx<0.2.0,>=0.1.6 (from gradio->PytorchWildlife)\n",
- " Downloading safehttpx-0.1.6-py3-none-any.whl.metadata (4.2 kB)\n",
- "Collecting semantic-version~=2.0 (from gradio->PytorchWildlife)\n",
- " Downloading semantic_version-2.10.0-py2.py3-none-any.whl.metadata (9.7 kB)\n",
- "Collecting starlette<1.0,>=0.40.0 (from gradio->PytorchWildlife)\n",
- " Downloading starlette-0.45.3-py3-none-any.whl.metadata (6.3 kB)\n",
- "Collecting tomlkit<0.14.0,>=0.12.0 (from gradio->PytorchWildlife)\n",
- " Downloading tomlkit-0.13.2-py3-none-any.whl.metadata (2.7 kB)\n",
- "Requirement already satisfied: typer<1.0,>=0.12 in /usr/local/lib/python3.11/dist-packages (from gradio->PytorchWildlife) (0.15.1)\n",
- "Requirement already satisfied: typing-extensions~=4.0 in /usr/local/lib/python3.11/dist-packages (from gradio->PytorchWildlife) (4.12.2)\n",
- "Collecting uvicorn>=0.14.0 (from gradio->PytorchWildlife)\n",
- " Downloading uvicorn-0.34.0-py3-none-any.whl.metadata (6.5 kB)\n",
- "Requirement already satisfied: fsspec in /usr/local/lib/python3.11/dist-packages (from gradio-client==1.6.0->gradio->PytorchWildlife) (2024.10.0)\n",
- "Requirement already satisfied: websockets<15.0,>=10.0 in /usr/local/lib/python3.11/dist-packages (from gradio-client==1.6.0->gradio->PytorchWildlife) (14.2)\n",
- "Requirement already satisfied: joblib>=1.2.0 in /usr/local/lib/python3.11/dist-packages (from scikit-learn->PytorchWildlife) (1.4.2)\n",
- "Requirement already satisfied: threadpoolctl>=3.1.0 in /usr/local/lib/python3.11/dist-packages (from scikit-learn->PytorchWildlife) (3.5.0)\n",
- "Requirement already satisfied: filelock in /usr/local/lib/python3.11/dist-packages (from torch->PytorchWildlife) (3.17.0)\n",
- "Requirement already satisfied: sympy in /usr/local/lib/python3.11/dist-packages (from torch->PytorchWildlife) (1.13.1)\n",
- "Requirement already satisfied: networkx in /usr/local/lib/python3.11/dist-packages (from torch->PytorchWildlife) (3.4.2)\n",
- "Requirement already satisfied: nvidia-cuda-nvrtc-cu12==12.1.105 in /usr/local/lib/python3.11/dist-packages (from torch->PytorchWildlife) (12.1.105)\n",
- "Requirement already satisfied: nvidia-cuda-runtime-cu12==12.1.105 in /usr/local/lib/python3.11/dist-packages (from torch->PytorchWildlife) (12.1.105)\n",
- "Requirement already satisfied: nvidia-cuda-cupti-cu12==12.1.105 in /usr/local/lib/python3.11/dist-packages (from torch->PytorchWildlife) (12.1.105)\n",
- "Requirement already satisfied: nvidia-cudnn-cu12==8.9.2.26 in /usr/local/lib/python3.11/dist-packages (from torch->PytorchWildlife) (8.9.2.26)\n",
- "Requirement already satisfied: nvidia-cublas-cu12==12.1.3.1 in /usr/local/lib/python3.11/dist-packages (from torch->PytorchWildlife) (12.1.3.1)\n",
- "Requirement already satisfied: nvidia-cufft-cu12==11.0.2.54 in /usr/local/lib/python3.11/dist-packages (from torch->PytorchWildlife) (11.0.2.54)\n",
- "Requirement already satisfied: nvidia-curand-cu12==10.3.2.106 in /usr/local/lib/python3.11/dist-packages (from torch->PytorchWildlife) (10.3.2.106)\n",
- "Requirement already satisfied: nvidia-cusolver-cu12==11.4.5.107 in /usr/local/lib/python3.11/dist-packages (from torch->PytorchWildlife) (11.4.5.107)\n",
- "Requirement already satisfied: nvidia-cusparse-cu12==12.1.0.106 in /usr/local/lib/python3.11/dist-packages (from torch->PytorchWildlife) (12.1.0.106)\n",
- "Requirement already satisfied: nvidia-nccl-cu12==2.19.3 in /usr/local/lib/python3.11/dist-packages (from torch->PytorchWildlife) (2.19.3)\n",
- "Requirement already satisfied: nvidia-nvtx-cu12==12.1.105 in /usr/local/lib/python3.11/dist-packages (from torch->PytorchWildlife) (12.1.105)\n",
- "Requirement already satisfied: triton==2.2.0 in /usr/local/lib/python3.11/dist-packages (from torch->PytorchWildlife) (2.2.0)\n",
- "Requirement already satisfied: nvidia-nvjitlink-cu12 in /usr/local/lib/python3.11/dist-packages (from nvidia-cusolver-cu12==11.4.5.107->torch->PytorchWildlife) (12.6.85)\n",
- "Requirement already satisfied: opencv-python>=4.6.0 in /usr/local/lib/python3.11/dist-packages (from ultralytics->PytorchWildlife) (4.10.0.84)\n",
- "Requirement already satisfied: requests>=2.23.0 in /usr/local/lib/python3.11/dist-packages (from ultralytics->PytorchWildlife) (2.32.3)\n",
- "Requirement already satisfied: psutil in /usr/local/lib/python3.11/dist-packages (from ultralytics->PytorchWildlife) (5.9.5)\n",
- "Requirement already satisfied: py-cpuinfo in /usr/local/lib/python3.11/dist-packages (from ultralytics->PytorchWildlife) (9.0.0)\n",
- "Requirement already satisfied: seaborn>=0.11.0 in /usr/local/lib/python3.11/dist-packages (from ultralytics->PytorchWildlife) (0.13.2)\n",
- "Collecting ultralytics-thop>=2.0.0 (from ultralytics->PytorchWildlife)\n",
- " Downloading ultralytics_thop-2.0.14-py3-none-any.whl.metadata (9.4 kB)\n",
- "Collecting protobuf<=3.20.1 (from ultralytics-yolov5->PytorchWildlife)\n",
- " Downloading protobuf-3.20.1-py2.py3-none-any.whl.metadata (720 bytes)\n",
- "Collecting fire (from ultralytics-yolov5->PytorchWildlife)\n",
- " Downloading fire-0.7.0.tar.gz (87 kB)\n",
- "\u001b[2K \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m87.2/87.2 kB\u001b[0m \u001b[31m8.8 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
- "\u001b[?25h Preparing metadata (setup.py) ... \u001b[?25l\u001b[?25hdone\n",
- "Requirement already satisfied: tensorboard>=2.4.1 in /usr/local/lib/python3.11/dist-packages (from ultralytics-yolov5->PytorchWildlife) (2.17.1)\n",
- "Requirement already satisfied: ipython in /usr/local/lib/python3.11/dist-packages (from ultralytics-yolov5->PytorchWildlife) (7.34.0)\n",
- "Collecting thop (from ultralytics-yolov5->PytorchWildlife)\n",
- " Downloading thop-0.1.1.post2209072238-py3-none-any.whl.metadata (2.7 kB)\n",
- "Requirement already satisfied: idna>=2.8 in /usr/local/lib/python3.11/dist-packages (from anyio<5.0,>=3.0->gradio->PytorchWildlife) (3.10)\n",
- "Requirement already satisfied: sniffio>=1.1 in /usr/local/lib/python3.11/dist-packages (from anyio<5.0,>=3.0->gradio->PytorchWildlife) (1.3.1)\n",
- "Requirement already satisfied: certifi in /usr/local/lib/python3.11/dist-packages (from httpx>=0.24.1->gradio->PytorchWildlife) (2024.12.14)\n",
- "Requirement already satisfied: httpcore==1.* in /usr/local/lib/python3.11/dist-packages (from httpx>=0.24.1->gradio->PytorchWildlife) (1.0.7)\n",
- "Requirement already satisfied: h11<0.15,>=0.13 in /usr/local/lib/python3.11/dist-packages (from httpcore==1.*->httpx>=0.24.1->gradio->PytorchWildlife) (0.14.0)\n",
- "Requirement already satisfied: contourpy>=1.0.1 in /usr/local/lib/python3.11/dist-packages (from matplotlib>=3.6.0->supervision==0.23.0->PytorchWildlife) (1.3.1)\n",
- "Requirement already satisfied: cycler>=0.10 in /usr/local/lib/python3.11/dist-packages (from matplotlib>=3.6.0->supervision==0.23.0->PytorchWildlife) (0.12.1)\n",
- "Requirement already satisfied: fonttools>=4.22.0 in /usr/local/lib/python3.11/dist-packages (from matplotlib>=3.6.0->supervision==0.23.0->PytorchWildlife) (4.55.5)\n",
- "Requirement already satisfied: kiwisolver>=1.3.1 in /usr/local/lib/python3.11/dist-packages (from matplotlib>=3.6.0->supervision==0.23.0->PytorchWildlife) (1.4.8)\n",
- "Requirement already satisfied: pyparsing>=2.3.1 in /usr/local/lib/python3.11/dist-packages (from matplotlib>=3.6.0->supervision==0.23.0->PytorchWildlife) (3.2.1)\n",
- "Requirement already satisfied: python-dateutil>=2.7 in /usr/local/lib/python3.11/dist-packages (from matplotlib>=3.6.0->supervision==0.23.0->PytorchWildlife) (2.8.2)\n",
- "Requirement already satisfied: pytz>=2020.1 in /usr/local/lib/python3.11/dist-packages (from pandas<3.0,>=1.0->gradio->PytorchWildlife) (2024.2)\n",
- "Requirement already satisfied: tzdata>=2022.7 in /usr/local/lib/python3.11/dist-packages (from pandas<3.0,>=1.0->gradio->PytorchWildlife) (2025.1)\n",
- "Requirement already satisfied: annotated-types>=0.6.0 in /usr/local/lib/python3.11/dist-packages (from pydantic>=2.0->gradio->PytorchWildlife) (0.7.0)\n",
- "Requirement already satisfied: pydantic-core==2.27.2 in /usr/local/lib/python3.11/dist-packages (from pydantic>=2.0->gradio->PytorchWildlife) (2.27.2)\n",
- "Requirement already satisfied: charset-normalizer<4,>=2 in /usr/local/lib/python3.11/dist-packages (from requests>=2.23.0->ultralytics->PytorchWildlife) (3.4.1)\n",
- "Requirement already satisfied: urllib3<3,>=1.21.1 in /usr/local/lib/python3.11/dist-packages (from requests>=2.23.0->ultralytics->PytorchWildlife) (2.3.0)\n",
- "Requirement already satisfied: absl-py>=0.4 in /usr/local/lib/python3.11/dist-packages (from tensorboard>=2.4.1->ultralytics-yolov5->PytorchWildlife) (1.4.0)\n",
- "Requirement already satisfied: grpcio>=1.48.2 in /usr/local/lib/python3.11/dist-packages (from tensorboard>=2.4.1->ultralytics-yolov5->PytorchWildlife) (1.69.0)\n",
- "Requirement already satisfied: markdown>=2.6.8 in /usr/local/lib/python3.11/dist-packages (from tensorboard>=2.4.1->ultralytics-yolov5->PytorchWildlife) (3.7)\n",
- "Requirement already satisfied: six>1.9 in /usr/local/lib/python3.11/dist-packages (from tensorboard>=2.4.1->ultralytics-yolov5->PytorchWildlife) (1.17.0)\n",
- "Requirement already satisfied: tensorboard-data-server<0.8.0,>=0.7.0 in /usr/local/lib/python3.11/dist-packages (from tensorboard>=2.4.1->ultralytics-yolov5->PytorchWildlife) (0.7.2)\n",
- "Requirement already satisfied: werkzeug>=1.0.1 in /usr/local/lib/python3.11/dist-packages (from tensorboard>=2.4.1->ultralytics-yolov5->PytorchWildlife) (3.1.3)\n",
- "Requirement already satisfied: click>=8.0.0 in /usr/local/lib/python3.11/dist-packages (from typer<1.0,>=0.12->gradio->PytorchWildlife) (8.1.8)\n",
- "Requirement already satisfied: shellingham>=1.3.0 in /usr/local/lib/python3.11/dist-packages (from typer<1.0,>=0.12->gradio->PytorchWildlife) (1.5.4)\n",
- "Requirement already satisfied: rich>=10.11.0 in /usr/local/lib/python3.11/dist-packages (from typer<1.0,>=0.12->gradio->PytorchWildlife) (13.9.4)\n",
- "Requirement already satisfied: termcolor in /usr/local/lib/python3.11/dist-packages (from fire->ultralytics-yolov5->PytorchWildlife) (2.5.0)\n",
- "Collecting jedi>=0.16 (from ipython->ultralytics-yolov5->PytorchWildlife)\n",
- " Downloading jedi-0.19.2-py2.py3-none-any.whl.metadata (22 kB)\n",
- "Requirement already satisfied: decorator in /usr/local/lib/python3.11/dist-packages (from ipython->ultralytics-yolov5->PytorchWildlife) (4.4.2)\n",
- "Requirement already satisfied: pickleshare in /usr/local/lib/python3.11/dist-packages (from ipython->ultralytics-yolov5->PytorchWildlife) (0.7.5)\n",
- "Requirement already satisfied: traitlets>=4.2 in /usr/local/lib/python3.11/dist-packages (from ipython->ultralytics-yolov5->PytorchWildlife) (5.7.1)\n",
- "Requirement already satisfied: prompt-toolkit!=3.0.0,!=3.0.1,<3.1.0,>=2.0.0 in /usr/local/lib/python3.11/dist-packages (from ipython->ultralytics-yolov5->PytorchWildlife) (3.0.50)\n",
- "Requirement already satisfied: pygments in /usr/local/lib/python3.11/dist-packages (from ipython->ultralytics-yolov5->PytorchWildlife) (2.18.0)\n",
- "Requirement already satisfied: backcall in /usr/local/lib/python3.11/dist-packages (from ipython->ultralytics-yolov5->PytorchWildlife) (0.2.0)\n",
- "Requirement already satisfied: matplotlib-inline in /usr/local/lib/python3.11/dist-packages (from ipython->ultralytics-yolov5->PytorchWildlife) (0.1.7)\n",
- "Requirement already satisfied: pexpect>4.3 in /usr/local/lib/python3.11/dist-packages (from ipython->ultralytics-yolov5->PytorchWildlife) (4.9.0)\n",
- "Requirement already satisfied: mpmath<1.4,>=1.1.0 in /usr/local/lib/python3.11/dist-packages (from sympy->torch->PytorchWildlife) (1.3.0)\n",
- "Requirement already satisfied: parso<0.9.0,>=0.8.4 in /usr/local/lib/python3.11/dist-packages (from jedi>=0.16->ipython->ultralytics-yolov5->PytorchWildlife) (0.8.4)\n",
- "Requirement already satisfied: ptyprocess>=0.5 in /usr/local/lib/python3.11/dist-packages (from pexpect>4.3->ipython->ultralytics-yolov5->PytorchWildlife) (0.7.0)\n",
- "Requirement already satisfied: wcwidth in /usr/local/lib/python3.11/dist-packages (from prompt-toolkit!=3.0.0,!=3.0.1,<3.1.0,>=2.0.0->ipython->ultralytics-yolov5->PytorchWildlife) (0.2.13)\n",
- "Requirement already satisfied: markdown-it-py>=2.2.0 in /usr/local/lib/python3.11/dist-packages (from rich>=10.11.0->typer<1.0,>=0.12->gradio->PytorchWildlife) (3.0.0)\n",
- "Requirement already satisfied: mdurl~=0.1 in /usr/local/lib/python3.11/dist-packages (from markdown-it-py>=2.2.0->rich>=10.11.0->typer<1.0,>=0.12->gradio->PytorchWildlife) (0.1.2)\n",
- "Downloading supervision-0.23.0-py3-none-any.whl (151 kB)\n",
- "\u001b[2K \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m151.5/151.5 kB\u001b[0m \u001b[31m13.8 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
- "\u001b[?25hDownloading gradio-5.13.1-py3-none-any.whl (57.6 MB)\n",
- "\u001b[2K \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m57.6/57.6 MB\u001b[0m \u001b[31m12.0 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
- "\u001b[?25hDownloading gradio_client-1.6.0-py3-none-any.whl (321 kB)\n",
- "\u001b[2K \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m321.8/321.8 kB\u001b[0m \u001b[31m22.4 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
- "\u001b[?25hDownloading ultralytics-8.3.68-py3-none-any.whl (913 kB)\n",
- "\u001b[2K \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m913.6/913.6 kB\u001b[0m \u001b[31m50.9 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
- "\u001b[?25hDownloading aiofiles-23.2.1-py3-none-any.whl (15 kB)\n",
- "Downloading fastapi-0.115.7-py3-none-any.whl (94 kB)\n",
- "\u001b[2K \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m94.8/94.8 kB\u001b[0m \u001b[31m8.6 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
- "\u001b[?25hDownloading MarkupSafe-2.1.5-cp311-cp311-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (28 kB)\n",
- "Downloading protobuf-3.20.1-py2.py3-none-any.whl (162 kB)\n",
- "\u001b[2K \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m162.1/162.1 kB\u001b[0m \u001b[31m15.4 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
- "\u001b[?25hDownloading python_multipart-0.0.20-py3-none-any.whl (24 kB)\n",
- "Downloading ruff-0.9.3-py3-none-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (12.4 MB)\n",
- "\u001b[2K \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m12.4/12.4 MB\u001b[0m \u001b[31m70.5 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
- "\u001b[?25hDownloading safehttpx-0.1.6-py3-none-any.whl (8.7 kB)\n",
- "Downloading semantic_version-2.10.0-py2.py3-none-any.whl (15 kB)\n",
- "Downloading starlette-0.45.3-py3-none-any.whl (71 kB)\n",
- "\u001b[2K \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m71.5/71.5 kB\u001b[0m \u001b[31m7.2 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
- "\u001b[?25hDownloading tomlkit-0.13.2-py3-none-any.whl (37 kB)\n",
- "Downloading ultralytics_thop-2.0.14-py3-none-any.whl (26 kB)\n",
- "Downloading uvicorn-0.34.0-py3-none-any.whl (62 kB)\n",
- "\u001b[2K \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m62.3/62.3 kB\u001b[0m \u001b[31m6.5 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
- "\u001b[?25hDownloading ffmpy-0.5.0-py3-none-any.whl (6.0 kB)\n",
- "Downloading pydub-0.25.1-py2.py3-none-any.whl (32 kB)\n",
- "Downloading thop-0.1.1.post2209072238-py3-none-any.whl (15 kB)\n",
- "Downloading jedi-0.19.2-py2.py3-none-any.whl (1.6 MB)\n",
- "\u001b[2K \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m1.6/1.6 MB\u001b[0m \u001b[31m59.4 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
- "\u001b[?25hBuilding wheels for collected packages: PytorchWildlife, ultralytics-yolov5, wget, fire\n",
- " Building wheel for PytorchWildlife (setup.py) ... \u001b[?25l\u001b[?25hdone\n",
- " Created wheel for PytorchWildlife: filename=PytorchWildlife-1.2.0-py3-none-any.whl size=61205 sha256=577066db4bab170fe39b2be823d341786cfc723632496fa015b8514eb9e60a20\n",
- " Stored in directory: /root/.cache/pip/wheels/40/07/6b/c1aba410c6ff0ce5c1743b48e9e50075c14c43b2653c0f620e\n",
- " Building wheel for ultralytics-yolov5 (pyproject.toml) ... \u001b[?25l\u001b[?25hdone\n",
- " Created wheel for ultralytics-yolov5: filename=ultralytics_yolov5-0.1.1-py3-none-any.whl size=861719 sha256=fad57d2bbcf5f3db6c153cc6fb19dfceeb48fac1f46c5f1347742ab3103c21c6\n",
- " Stored in directory: /root/.cache/pip/wheels/5a/cc/de/9cf3e09352897d22b767f47d4ba35049fed1e7b3404b91c0dd\n",
- " Building wheel for wget (setup.py) ... \u001b[?25l\u001b[?25hdone\n",
- " Created wheel for wget: filename=wget-3.2-py3-none-any.whl size=9656 sha256=6867dc3459705147a84192d1f2828c747d058e5a8af446280d9f9ca01bebf8ba\n",
- " Stored in directory: /root/.cache/pip/wheels/40/b3/0f/a40dbd1c6861731779f62cc4babcb234387e11d697df70ee97\n",
- " Building wheel for fire (setup.py) ... \u001b[?25l\u001b[?25hdone\n",
- " Created wheel for fire: filename=fire-0.7.0-py3-none-any.whl size=114249 sha256=3efeec798f4f3f4035794c71223ec6d2221210df972f6c50fe54092a46a0e23f\n",
- " Stored in directory: /root/.cache/pip/wheels/46/54/24/1624fd5b8674eb1188623f7e8e17cdf7c0f6c24b609dfb8a89\n",
- "Successfully built PytorchWildlife ultralytics-yolov5 wget fire\n",
- "Installing collected packages: wget, pydub, uvicorn, tomlkit, semantic-version, ruff, python-multipart, protobuf, markupsafe, jedi, fire, ffmpy, aiofiles, starlette, supervision, safehttpx, gradio-client, fastapi, ultralytics-thop, thop, gradio, ultralytics-yolov5, ultralytics, PytorchWildlife\n",
- " Attempting uninstall: protobuf\n",
- " Found existing installation: protobuf 4.25.5\n",
- " Uninstalling protobuf-4.25.5:\n",
- " Successfully uninstalled protobuf-4.25.5\n",
- " Attempting uninstall: markupsafe\n",
- " Found existing installation: MarkupSafe 3.0.2\n",
- " Uninstalling MarkupSafe-3.0.2:\n",
- " Successfully uninstalled MarkupSafe-3.0.2\n",
- "\u001b[31mERROR: pip's dependency resolver does not currently take into account all the packages that are installed. This behaviour is the source of the following dependency conflicts.\n",
- "google-cloud-pubsub 2.25.0 requires protobuf!=4.21.0,!=4.21.1,!=4.21.2,!=4.21.3,!=4.21.4,!=4.21.5,<6.0.0dev,>=3.20.2, but you have protobuf 3.20.1 which is incompatible.\n",
- "google-cloud-bigtable 2.28.1 requires protobuf!=4.21.0,!=4.21.1,!=4.21.2,!=4.21.3,!=4.21.4,!=4.21.5,<6.0.0dev,>=3.20.2, but you have protobuf 3.20.1 which is incompatible.\n",
- "google-cloud-aiplatform 1.74.0 requires protobuf!=4.21.0,!=4.21.1,!=4.21.2,!=4.21.3,!=4.21.4,!=4.21.5,<6.0.0dev,>=3.20.2, but you have protobuf 3.20.1 which is incompatible.\n",
- "google-cloud-bigquery-connection 1.17.0 requires protobuf!=4.21.0,!=4.21.1,!=4.21.2,!=4.21.3,!=4.21.4,!=4.21.5,<6.0.0dev,>=3.20.2, but you have protobuf 3.20.1 which is incompatible.\n",
- "google-cloud-resource-manager 1.14.0 requires protobuf!=4.21.0,!=4.21.1,!=4.21.2,!=4.21.3,!=4.21.4,!=4.21.5,<6.0.0dev,>=3.20.2, but you have protobuf 3.20.1 which is incompatible.\n",
- "google-cloud-bigquery-storage 2.27.0 requires 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,<6.0.0dev,>=3.20.2, but you have protobuf 3.20.1 which is incompatible.\n",
- "tensorflow-metadata 1.16.1 requires protobuf<6.0.0dev,>=4.25.2; python_version >= \"3.11\", but you have protobuf 3.20.1 which is incompatible.\n",
- "google-cloud-translate 3.19.0 requires protobuf!=4.21.0,!=4.21.1,!=4.21.2,!=4.21.3,!=4.21.4,!=4.21.5,<6.0.0dev,>=3.20.2, but you have protobuf 3.20.1 which is incompatible.\n",
- "google-cloud-firestore 2.19.0 requires 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,<6.0.0dev,>=3.20.2, but you have protobuf 3.20.1 which is incompatible.\n",
- "google-cloud-functions 1.19.0 requires protobuf!=4.21.0,!=4.21.1,!=4.21.2,!=4.21.3,!=4.21.4,!=4.21.5,<6.0.0dev,>=3.20.2, but you have protobuf 3.20.1 which is incompatible.\n",
- "googleapis-common-protos 1.66.0 requires protobuf!=3.20.0,!=3.20.1,!=4.21.1,!=4.21.2,!=4.21.3,!=4.21.4,!=4.21.5,<6.0.0.dev0,>=3.20.2, but you have protobuf 3.20.1 which is incompatible.\n",
- "grpcio-status 1.62.3 requires protobuf>=4.21.6, but you have protobuf 3.20.1 which is incompatible.\n",
- "google-cloud-language 2.16.0 requires protobuf!=4.21.0,!=4.21.1,!=4.21.2,!=4.21.3,!=4.21.4,!=4.21.5,<6.0.0dev,>=3.20.2, but you have protobuf 3.20.1 which is incompatible.\n",
- "google-ai-generativelanguage 0.6.15 requires protobuf!=4.21.0,!=4.21.1,!=4.21.2,!=4.21.3,!=4.21.4,!=4.21.5,<6.0.0dev,>=3.20.2, but you have protobuf 3.20.1 which is incompatible.\n",
- "google-cloud-iam 2.17.0 requires protobuf!=4.21.0,!=4.21.1,!=4.21.2,!=4.21.3,!=4.21.4,!=4.21.5,<6.0.0dev,>=3.20.2, but you have protobuf 3.20.1 which is incompatible.\n",
- "google-cloud-datastore 2.20.2 requires 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,<6.0.0dev,>=3.20.2, but you have protobuf 3.20.1 which is incompatible.\n",
- "google-api-core 2.19.2 requires 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,<6.0.0.dev0,>=3.19.5, but you have protobuf 3.20.1 which is incompatible.\n",
- "tensorflow 2.17.1 requires protobuf!=4.21.0,!=4.21.1,!=4.21.2,!=4.21.3,!=4.21.4,!=4.21.5,<5.0.0dev,>=3.20.3, but you have protobuf 3.20.1 which is incompatible.\n",
- "grpc-google-iam-v1 0.14.0 requires protobuf!=4.21.1,!=4.21.2,!=4.21.3,!=4.21.4,!=4.21.5,<6.0.0dev,>=3.20.2, but you have protobuf 3.20.1 which is incompatible.\u001b[0m\u001b[31m\n",
- "\u001b[0mSuccessfully installed PytorchWildlife-1.2.0 aiofiles-23.2.1 fastapi-0.115.7 ffmpy-0.5.0 fire-0.7.0 gradio-5.13.1 gradio-client-1.6.0 jedi-0.19.2 markupsafe-2.1.5 protobuf-3.20.1 pydub-0.25.1 python-multipart-0.0.20 ruff-0.9.3 safehttpx-0.1.6 semantic-version-2.10.0 starlette-0.45.3 supervision-0.23.0 thop-0.1.1.post2209072238 tomlkit-0.13.2 ultralytics-8.3.68 ultralytics-thop-2.0.14 ultralytics-yolov5-0.1.1 uvicorn-0.34.0 wget-3.2\n"
- ]
- },
- {
- "output_type": "display_data",
- "data": {
- "application/vnd.colab-display-data+json": {
- "pip_warning": {
- "packages": [
- "google"
- ]
- },
- "id": "bd77490f30e34da7984c182933e8a72f"
- }
- },
- "metadata": {}
- }
- ],
- "source": [
- " #restart 1\n",
- " !sudo apt-get install python3-distutils\n",
- " !pip install PytorchWildlife"
- ]
- },
- {
- "cell_type": "code",
- "execution_count": null,
- "id": "LdcriN_jp6B4",
- "metadata": {
- "id": "LdcriN_jp6B4",
- "colab": {
- "base_uri": "https://localhost:8080/",
- "height": 992
- },
- "outputId": "c2f2b64f-803b-488f-c752-46b28c80f296"
- },
- "outputs": [
- {
- "output_type": "stream",
- "name": "stdout",
- "text": [
- "Requirement already satisfied: ipython in /usr/local/lib/python3.11/dist-packages (7.34.0)\n",
- "Collecting ipython\n",
- " Downloading ipython-8.31.0-py3-none-any.whl.metadata (4.9 kB)\n",
- "Requirement already satisfied: decorator in /usr/local/lib/python3.11/dist-packages (from ipython) (4.4.2)\n",
- "Requirement already satisfied: jedi>=0.16 in /usr/local/lib/python3.11/dist-packages (from ipython) (0.19.2)\n",
- "Requirement already satisfied: matplotlib-inline in /usr/local/lib/python3.11/dist-packages (from ipython) (0.1.7)\n",
- "Requirement already satisfied: pexpect>4.3 in /usr/local/lib/python3.11/dist-packages (from ipython) (4.9.0)\n",
- "Requirement already satisfied: prompt_toolkit<3.1.0,>=3.0.41 in /usr/local/lib/python3.11/dist-packages (from ipython) (3.0.50)\n",
- "Requirement already satisfied: pygments>=2.4.0 in /usr/local/lib/python3.11/dist-packages (from ipython) (2.18.0)\n",
- "Collecting stack_data (from ipython)\n",
- " Downloading stack_data-0.6.3-py3-none-any.whl.metadata (18 kB)\n",
- "Collecting traitlets>=5.13.0 (from ipython)\n",
- " Downloading traitlets-5.14.3-py3-none-any.whl.metadata (10 kB)\n",
- "Requirement already satisfied: typing_extensions>=4.6 in /usr/local/lib/python3.11/dist-packages (from ipython) (4.12.2)\n",
- "Requirement already satisfied: parso<0.9.0,>=0.8.4 in /usr/local/lib/python3.11/dist-packages (from jedi>=0.16->ipython) (0.8.4)\n",
- "Requirement already satisfied: ptyprocess>=0.5 in /usr/local/lib/python3.11/dist-packages (from pexpect>4.3->ipython) (0.7.0)\n",
- "Requirement already satisfied: wcwidth in /usr/local/lib/python3.11/dist-packages (from prompt_toolkit<3.1.0,>=3.0.41->ipython) (0.2.13)\n",
- "Collecting executing>=1.2.0 (from stack_data->ipython)\n",
- " Downloading executing-2.2.0-py2.py3-none-any.whl.metadata (8.9 kB)\n",
- "Collecting asttokens>=2.1.0 (from stack_data->ipython)\n",
- " Downloading asttokens-3.0.0-py3-none-any.whl.metadata (4.7 kB)\n",
- "Collecting pure-eval (from stack_data->ipython)\n",
- " Downloading pure_eval-0.2.3-py3-none-any.whl.metadata (6.3 kB)\n",
- "Downloading ipython-8.31.0-py3-none-any.whl (821 kB)\n",
- "\u001b[2K \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m821.6/821.6 kB\u001b[0m \u001b[31m39.7 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
- "\u001b[?25hDownloading traitlets-5.14.3-py3-none-any.whl (85 kB)\n",
- "\u001b[2K \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m85.4/85.4 kB\u001b[0m \u001b[31m8.7 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
- "\u001b[?25hDownloading stack_data-0.6.3-py3-none-any.whl (24 kB)\n",
- "Downloading asttokens-3.0.0-py3-none-any.whl (26 kB)\n",
- "Downloading executing-2.2.0-py2.py3-none-any.whl (26 kB)\n",
- "Downloading pure_eval-0.2.3-py3-none-any.whl (11 kB)\n",
- "Installing collected packages: pure-eval, traitlets, executing, asttokens, stack_data, ipython\n",
- " Attempting uninstall: traitlets\n",
- " Found existing installation: traitlets 5.7.1\n",
- " Uninstalling traitlets-5.7.1:\n",
- " Successfully uninstalled traitlets-5.7.1\n",
- " Attempting uninstall: ipython\n",
- " Found existing installation: ipython 7.34.0\n",
- " Uninstalling ipython-7.34.0:\n",
- " Successfully uninstalled ipython-7.34.0\n",
- "\u001b[31mERROR: pip's dependency resolver does not currently take into account all the packages that are installed. This behaviour is the source of the following dependency conflicts.\n",
- "google-colab 1.0.0 requires ipython==7.34.0, but you have ipython 8.31.0 which is incompatible.\u001b[0m\u001b[31m\n",
- "\u001b[0mSuccessfully installed asttokens-3.0.0 executing-2.2.0 ipython-8.31.0 pure-eval-0.2.3 stack_data-0.6.3 traitlets-5.14.3\n"
- ]
- },
- {
- "output_type": "display_data",
- "data": {
- "application/vnd.colab-display-data+json": {
- "pip_warning": {
- "packages": [
- "IPython"
- ]
- },
- "id": "a4e08109536e46b1bc8347066bfbfc28"
- }
- },
- "metadata": {}
- }
- ],
- "source": [
- "!pip install -U ipython"
- ]
- },
- {
- "cell_type": "code",
- "execution_count": null,
- "id": "NJt-E0I63pYe",
- "metadata": {
- "id": "NJt-E0I63pYe",
- "colab": {
- "base_uri": "https://localhost:8080/",
- "height": 1000
- },
- "outputId": "73c256fc-f645-4025-fec8-adf72621e6ec"
- },
- "outputs": [
- {
- "output_type": "stream",
- "name": "stdout",
- "text": [
- "Requirement already satisfied: pytorchWildlife in /usr/local/lib/python3.11/dist-packages (1.2.0)\n",
- "Requirement already satisfied: torch in /usr/local/lib/python3.11/dist-packages (from pytorchWildlife) (2.2.1)\n",
- "Requirement already satisfied: torchvision in /usr/local/lib/python3.11/dist-packages (from pytorchWildlife) (0.17.1)\n",
- "Requirement already satisfied: torchaudio in /usr/local/lib/python3.11/dist-packages (from pytorchWildlife) (2.2.1)\n",
- "Requirement already satisfied: tqdm in /usr/local/lib/python3.11/dist-packages (from pytorchWildlife) (4.67.1)\n",
- "Requirement already satisfied: Pillow in /usr/local/lib/python3.11/dist-packages (from pytorchWildlife) (11.1.0)\n",
- "Requirement already satisfied: supervision==0.23.0 in /usr/local/lib/python3.11/dist-packages (from pytorchWildlife) (0.23.0)\n",
- "Requirement already satisfied: gradio in /usr/local/lib/python3.11/dist-packages (from pytorchWildlife) (5.13.1)\n",
- "Requirement already satisfied: ultralytics-yolov5 in /usr/local/lib/python3.11/dist-packages (from pytorchWildlife) (0.1.1)\n",
- "Requirement already satisfied: ultralytics in /usr/local/lib/python3.11/dist-packages (from pytorchWildlife) (8.3.68)\n",
- "Requirement already satisfied: chardet in /usr/local/lib/python3.11/dist-packages (from pytorchWildlife) (5.2.0)\n",
- "Requirement already satisfied: wget in /usr/local/lib/python3.11/dist-packages (from pytorchWildlife) (3.2)\n",
- "Requirement already satisfied: setuptools in /usr/local/lib/python3.11/dist-packages (from pytorchWildlife) (75.1.0)\n",
- "Requirement already satisfied: scikit-learn in /usr/local/lib/python3.11/dist-packages (from pytorchWildlife) (1.6.1)\n",
- "Requirement already satisfied: defusedxml<0.8.0,>=0.7.1 in /usr/local/lib/python3.11/dist-packages (from supervision==0.23.0->pytorchWildlife) (0.7.1)\n",
- "Requirement already satisfied: matplotlib>=3.6.0 in /usr/local/lib/python3.11/dist-packages (from supervision==0.23.0->pytorchWildlife) (3.10.0)\n",
- "Requirement already satisfied: numpy>=1.23.3 in /usr/local/lib/python3.11/dist-packages (from supervision==0.23.0->pytorchWildlife) (1.26.4)\n",
- "Requirement already satisfied: opencv-python-headless>=4.5.5.64 in /usr/local/lib/python3.11/dist-packages (from supervision==0.23.0->pytorchWildlife) (4.11.0.86)\n",
- "Requirement already satisfied: pyyaml>=5.3 in /usr/local/lib/python3.11/dist-packages (from supervision==0.23.0->pytorchWildlife) (6.0.2)\n",
- "Requirement already satisfied: scipy<2.0.0,>=1.10.0 in /usr/local/lib/python3.11/dist-packages (from supervision==0.23.0->pytorchWildlife) (1.13.1)\n",
- "Requirement already satisfied: aiofiles<24.0,>=22.0 in /usr/local/lib/python3.11/dist-packages (from gradio->pytorchWildlife) (23.2.1)\n",
- "Requirement already satisfied: anyio<5.0,>=3.0 in /usr/local/lib/python3.11/dist-packages (from gradio->pytorchWildlife) (3.7.1)\n",
- "Requirement already satisfied: fastapi<1.0,>=0.115.2 in /usr/local/lib/python3.11/dist-packages (from gradio->pytorchWildlife) (0.115.7)\n",
- "Requirement already satisfied: ffmpy in /usr/local/lib/python3.11/dist-packages (from gradio->pytorchWildlife) (0.5.0)\n",
- "Requirement already satisfied: gradio-client==1.6.0 in /usr/local/lib/python3.11/dist-packages (from gradio->pytorchWildlife) (1.6.0)\n",
- "Requirement already satisfied: httpx>=0.24.1 in /usr/local/lib/python3.11/dist-packages (from gradio->pytorchWildlife) (0.28.1)\n",
- "Requirement already satisfied: huggingface-hub>=0.25.1 in /usr/local/lib/python3.11/dist-packages (from gradio->pytorchWildlife) (0.27.1)\n",
- "Requirement already satisfied: jinja2<4.0 in /usr/local/lib/python3.11/dist-packages (from gradio->pytorchWildlife) (3.1.5)\n",
- "Requirement already satisfied: markupsafe~=2.0 in /usr/local/lib/python3.11/dist-packages (from gradio->pytorchWildlife) (2.1.5)\n",
- "Requirement already satisfied: orjson~=3.0 in /usr/local/lib/python3.11/dist-packages (from gradio->pytorchWildlife) (3.10.15)\n",
- "Requirement already satisfied: packaging in /usr/local/lib/python3.11/dist-packages (from gradio->pytorchWildlife) (24.2)\n",
- "Requirement already satisfied: pandas<3.0,>=1.0 in /usr/local/lib/python3.11/dist-packages (from gradio->pytorchWildlife) (2.2.2)\n",
- "Requirement already satisfied: pydantic>=2.0 in /usr/local/lib/python3.11/dist-packages (from gradio->pytorchWildlife) (2.10.5)\n",
- "Requirement already satisfied: pydub in /usr/local/lib/python3.11/dist-packages (from gradio->pytorchWildlife) (0.25.1)\n",
- "Requirement already satisfied: python-multipart>=0.0.18 in /usr/local/lib/python3.11/dist-packages (from gradio->pytorchWildlife) (0.0.20)\n",
- "Requirement already satisfied: ruff>=0.2.2 in /usr/local/lib/python3.11/dist-packages (from gradio->pytorchWildlife) (0.9.3)\n",
- "Requirement already satisfied: safehttpx<0.2.0,>=0.1.6 in /usr/local/lib/python3.11/dist-packages (from gradio->pytorchWildlife) (0.1.6)\n",
- "Requirement already satisfied: semantic-version~=2.0 in /usr/local/lib/python3.11/dist-packages (from gradio->pytorchWildlife) (2.10.0)\n",
- "Requirement already satisfied: starlette<1.0,>=0.40.0 in /usr/local/lib/python3.11/dist-packages (from gradio->pytorchWildlife) (0.45.3)\n",
- "Requirement already satisfied: tomlkit<0.14.0,>=0.12.0 in /usr/local/lib/python3.11/dist-packages (from gradio->pytorchWildlife) (0.13.2)\n",
- "Requirement already satisfied: typer<1.0,>=0.12 in /usr/local/lib/python3.11/dist-packages (from gradio->pytorchWildlife) (0.15.1)\n",
- "Requirement already satisfied: typing-extensions~=4.0 in /usr/local/lib/python3.11/dist-packages (from gradio->pytorchWildlife) (4.12.2)\n",
- "Requirement already satisfied: uvicorn>=0.14.0 in /usr/local/lib/python3.11/dist-packages (from gradio->pytorchWildlife) (0.34.0)\n",
- "Requirement already satisfied: fsspec in /usr/local/lib/python3.11/dist-packages (from gradio-client==1.6.0->gradio->pytorchWildlife) (2024.10.0)\n",
- "Requirement already satisfied: websockets<15.0,>=10.0 in /usr/local/lib/python3.11/dist-packages (from gradio-client==1.6.0->gradio->pytorchWildlife) (14.2)\n",
- "Requirement already satisfied: joblib>=1.2.0 in /usr/local/lib/python3.11/dist-packages (from scikit-learn->pytorchWildlife) (1.4.2)\n",
- "Requirement already satisfied: threadpoolctl>=3.1.0 in /usr/local/lib/python3.11/dist-packages (from scikit-learn->pytorchWildlife) (3.5.0)\n",
- "Requirement already satisfied: filelock in /usr/local/lib/python3.11/dist-packages (from torch->pytorchWildlife) (3.17.0)\n",
- "Requirement already satisfied: sympy in /usr/local/lib/python3.11/dist-packages (from torch->pytorchWildlife) (1.13.1)\n",
- "Requirement already satisfied: networkx in /usr/local/lib/python3.11/dist-packages (from torch->pytorchWildlife) (3.4.2)\n",
- "Requirement already satisfied: nvidia-cuda-nvrtc-cu12==12.1.105 in /usr/local/lib/python3.11/dist-packages (from torch->pytorchWildlife) (12.1.105)\n",
- "Requirement already satisfied: nvidia-cuda-runtime-cu12==12.1.105 in /usr/local/lib/python3.11/dist-packages (from torch->pytorchWildlife) (12.1.105)\n",
- "Requirement already satisfied: nvidia-cuda-cupti-cu12==12.1.105 in /usr/local/lib/python3.11/dist-packages (from torch->pytorchWildlife) (12.1.105)\n",
- "Requirement already satisfied: nvidia-cudnn-cu12==8.9.2.26 in /usr/local/lib/python3.11/dist-packages (from torch->pytorchWildlife) (8.9.2.26)\n",
- "Requirement already satisfied: nvidia-cublas-cu12==12.1.3.1 in /usr/local/lib/python3.11/dist-packages (from torch->pytorchWildlife) (12.1.3.1)\n",
- "Requirement already satisfied: nvidia-cufft-cu12==11.0.2.54 in /usr/local/lib/python3.11/dist-packages (from torch->pytorchWildlife) (11.0.2.54)\n",
- "Requirement already satisfied: nvidia-curand-cu12==10.3.2.106 in /usr/local/lib/python3.11/dist-packages (from torch->pytorchWildlife) (10.3.2.106)\n",
- "Requirement already satisfied: nvidia-cusolver-cu12==11.4.5.107 in /usr/local/lib/python3.11/dist-packages (from torch->pytorchWildlife) (11.4.5.107)\n",
- "Requirement already satisfied: nvidia-cusparse-cu12==12.1.0.106 in /usr/local/lib/python3.11/dist-packages (from torch->pytorchWildlife) (12.1.0.106)\n",
- "Requirement already satisfied: nvidia-nccl-cu12==2.19.3 in /usr/local/lib/python3.11/dist-packages (from torch->pytorchWildlife) (2.19.3)\n",
- "Requirement already satisfied: nvidia-nvtx-cu12==12.1.105 in /usr/local/lib/python3.11/dist-packages (from torch->pytorchWildlife) (12.1.105)\n",
- "Requirement already satisfied: triton==2.2.0 in /usr/local/lib/python3.11/dist-packages (from torch->pytorchWildlife) (2.2.0)\n",
- "Requirement already satisfied: nvidia-nvjitlink-cu12 in /usr/local/lib/python3.11/dist-packages (from nvidia-cusolver-cu12==11.4.5.107->torch->pytorchWildlife) (12.6.85)\n",
- "Requirement already satisfied: opencv-python>=4.6.0 in /usr/local/lib/python3.11/dist-packages (from ultralytics->pytorchWildlife) (4.10.0.84)\n",
- "Requirement already satisfied: requests>=2.23.0 in /usr/local/lib/python3.11/dist-packages (from ultralytics->pytorchWildlife) (2.32.3)\n",
- "Requirement already satisfied: psutil in /usr/local/lib/python3.11/dist-packages (from ultralytics->pytorchWildlife) (5.9.5)\n",
- "Requirement already satisfied: py-cpuinfo in /usr/local/lib/python3.11/dist-packages (from ultralytics->pytorchWildlife) (9.0.0)\n",
- "Requirement already satisfied: seaborn>=0.11.0 in /usr/local/lib/python3.11/dist-packages (from ultralytics->pytorchWildlife) (0.13.2)\n",
- "Requirement already satisfied: ultralytics-thop>=2.0.0 in /usr/local/lib/python3.11/dist-packages (from ultralytics->pytorchWildlife) (2.0.14)\n",
- "Requirement already satisfied: protobuf<=3.20.1 in /usr/local/lib/python3.11/dist-packages (from ultralytics-yolov5->pytorchWildlife) (3.20.1)\n",
- "Requirement already satisfied: fire in /usr/local/lib/python3.11/dist-packages (from ultralytics-yolov5->pytorchWildlife) (0.7.0)\n",
- "Requirement already satisfied: tensorboard>=2.4.1 in /usr/local/lib/python3.11/dist-packages (from ultralytics-yolov5->pytorchWildlife) (2.17.1)\n",
- "Requirement already satisfied: ipython in /usr/local/lib/python3.11/dist-packages (from ultralytics-yolov5->pytorchWildlife) (8.31.0)\n",
- "Requirement already satisfied: thop in /usr/local/lib/python3.11/dist-packages (from ultralytics-yolov5->pytorchWildlife) (0.1.1.post2209072238)\n",
- "Requirement already satisfied: idna>=2.8 in /usr/local/lib/python3.11/dist-packages (from anyio<5.0,>=3.0->gradio->pytorchWildlife) (3.10)\n",
- "Requirement already satisfied: sniffio>=1.1 in /usr/local/lib/python3.11/dist-packages (from anyio<5.0,>=3.0->gradio->pytorchWildlife) (1.3.1)\n",
- "Requirement already satisfied: certifi in /usr/local/lib/python3.11/dist-packages (from httpx>=0.24.1->gradio->pytorchWildlife) (2024.12.14)\n",
- "Requirement already satisfied: httpcore==1.* in /usr/local/lib/python3.11/dist-packages (from httpx>=0.24.1->gradio->pytorchWildlife) (1.0.7)\n",
- "Requirement already satisfied: h11<0.15,>=0.13 in /usr/local/lib/python3.11/dist-packages (from httpcore==1.*->httpx>=0.24.1->gradio->pytorchWildlife) (0.14.0)\n",
- "Requirement already satisfied: contourpy>=1.0.1 in /usr/local/lib/python3.11/dist-packages (from matplotlib>=3.6.0->supervision==0.23.0->pytorchWildlife) (1.3.1)\n",
- "Requirement already satisfied: cycler>=0.10 in /usr/local/lib/python3.11/dist-packages (from matplotlib>=3.6.0->supervision==0.23.0->pytorchWildlife) (0.12.1)\n",
- "Requirement already satisfied: fonttools>=4.22.0 in /usr/local/lib/python3.11/dist-packages (from matplotlib>=3.6.0->supervision==0.23.0->pytorchWildlife) (4.55.5)\n",
- "Requirement already satisfied: kiwisolver>=1.3.1 in /usr/local/lib/python3.11/dist-packages (from matplotlib>=3.6.0->supervision==0.23.0->pytorchWildlife) (1.4.8)\n",
- "Requirement already satisfied: pyparsing>=2.3.1 in /usr/local/lib/python3.11/dist-packages (from matplotlib>=3.6.0->supervision==0.23.0->pytorchWildlife) (3.2.1)\n",
- "Requirement already satisfied: python-dateutil>=2.7 in /usr/local/lib/python3.11/dist-packages (from matplotlib>=3.6.0->supervision==0.23.0->pytorchWildlife) (2.8.2)\n",
- "Requirement already satisfied: pytz>=2020.1 in /usr/local/lib/python3.11/dist-packages (from pandas<3.0,>=1.0->gradio->pytorchWildlife) (2024.2)\n",
- "Requirement already satisfied: tzdata>=2022.7 in /usr/local/lib/python3.11/dist-packages (from pandas<3.0,>=1.0->gradio->pytorchWildlife) (2025.1)\n",
- "Requirement already satisfied: annotated-types>=0.6.0 in /usr/local/lib/python3.11/dist-packages (from pydantic>=2.0->gradio->pytorchWildlife) (0.7.0)\n",
- "Requirement already satisfied: pydantic-core==2.27.2 in /usr/local/lib/python3.11/dist-packages (from pydantic>=2.0->gradio->pytorchWildlife) (2.27.2)\n",
- "Requirement already satisfied: charset-normalizer<4,>=2 in /usr/local/lib/python3.11/dist-packages (from requests>=2.23.0->ultralytics->pytorchWildlife) (3.4.1)\n",
- "Requirement already satisfied: urllib3<3,>=1.21.1 in /usr/local/lib/python3.11/dist-packages (from requests>=2.23.0->ultralytics->pytorchWildlife) (2.3.0)\n",
- "Requirement already satisfied: absl-py>=0.4 in /usr/local/lib/python3.11/dist-packages (from tensorboard>=2.4.1->ultralytics-yolov5->pytorchWildlife) (1.4.0)\n",
- "Requirement already satisfied: grpcio>=1.48.2 in /usr/local/lib/python3.11/dist-packages (from tensorboard>=2.4.1->ultralytics-yolov5->pytorchWildlife) (1.69.0)\n",
- "Requirement already satisfied: markdown>=2.6.8 in /usr/local/lib/python3.11/dist-packages (from tensorboard>=2.4.1->ultralytics-yolov5->pytorchWildlife) (3.7)\n",
- "Requirement already satisfied: six>1.9 in /usr/local/lib/python3.11/dist-packages (from tensorboard>=2.4.1->ultralytics-yolov5->pytorchWildlife) (1.17.0)\n",
- "Requirement already satisfied: tensorboard-data-server<0.8.0,>=0.7.0 in /usr/local/lib/python3.11/dist-packages (from tensorboard>=2.4.1->ultralytics-yolov5->pytorchWildlife) (0.7.2)\n",
- "Requirement already satisfied: werkzeug>=1.0.1 in /usr/local/lib/python3.11/dist-packages (from tensorboard>=2.4.1->ultralytics-yolov5->pytorchWildlife) (3.1.3)\n",
- "Requirement already satisfied: click>=8.0.0 in /usr/local/lib/python3.11/dist-packages (from typer<1.0,>=0.12->gradio->pytorchWildlife) (8.1.8)\n",
- "Requirement already satisfied: shellingham>=1.3.0 in /usr/local/lib/python3.11/dist-packages (from typer<1.0,>=0.12->gradio->pytorchWildlife) (1.5.4)\n",
- "Requirement already satisfied: rich>=10.11.0 in /usr/local/lib/python3.11/dist-packages (from typer<1.0,>=0.12->gradio->pytorchWildlife) (13.9.4)\n",
- "Requirement already satisfied: termcolor in /usr/local/lib/python3.11/dist-packages (from fire->ultralytics-yolov5->pytorchWildlife) (2.5.0)\n",
- "Requirement already satisfied: decorator in /usr/local/lib/python3.11/dist-packages (from ipython->ultralytics-yolov5->pytorchWildlife) (4.4.2)\n",
- "Requirement already satisfied: jedi>=0.16 in /usr/local/lib/python3.11/dist-packages (from ipython->ultralytics-yolov5->pytorchWildlife) (0.19.2)\n",
- "Requirement already satisfied: matplotlib-inline in /usr/local/lib/python3.11/dist-packages (from ipython->ultralytics-yolov5->pytorchWildlife) (0.1.7)\n",
- "Requirement already satisfied: pexpect>4.3 in /usr/local/lib/python3.11/dist-packages (from ipython->ultralytics-yolov5->pytorchWildlife) (4.9.0)\n",
- "Requirement already satisfied: prompt_toolkit<3.1.0,>=3.0.41 in /usr/local/lib/python3.11/dist-packages (from ipython->ultralytics-yolov5->pytorchWildlife) (3.0.50)\n",
- "Requirement already satisfied: pygments>=2.4.0 in /usr/local/lib/python3.11/dist-packages (from ipython->ultralytics-yolov5->pytorchWildlife) (2.18.0)\n",
- "Requirement already satisfied: stack_data in /usr/local/lib/python3.11/dist-packages (from ipython->ultralytics-yolov5->pytorchWildlife) (0.6.3)\n",
- "Requirement already satisfied: traitlets>=5.13.0 in /usr/local/lib/python3.11/dist-packages (from ipython->ultralytics-yolov5->pytorchWildlife) (5.14.3)\n",
- "Requirement already satisfied: mpmath<1.4,>=1.1.0 in /usr/local/lib/python3.11/dist-packages (from sympy->torch->pytorchWildlife) (1.3.0)\n",
- "Requirement already satisfied: parso<0.9.0,>=0.8.4 in /usr/local/lib/python3.11/dist-packages (from jedi>=0.16->ipython->ultralytics-yolov5->pytorchWildlife) (0.8.4)\n",
- "Requirement already satisfied: ptyprocess>=0.5 in /usr/local/lib/python3.11/dist-packages (from pexpect>4.3->ipython->ultralytics-yolov5->pytorchWildlife) (0.7.0)\n",
- "Requirement already satisfied: wcwidth in /usr/local/lib/python3.11/dist-packages (from prompt_toolkit<3.1.0,>=3.0.41->ipython->ultralytics-yolov5->pytorchWildlife) (0.2.13)\n",
- "Requirement already satisfied: markdown-it-py>=2.2.0 in /usr/local/lib/python3.11/dist-packages (from rich>=10.11.0->typer<1.0,>=0.12->gradio->pytorchWildlife) (3.0.0)\n",
- "Requirement already satisfied: executing>=1.2.0 in /usr/local/lib/python3.11/dist-packages (from stack_data->ipython->ultralytics-yolov5->pytorchWildlife) (2.2.0)\n",
- "Requirement already satisfied: asttokens>=2.1.0 in /usr/local/lib/python3.11/dist-packages (from stack_data->ipython->ultralytics-yolov5->pytorchWildlife) (3.0.0)\n",
- "Requirement already satisfied: pure-eval in /usr/local/lib/python3.11/dist-packages (from stack_data->ipython->ultralytics-yolov5->pytorchWildlife) (0.2.3)\n",
- "Requirement already satisfied: mdurl~=0.1 in /usr/local/lib/python3.11/dist-packages (from markdown-it-py>=2.2.0->rich>=10.11.0->typer<1.0,>=0.12->gradio->pytorchWildlife) (0.1.2)\n"
- ]
- },
- {
- "output_type": "error",
- "ename": "ModuleNotFoundError",
- "evalue": "No module named 'pytorchWildlife'",
- "traceback": [
- "\u001b[0;31m---------------------------------------------------------------------------\u001b[0m",
- "\u001b[0;31mModuleNotFoundError\u001b[0m Traceback (most recent call last)",
- "\u001b[0;32m\u001b[0m in \u001b[0;36m\u001b[0;34m()\u001b[0m\n\u001b[1;32m 1\u001b[0m \u001b[0mget_ipython\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0msystem\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0;34m'pip install pytorchWildlife'\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0;32m----> 2\u001b[0;31m \u001b[0;32mimport\u001b[0m \u001b[0mpytorchWildlife\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0m\u001b[1;32m 3\u001b[0m \u001b[0mprint\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0mPytorchWildlife\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0m__version__\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m 4\u001b[0m \u001b[0mprint\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0mPytorchWildlife\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0m__file__\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n",
- "\u001b[0;31mModuleNotFoundError\u001b[0m: No module named 'pytorchWildlife'",
- "",
- "\u001b[0;31m---------------------------------------------------------------------------\u001b[0;32m\nNOTE: If your import is failing due to a missing package, you can\nmanually install dependencies using either !pip or !apt.\n\nTo view examples of installing some common dependencies, click the\n\"Open Examples\" button below.\n\u001b[0;31m---------------------------------------------------------------------------\u001b[0m\n"
- ],
- "errorDetails": {
- "actions": [
- {
- "action": "open_url",
- "actionText": "Open Examples",
- "url": "/notebooks/snippets/importing_libraries.ipynb"
- }
- ]
- }
- }
- ],
- "source": [
- "!pip install pytorchWildlife\n",
- "import pytorchWildlife\n",
- "print(PytorchWildlife.__version__)\n",
- "print(PytorchWildlife.__file__)"
- ]
- },
- {
- "cell_type": "markdown",
- "id": "1197e180",
- "metadata": {
- "id": "1197e180"
- },
- "source": [
- "This tutorial guides you on how to use PyTorchWildlife for image detection. We will go through the process of setting up the environment, defining the detection model, as well as performing inference and saving the results in different ways.\n",
- "\n",
- "## Prerequisites\n",
- "Install PytorchWildlife running the following commands:\n",
- "```bash\n",
- "conda create -n pytorch_wildlife python=3.8 -y\n",
- "conda activate pytorch_wildlife\n",
- "pip install PytorchWildlife\n",
- "```\n",
- "Also, make sure you have a CUDA-capable GPU if you intend to run the model on a GPU. This notebook can also run on CPU.\n",
- "\n",
- "## Importing libraries\n",
- "First, we'll start by importing the necessary libraries and modules."
- ]
- },
- {
- "cell_type": "code",
- "execution_count": null,
- "id": "c44e7713",
- "metadata": {
- "id": "c44e7713"
- },
- "outputs": [],
- "source": [
- "import numpy as np\n",
- "from PIL import Image\n",
- "import torch\n",
- "from torch.utils.data import DataLoader\n",
- "from PytorchWildlife.models import detection as pw_detection\n",
- "from PytorchWildlife.models import classification as pw_classification\n",
- "\n",
- "from PytorchWildlife.data import transforms as pw_trans\n",
- "from PytorchWildlife.data import datasets as pw_data\n",
- "from PytorchWildlife import utils as pw_utils"
- ]
- },
- {
- "cell_type": "markdown",
- "id": "3a23982f",
- "metadata": {
- "id": "3a23982f"
- },
- "source": [
- "## Setting GPU\n",
- "If you are using a GPU for this exercise, please specify which GPU to use for the computations. By default, GPU number 0 is used. Adjust this as per your setup. You don't need to run this cell if you are using a CPU."
- ]
- },
- {
- "cell_type": "code",
- "execution_count": null,
- "id": "8f622040",
- "metadata": {
- "id": "8f622040"
- },
- "outputs": [],
- "source": [
- "torch.cuda.set_device(0) # Only use if you are running on GPU."
- ]
- },
- {
- "cell_type": "markdown",
- "id": "6abd07b5",
- "metadata": {
- "id": "6abd07b5"
- },
- "source": [
- "## Model Initialization\n",
- "We will initialize the MegaDetectorV5 model for image detection. This model is designed for detecting animals in images."
- ]
- },
- {
- "cell_type": "code",
- "execution_count": null,
- "id": "bnRqmMVf-N0q",
- "metadata": {
- "id": "bnRqmMVf-N0q"
- },
- "outputs": [],
- "source": [
- "!venv export"
- ]
- },
- {
- "cell_type": "code",
- "execution_count": null,
- "id": "ufjnWYWUwwGf",
- "metadata": {
- "collapsed": true,
- "id": "ufjnWYWUwwGf"
- },
- "outputs": [],
- "source": [
- "# new RR, 240926\n",
- "# !pip show ultralytics\n",
- "!pip install ultralytics"
- ]
- },
- {
- "cell_type": "code",
- "execution_count": null,
- "id": "UInvY5lRjRjs",
- "metadata": {
- "id": "UInvY5lRjRjs"
- },
- "outputs": [],
- "source": [
- "# from yolov8_base import *\n",
- "# from megadetectorv6 import *\n",
- "\n",
- "#%%\n",
- "# Initializing the MegaDetectorV5 model for image detection\n",
- "# detection_model = pw_detection.MegaDetectorV5(device=DEVICE, pretrained=True)\n",
- "#detection_model = pw_detection.MegaDetectorV5(device=DEVICE, weights='https://zenodo.org/records/11192829/files/MDV6b-yolov9c.pt', pretrained=False) # For beta testing, you need to specify the path to the weights file."
- ]
- },
- {
- "cell_type": "markdown",
- "id": "umnCY0tcdQ2T",
- "metadata": {
- "id": "umnCY0tcdQ2T"
- },
- "source": []
- },
- {
- "cell_type": "code",
- "execution_count": null,
- "id": "ilTFFWPqw9C2",
- "metadata": {
- "id": "ilTFFWPqw9C2"
- },
- "outputs": [],
- "source": [
- "#DEVICE = \"cpu\" # Use \"cuda\" if GPU is available \"cpu\" if no GPU is available\n",
- "\n",
- "#detection_model = pw_detection.MegaDetectorV6(device=DEVICE, weights='../MDV6b-yolov9c.pt', pretrained=False)"
- ]
- },
- {
- "cell_type": "code",
- "execution_count": null,
- "id": "2fYOIiEKnrJ2",
- "metadata": {
- "id": "2fYOIiEKnrJ2"
- },
- "outputs": [],
- "source": []
- },
- {
- "cell_type": "code",
- "execution_count": null,
- "id": "eb25db43",
- "metadata": {
- "id": "eb25db43"
- },
- "outputs": [],
- "source": [
- "DEVICE = \"cuda\" # Use \"cuda\" if GPU is available \"cpu\" if no GPU is available\n",
- "detection_model = pw_detection.MegaDetectorV6(device = DEVICE)\n",
- "#detection_model = pw_detection.MegaDetectorV5(device=DEVICE, weights='https://zenodo.org/records/11192829/files/MDV6b-yolov9c.pt', pretrained=False) # For beta testing, you need to specify the path to the weights file."
- ]
- },
- {
- "cell_type": "markdown",
- "id": "1e57dcca",
- "metadata": {
- "id": "1e57dcca"
- },
- "source": [
- "## Single Image Detection\n",
- "We will first perform detection on a single image. Make sure to verify that you have the image in the specified path."
- ]
- },
- {
- "cell_type": "code",
- "execution_count": null,
- "id": "0d730b20",
- "metadata": {
- "id": "0d730b20"
- },
- "outputs": [],
- "source": [
- "tgt_img_path = \"/content/Screenshot 2025-01-28 at 3.27.56 PM.png\"\n",
- "img = np.array(Image.open(tgt_img_path).convert(\"RGB\"))\n",
- "transform = pw_trans.MegaDetector_v5_Transform(target_size=detection_model.IMAGE_SIZE,\n",
- " stride=detection_model.STRIDE)\n",
- "results = detection_model.single_image_detection(transform(img), img.shape, tgt_img_path)\n",
- "pw_utils.save_detection_images(results, \"./demo_output\")\n",
- "\n",
- "# results 객체에서 detections 객체를 가져옵니다.\n",
- "detections = results['detections']\n",
- "\n",
- "# confidence 값을 추출하는 코드를 작성합니다.\n",
- "# 이는 detections 객체의 구현에 따라 다를 수 있으므로, 다음 라인은 예시일 뿐입니다.\n",
- "# 실제 구현에서는 detections.confidence 또는 비슷한 메소드가 필요할 수 있습니다.\n",
- "confidence_scores = detections.confidence # 가정된 접근 방법\n",
- "\n",
- "# 점수가 0.7 이상인지 검사하여 출력합니다.\n",
- "output = 1 if any(score > 0.7 for score in confidence_scores) else 0\n",
- "\n",
- "print(output)"
- ]
- },
- {
- "cell_type": "code",
- "source": [],
- "metadata": {
- "id": "rYl54P3WwXMy"
- },
- "id": "rYl54P3WwXMy",
- "execution_count": null,
- "outputs": []
- }
- ],
- "metadata": {
- "accelerator": "GPU",
- "colab": {
- "gpuType": "T4",
- "provenance": [],
- "include_colab_link": true
- },
- "kernelspec": {
- "display_name": "Python 3",
- "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.8.18"
- }
- },
- "nbformat": 4,
- "nbformat_minor": 5
-}
\ No newline at end of file
|