{ "cells": [ { "cell_type": "code", "execution_count": 10, "metadata": { "colab": { "base_uri": "https://localhost:8080/" }, "id": "rPv_jkuhaZN8", "outputId": "f452d923-7e23-4ced-d295-6000c19f0e83", "tags": [] }, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "Requirement already satisfied: ultralytics in c:\\users\\zeus\\appdata\\local\\packages\\pythonsoftwarefoundation.python.3.11_qbz5n2kfra8p0\\localcache\\local-packages\\python311\\site-packages (8.1.4)\n", "Requirement already satisfied: matplotlib>=3.3.0 in c:\\users\\zeus\\appdata\\local\\packages\\pythonsoftwarefoundation.python.3.11_qbz5n2kfra8p0\\localcache\\local-packages\\python311\\site-packages (from ultralytics) (3.8.2)\n", "Requirement already satisfied: numpy>=1.22.2 in c:\\users\\zeus\\appdata\\local\\packages\\pythonsoftwarefoundation.python.3.11_qbz5n2kfra8p0\\localcache\\local-packages\\python311\\site-packages (from ultralytics) (1.26.0)\n", "Requirement already satisfied: opencv-python>=4.6.0 in c:\\users\\zeus\\appdata\\local\\packages\\pythonsoftwarefoundation.python.3.11_qbz5n2kfra8p0\\localcache\\local-packages\\python311\\site-packages (from ultralytics) (4.8.1.78)\n", "Requirement already satisfied: pillow>=7.1.2 in c:\\users\\zeus\\appdata\\local\\packages\\pythonsoftwarefoundation.python.3.11_qbz5n2kfra8p0\\localcache\\local-packages\\python311\\site-packages (from ultralytics) (9.5.0)\n", "Requirement already satisfied: pyyaml>=5.3.1 in c:\\users\\zeus\\appdata\\local\\packages\\pythonsoftwarefoundation.python.3.11_qbz5n2kfra8p0\\localcache\\local-packages\\python311\\site-packages (from ultralytics) (6.0.1)\n", "Requirement already satisfied: requests>=2.23.0 in c:\\users\\zeus\\appdata\\local\\packages\\pythonsoftwarefoundation.python.3.11_qbz5n2kfra8p0\\localcache\\local-packages\\python311\\site-packages (from ultralytics) (2.31.0)\n", "Requirement already satisfied: scipy>=1.4.1 in c:\\users\\zeus\\appdata\\local\\packages\\pythonsoftwarefoundation.python.3.11_qbz5n2kfra8p0\\localcache\\local-packages\\python311\\site-packages (from ultralytics) (1.11.4)\n", "Requirement already satisfied: torch>=1.8.0 in c:\\users\\zeus\\appdata\\local\\packages\\pythonsoftwarefoundation.python.3.11_qbz5n2kfra8p0\\localcache\\local-packages\\python311\\site-packages (from ultralytics) (2.1.2)\n", "Requirement already satisfied: torchvision>=0.9.0 in c:\\users\\zeus\\appdata\\local\\packages\\pythonsoftwarefoundation.python.3.11_qbz5n2kfra8p0\\localcache\\local-packages\\python311\\site-packages (from ultralytics) (0.16.2)\n", "Requirement already satisfied: tqdm>=4.64.0 in c:\\users\\zeus\\appdata\\local\\packages\\pythonsoftwarefoundation.python.3.11_qbz5n2kfra8p0\\localcache\\local-packages\\python311\\site-packages (from ultralytics) (4.66.1)\n", "Requirement already satisfied: psutil in c:\\users\\zeus\\appdata\\local\\packages\\pythonsoftwarefoundation.python.3.11_qbz5n2kfra8p0\\localcache\\local-packages\\python311\\site-packages (from ultralytics) (5.9.7)\n", "Requirement already satisfied: py-cpuinfo in c:\\users\\zeus\\appdata\\local\\packages\\pythonsoftwarefoundation.python.3.11_qbz5n2kfra8p0\\localcache\\local-packages\\python311\\site-packages (from ultralytics) (9.0.0)\n", "Requirement already satisfied: thop>=0.1.1 in c:\\users\\zeus\\appdata\\local\\packages\\pythonsoftwarefoundation.python.3.11_qbz5n2kfra8p0\\localcache\\local-packages\\python311\\site-packages (from ultralytics) (0.1.1.post2209072238)\n", "Requirement already satisfied: pandas>=1.1.4 in c:\\users\\zeus\\appdata\\local\\packages\\pythonsoftwarefoundation.python.3.11_qbz5n2kfra8p0\\localcache\\local-packages\\python311\\site-packages (from ultralytics) (2.1.0)\n", "Requirement already satisfied: seaborn>=0.11.0 in c:\\users\\zeus\\appdata\\local\\packages\\pythonsoftwarefoundation.python.3.11_qbz5n2kfra8p0\\localcache\\local-packages\\python311\\site-packages (from ultralytics) (0.13.1)\n", "Requirement already satisfied: contourpy>=1.0.1 in c:\\users\\zeus\\appdata\\local\\packages\\pythonsoftwarefoundation.python.3.11_qbz5n2kfra8p0\\localcache\\local-packages\\python311\\site-packages (from matplotlib>=3.3.0->ultralytics) (1.2.0)\n", "Requirement already satisfied: cycler>=0.10 in c:\\users\\zeus\\appdata\\local\\packages\\pythonsoftwarefoundation.python.3.11_qbz5n2kfra8p0\\localcache\\local-packages\\python311\\site-packages (from matplotlib>=3.3.0->ultralytics) (0.12.1)\n", "Requirement already satisfied: fonttools>=4.22.0 in c:\\users\\zeus\\appdata\\local\\packages\\pythonsoftwarefoundation.python.3.11_qbz5n2kfra8p0\\localcache\\local-packages\\python311\\site-packages (from matplotlib>=3.3.0->ultralytics) (4.47.0)\n", "Requirement already satisfied: kiwisolver>=1.3.1 in c:\\users\\zeus\\appdata\\local\\packages\\pythonsoftwarefoundation.python.3.11_qbz5n2kfra8p0\\localcache\\local-packages\\python311\\site-packages (from matplotlib>=3.3.0->ultralytics) (1.4.5)\n", "Requirement already satisfied: packaging>=20.0 in c:\\users\\zeus\\appdata\\local\\packages\\pythonsoftwarefoundation.python.3.11_qbz5n2kfra8p0\\localcache\\local-packages\\python311\\site-packages (from matplotlib>=3.3.0->ultralytics) (23.1)\n", "Requirement already satisfied: pyparsing>=2.3.1 in c:\\users\\zeus\\appdata\\local\\packages\\pythonsoftwarefoundation.python.3.11_qbz5n2kfra8p0\\localcache\\local-packages\\python311\\site-packages (from matplotlib>=3.3.0->ultralytics) (3.1.1)\n", "Requirement already satisfied: python-dateutil>=2.7 in c:\\users\\zeus\\appdata\\local\\packages\\pythonsoftwarefoundation.python.3.11_qbz5n2kfra8p0\\localcache\\local-packages\\python311\\site-packages (from matplotlib>=3.3.0->ultralytics) (2.8.2)\n", "Requirement already satisfied: pytz>=2020.1 in c:\\users\\zeus\\appdata\\local\\packages\\pythonsoftwarefoundation.python.3.11_qbz5n2kfra8p0\\localcache\\local-packages\\python311\\site-packages (from pandas>=1.1.4->ultralytics) (2023.3.post1)\n", "Requirement already satisfied: tzdata>=2022.1 in c:\\users\\zeus\\appdata\\local\\packages\\pythonsoftwarefoundation.python.3.11_qbz5n2kfra8p0\\localcache\\local-packages\\python311\\site-packages (from pandas>=1.1.4->ultralytics) (2023.3)\n", "Requirement already satisfied: charset-normalizer<4,>=2 in c:\\users\\zeus\\appdata\\local\\packages\\pythonsoftwarefoundation.python.3.11_qbz5n2kfra8p0\\localcache\\local-packages\\python311\\site-packages (from requests>=2.23.0->ultralytics) (3.2.0)\n", "Requirement already satisfied: idna<4,>=2.5 in c:\\users\\zeus\\appdata\\local\\packages\\pythonsoftwarefoundation.python.3.11_qbz5n2kfra8p0\\localcache\\local-packages\\python311\\site-packages (from requests>=2.23.0->ultralytics) (3.4)\n", "Requirement already satisfied: urllib3<3,>=1.21.1 in c:\\users\\zeus\\appdata\\local\\packages\\pythonsoftwarefoundation.python.3.11_qbz5n2kfra8p0\\localcache\\local-packages\\python311\\site-packages (from requests>=2.23.0->ultralytics) (1.26.16)\n", "Requirement already satisfied: certifi>=2017.4.17 in c:\\users\\zeus\\appdata\\local\\packages\\pythonsoftwarefoundation.python.3.11_qbz5n2kfra8p0\\localcache\\local-packages\\python311\\site-packages (from requests>=2.23.0->ultralytics) (2023.7.22)\n", "Requirement already satisfied: filelock in c:\\users\\zeus\\appdata\\local\\packages\\pythonsoftwarefoundation.python.3.11_qbz5n2kfra8p0\\localcache\\local-packages\\python311\\site-packages (from torch>=1.8.0->ultralytics) (3.13.1)\n", "Requirement already satisfied: typing-extensions in c:\\users\\zeus\\appdata\\local\\packages\\pythonsoftwarefoundation.python.3.11_qbz5n2kfra8p0\\localcache\\local-packages\\python311\\site-packages (from torch>=1.8.0->ultralytics) (4.8.0)\n", "Requirement already satisfied: sympy in c:\\users\\zeus\\appdata\\local\\packages\\pythonsoftwarefoundation.python.3.11_qbz5n2kfra8p0\\localcache\\local-packages\\python311\\site-packages (from torch>=1.8.0->ultralytics) (1.12)\n", "Requirement already satisfied: networkx in c:\\users\\zeus\\appdata\\local\\packages\\pythonsoftwarefoundation.python.3.11_qbz5n2kfra8p0\\localcache\\local-packages\\python311\\site-packages (from torch>=1.8.0->ultralytics) (3.2.1)\n", "Requirement already satisfied: jinja2 in c:\\users\\zeus\\appdata\\local\\packages\\pythonsoftwarefoundation.python.3.11_qbz5n2kfra8p0\\localcache\\local-packages\\python311\\site-packages (from torch>=1.8.0->ultralytics) (3.1.2)\n", "Requirement already satisfied: fsspec in c:\\users\\zeus\\appdata\\local\\packages\\pythonsoftwarefoundation.python.3.11_qbz5n2kfra8p0\\localcache\\local-packages\\python311\\site-packages (from torch>=1.8.0->ultralytics) (2023.12.2)\n", "Requirement already satisfied: colorama in c:\\users\\zeus\\appdata\\local\\packages\\pythonsoftwarefoundation.python.3.11_qbz5n2kfra8p0\\localcache\\local-packages\\python311\\site-packages (from tqdm>=4.64.0->ultralytics) (0.4.6)\n", "Requirement already satisfied: six>=1.5 in c:\\users\\zeus\\appdata\\local\\packages\\pythonsoftwarefoundation.python.3.11_qbz5n2kfra8p0\\localcache\\local-packages\\python311\\site-packages (from python-dateutil>=2.7->matplotlib>=3.3.0->ultralytics) (1.16.0)\n", "Requirement already satisfied: MarkupSafe>=2.0 in c:\\users\\zeus\\appdata\\local\\packages\\pythonsoftwarefoundation.python.3.11_qbz5n2kfra8p0\\localcache\\local-packages\\python311\\site-packages (from jinja2->torch>=1.8.0->ultralytics) (2.1.3)\n", "Requirement already satisfied: mpmath>=0.19 in c:\\users\\zeus\\appdata\\local\\packages\\pythonsoftwarefoundation.python.3.11_qbz5n2kfra8p0\\localcache\\local-packages\\python311\\site-packages (from sympy->torch>=1.8.0->ultralytics) (1.3.0)\n" ] }, { "name": "stderr", "output_type": "stream", "text": [ "\n", "[notice] A new release of pip is available: 23.3.2 -> 24.0\n", "[notice] To update, run: C:\\Users\\Zeus\\AppData\\Local\\Microsoft\\WindowsApps\\PythonSoftwareFoundation.Python.3.11_qbz5n2kfra8p0\\python.exe -m pip install --upgrade pip\n" ] } ], "source": [ "## install the packages\n", "! pip install ultralytics" ] }, { "cell_type": "code", "execution_count": 11, "metadata": { "colab": { "base_uri": "https://localhost:8080/" }, "id": "CDLyOpyraf2g", "outputId": "7a070817-581d-4d8d-892b-2db0695ed282", "tags": [] }, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "Requirement already satisfied: torch in c:\\users\\zeus\\appdata\\local\\packages\\pythonsoftwarefoundation.python.3.11_qbz5n2kfra8p0\\localcache\\local-packages\\python311\\site-packages (2.1.2)\n", "Requirement already satisfied: filelock in c:\\users\\zeus\\appdata\\local\\packages\\pythonsoftwarefoundation.python.3.11_qbz5n2kfra8p0\\localcache\\local-packages\\python311\\site-packages (from torch) (3.13.1)\n", "Requirement already satisfied: typing-extensions in c:\\users\\zeus\\appdata\\local\\packages\\pythonsoftwarefoundation.python.3.11_qbz5n2kfra8p0\\localcache\\local-packages\\python311\\site-packages (from torch) (4.8.0)\n", "Requirement already satisfied: sympy in c:\\users\\zeus\\appdata\\local\\packages\\pythonsoftwarefoundation.python.3.11_qbz5n2kfra8p0\\localcache\\local-packages\\python311\\site-packages (from torch) (1.12)\n", "Requirement already satisfied: networkx in c:\\users\\zeus\\appdata\\local\\packages\\pythonsoftwarefoundation.python.3.11_qbz5n2kfra8p0\\localcache\\local-packages\\python311\\site-packages (from torch) (3.2.1)\n", "Requirement already satisfied: jinja2 in c:\\users\\zeus\\appdata\\local\\packages\\pythonsoftwarefoundation.python.3.11_qbz5n2kfra8p0\\localcache\\local-packages\\python311\\site-packages (from torch) (3.1.2)\n", "Requirement already satisfied: fsspec in c:\\users\\zeus\\appdata\\local\\packages\\pythonsoftwarefoundation.python.3.11_qbz5n2kfra8p0\\localcache\\local-packages\\python311\\site-packages (from torch) (2023.12.2)\n", "Requirement already satisfied: MarkupSafe>=2.0 in c:\\users\\zeus\\appdata\\local\\packages\\pythonsoftwarefoundation.python.3.11_qbz5n2kfra8p0\\localcache\\local-packages\\python311\\site-packages (from jinja2->torch) (2.1.3)\n", "Requirement already satisfied: mpmath>=0.19 in c:\\users\\zeus\\appdata\\local\\packages\\pythonsoftwarefoundation.python.3.11_qbz5n2kfra8p0\\localcache\\local-packages\\python311\\site-packages (from sympy->torch) (1.3.0)\n" ] }, { "name": "stderr", "output_type": "stream", "text": [ "\n", "[notice] A new release of pip is available: 23.3.2 -> 24.0\n", "[notice] To update, run: C:\\Users\\Zeus\\AppData\\Local\\Microsoft\\WindowsApps\\PythonSoftwareFoundation.Python.3.11_qbz5n2kfra8p0\\python.exe -m pip install --upgrade pip\n" ] } ], "source": [ "! pip install torch" ] }, { "cell_type": "code", "execution_count": 8, "metadata": { "id": "7zwkmpyLainb", "tags": [] }, "outputs": [], "source": [ "# import packages\n", "\n", "import torch\n", "import cv2\n", "import os\n", "from tqdm.auto import tqdm\n", "import numpy as np\n", "import pandas as pd\n", "import tqdm" ] }, { "cell_type": "code", "execution_count": 26, "metadata": { "colab": { "base_uri": "https://localhost:8080/", "height": 425 }, "id": "-eVvOhSmaPr1", "outputId": "6e45e02a-0109-4235-ec27-4a8a31a7e2b4", "tags": [] }, "outputs": [ { "name": "stderr", "output_type": "stream", "text": [ " 0%| | 0/1 [00:00\n", "\n", "\n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", "
image_namepredictions
0chest-x-ray-in-normal-pregnancy (1).jpg[0.0]
1chest-x-ray-in-normal-pregnancy.jpg[0.0]
2normal-chest-radiograph-female-2.jpg[]
3normal-chest-radiograph-female-3.jpg[]
4normal-chest-radiograph-female-5.jpg[]
.........
95pneumothorax_40.jpg[]
96pneumothorax_41.jpg[]
97pneumothorax_42.jpg[0.0]
98pneumothorax_43.jpg[0.0, 0.0]
99pneumothorax_44.jpg[0.0]
\n", "

100 rows × 2 columns

\n", "" ], "text/plain": [ " image_name predictions\n", "0 chest-x-ray-in-normal-pregnancy (1).jpg [0.0]\n", "1 chest-x-ray-in-normal-pregnancy.jpg [0.0]\n", "2 normal-chest-radiograph-female-2.jpg []\n", "3 normal-chest-radiograph-female-3.jpg []\n", "4 normal-chest-radiograph-female-5.jpg []\n", ".. ... ...\n", "95 pneumothorax_40.jpg []\n", "96 pneumothorax_41.jpg []\n", "97 pneumothorax_42.jpg [0.0]\n", "98 pneumothorax_43.jpg [0.0, 0.0]\n", "99 pneumothorax_44.jpg [0.0]\n", "\n", "[100 rows x 2 columns]" ] }, "execution_count": 27, "metadata": {}, "output_type": "execute_result" } ], "source": [ "list_df[0].head(100)" ] }, { "cell_type": "code", "execution_count": 28, "metadata": {}, "outputs": [], "source": [ "## Saving each df (three class yolov8)\n", "\n", "df_pneumothorax=list_df[0]" ] }, { "cell_type": "code", "execution_count": 29, "metadata": {}, "outputs": [], "source": [ "## Exporting the individual dataframes to csv files (three class v8)\n", "\n", "\n", "df_pneumothorax.to_csv('pneumothorax_v8.csv')" ] }, { "cell_type": "code", "execution_count": 25, "metadata": {}, "outputs": [], "source": [ "# ## Saving each df\n", "# df_aortic=list_df[0]\n", "# df_atelec=list_df[1]\n", "# df_cal=list_df[2]\n", "# df_cardio=list_df[3]\n", "# df_ild=list_df[4]\n", "# df_opacity=list_df[5]\n", "# df_nodules=list_df[6]\n", "# df_effusion=list_df[7]\n", "# df_pneumothorax=list_df[8]" ] }, { "cell_type": "code", "execution_count": 26, "metadata": { "tags": [] }, "outputs": [], "source": [ "## Exporting the individual dataframes to csv files\n", "# df_aortic.to_csv('aortic_v8.csv')\n", "# df_atelec.to_csv('atelec_v8.csv')\n", "# df_cal.to_csv('calcifications_v8.csv')\n", "# df_cardio.to_csv('cardiomegaly_v8.csv')\n", "# df_ild.to_csv('ild_v8.csv')\n", "# df_opacity.to_csv('opacity_v8.csv')\n", "# df_nodules.to_csv('nodules_v8.csv')\n", "# df_effusion.to_csv('effusion_v8.csv')\n", "# df_pneumothorax.to_csv('pneumothorax_v8.csv')" ] } ], "metadata": { "accelerator": "GPU", "colab": { "gpuType": "T4", "provenance": [] }, "kernelspec": { "display_name": "Python 3 (ipykernel)", "language": "python", "name": "python3" }, "language_info": { "codemirror_mode": { "name": "ipython", "version": 3 }, "file_extension": ".py", "mimetype": "text/x-python", "name": "python", "nbconvert_exporter": "python", "pygments_lexer": "ipython3", "version": "3.11.8" } }, "nbformat": 4, "nbformat_minor": 4 }