{ "cells": [ { "cell_type": "code", "execution_count": 8, "metadata": {}, "outputs": [], "source": [ "import json\n", "import pandas as pd\n", "\n", "import os\n", "import random\n", "import shutil" ] }, { "cell_type": "code", "execution_count": 9, "metadata": {}, "outputs": [], "source": [ "with open('pred.json', 'r') as pred_file:\n", " predictions = json.load(pred_file)\n", "\n", "with open('GT.json', 'r') as gt_file:\n", " ground_truth = json.load(gt_file)" ] }, { "cell_type": "code", "execution_count": 10, "metadata": {}, "outputs": [ { "data": { "text/plain": [ "['./unique_images/img_20240703_064636_out.jpg',\n", " './unique_images/img_20240702_074550_out.jpg',\n", " './unique_images/img_20240620_055142_out.jpg',\n", " './unique_images/img_20240703_062335_out.jpg',\n", " './unique_images/img_20240703_060438_out.jpg',\n", " './unique_images/img_20240701_021942_out.jpg',\n", " './unique_images/img_20240620_061135_out.jpg',\n", " './unique_images/img_20240712_052209_out.jpg',\n", " './unique_images/img_20240708_021120_out.jpg',\n", " './unique_images/1_DSC5470.JPG',\n", " './unique_images/img_20240705_090326_out.jpg',\n", " './unique_images/img_20240620_064324_out.jpg',\n", " './unique_images/img_20240620_070604_out.jpg',\n", " './unique_images/1_DSC5400.JPG',\n", " './unique_images/_DSC1146.jpg',\n", " './unique_images/img_20240707_110503_out.jpg',\n", " './unique_images/img_20240701_071044_out.jpg',\n", " './unique_images/20240502_180948_12.jpg',\n", " './unique_images/img_20240708_141239_out.jpg',\n", " './unique_images/img_20240701_002229_out.jpg',\n", " './unique_images/img_20240712_125831_out.jpg',\n", " './unique_images/img_20240702_085738_out.jpg',\n", " './unique_images/img_20240715_031531_out.jpg',\n", " './unique_images/_DSC2034.jpg',\n", " './unique_images/img_20240710_090524_out.jpg',\n", " './unique_images/img_20240702_050024_out.jpg',\n", " './unique_images/img_20240705_075748_out.jpg',\n", " './unique_images/_DSC225606.jpg',\n", " './unique_images/img_20240621_074809_out.jpg',\n", " './unique_images/img_20240714_114651_out.jpg',\n", " './unique_images/img_20240708_120149_out.jpg',\n", " './unique_images/img_20240709_043019_out.jpg',\n", " './unique_images/img_20240705_084642_out.jpg',\n", " './unique_images/img_20240708_112929_out.jpg',\n", " './unique_images/KakaoTalk_20240501_224948237_12_23.jpg',\n", " './unique_images/img_20240712_082028_out.jpg',\n", " './unique_images/img_20240714_124221_out.jpg',\n", " './unique_images/img_20240702_020103_out.jpg',\n", " './unique_images/_DSC5669_03.jpg']" ] }, "execution_count": 10, "metadata": {}, "output_type": "execute_result" } ], "source": [ "# Convert JSON data to DataFrame\n", "df = pd.DataFrame(ground_truth)\n", "\n", "# Create unique_class column\n", "df['unique_class'] = df['damage_class'] + '_' + df['damage_level']\n", "\n", "\n", "prediction_df = pd.DataFrame(predictions)\n", "prediction_df['predicted_unique_class'] = prediction_df['damage_class'] + '_' + prediction_df['damage_level']\n", "\n", "# Get a random image for each unique class\n", "unique_classes = df['unique_class'].unique()\n", "random_images = df.groupby('unique_class').apply(lambda x: x.sample(1)).reset_index(drop=True)\n", "\n", "# Save random images to new directory\n", "output_dir = './unique_images'\n", "os.makedirs(output_dir, exist_ok=True)\n", "\n", "saved_images = []\n", "\n", "for _, row in random_images.iterrows():\n", " src = row['image_path']\n", " dest = os.path.join(output_dir, os.path.basename(src))\n", " shutil.copy(src, dest)\n", " saved_images.append(dest)\n", "\n", "saved_images" ] }, { "cell_type": "code", "execution_count": 12, "metadata": {}, "outputs": [ { "data": { "text/html": [ "
\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", " \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", " \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", " \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_pathunique_classpredicted_unique_class
0./TTA_TEST_SET/img_20240703_064636_out.jpgCOS_Arc_mark_level_1COS_Arc_mark_level_1
1./TTA_TEST_SET/img_20240702_074550_out.jpgCOS_Arc_mark_level_2COS_Arc_mark_level_2
2./TTA_TEST_SET/img_20240620_055142_out.jpgCOS_Arc_mark_level_3COS_Arc_mark_level_3
3./TTA_TEST_SET/img_20240703_062335_out.jpgCOS_Crack_level_1COS_Crack_level_1
4./TTA_TEST_SET/img_20240703_060438_out.jpgCOS_Crack_level_2COS_Crack_level_2
5./TTA_TEST_SET/img_20240701_021942_out.jpgCOS_Crack_level_3COS_Crack_level_3
6./TTA_TEST_SET/img_20240620_061135_out.jpgCOS_Fuse_link_corrosion_level_1COS_Fuse_link_corrosion_level_1
7./TTA_TEST_SET/img_20240712_052209_out.jpgCOS_Fuse_link_corrosion_level_2COS_Fuse_link_corrosion_level_2
8./TTA_TEST_SET/img_20240708_021120_out.jpgCOS_Fuse_link_corrosion_level_3COS_Fuse_link_corrosion_level_3
9./TTA_TEST_SET/1_DSC5470.JPGCOS_Normal_normalCOS_Normal_normal
10./TTA_TEST_SET/img_20240705_090326_out.jpgGS_Arc_mark_level_1GS_Arc_mark_level_1
11./TTA_TEST_SET/img_20240620_064324_out.jpgGS_Arc_mark_level_2GS_Arc_mark_level_2
12./TTA_TEST_SET/img_20240620_070604_out.jpgGS_Arc_mark_level_3GS_Arc_mark_level_3
13./TTA_TEST_SET/1_DSC5400.JPGGS_Normal_normalGS_Normal_normal
14./TTA_TEST_SET/_DSC1146.jpgInsulator_Installation_level_1Insulator_Installation_level_1
15./TTA_TEST_SET/img_20240707_110503_out.jpgInsulator_Installation_level_2Insulator_Installation_level_2
16./TTA_TEST_SET/img_20240701_071044_out.jpgInsulator_Installation_level_3Insulator_Installation_level_3
17./TTA_TEST_SET/20240502_180948_12.jpgInsulator_Normal_normalInsulator_Normal_normal
18./TTA_TEST_SET/img_20240708_141239_out.jpgLA_Arc_mark_level_1LA_Arc_mark_level_1
19./TTA_TEST_SET/img_20240701_002229_out.jpgLA_Arc_mark_level_2LA_Arc_mark_level_2
20./TTA_TEST_SET/img_20240712_125831_out.jpgLA_Arc_mark_level_3LA_Arc_mark_level_3
21./TTA_TEST_SET/img_20240702_085738_out.jpgLA_Crack_level_1LA_Crack_level_1
22./TTA_TEST_SET/img_20240715_031531_out.jpgLA_Crack_level_2LA_Crack_level_2
23./TTA_TEST_SET/_DSC2034.jpgLA_Crack_level_3LA_Crack_level_3
24./TTA_TEST_SET/img_20240710_090524_out.jpgLA_Installation_level_1LA_Installation_level_1
25./TTA_TEST_SET/img_20240702_050024_out.jpgLA_Installation_level_2LA_Installation_level_2
26./TTA_TEST_SET/img_20240705_075748_out.jpgLA_Installation_level_3LA_Installation_level_3
27./TTA_TEST_SET/_DSC225606.jpgLA_Normal_normalLA_Normal_normal
28./TTA_TEST_SET/img_20240621_074809_out.jpgLP_Crack_level_1LP_Crack_level_1
29./TTA_TEST_SET/img_20240714_114651_out.jpgLP_Crack_level_2LP_Crack_level_2
30./TTA_TEST_SET/img_20240708_120149_out.jpgLP_Crack_level_3LP_Crack_level_3
31./TTA_TEST_SET/img_20240709_043019_out.jpgLP_Installation_level_1LP_Installation_level_1
32./TTA_TEST_SET/img_20240705_084642_out.jpgLP_Installation_level_2LP_Installation_level_2
33./TTA_TEST_SET/img_20240708_112929_out.jpgLP_Installation_level_3LP_Installation_level_3
34./TTA_TEST_SET/KakaoTalk_20240501_224948237_12...LP_Normal_normalLP_Normal_normal
35./TTA_TEST_SET/img_20240712_082028_out.jpgPole_Crack_level_1Pole_Crack_level_1
36./TTA_TEST_SET/img_20240714_124221_out.jpgPole_Crack_level_2Pole_Crack_level_2
37./TTA_TEST_SET/img_20240702_020103_out.jpgPole_Crack_level_3Pole_Crack_level_3
38./TTA_TEST_SET/_DSC5669_03.jpgPole_Normal_normalPole_Normal_normal
\n", "
" ], "text/plain": [ " image_path \\\n", "0 ./TTA_TEST_SET/img_20240703_064636_out.jpg \n", "1 ./TTA_TEST_SET/img_20240702_074550_out.jpg \n", "2 ./TTA_TEST_SET/img_20240620_055142_out.jpg \n", "3 ./TTA_TEST_SET/img_20240703_062335_out.jpg \n", "4 ./TTA_TEST_SET/img_20240703_060438_out.jpg \n", "5 ./TTA_TEST_SET/img_20240701_021942_out.jpg \n", "6 ./TTA_TEST_SET/img_20240620_061135_out.jpg \n", "7 ./TTA_TEST_SET/img_20240712_052209_out.jpg \n", "8 ./TTA_TEST_SET/img_20240708_021120_out.jpg \n", "9 ./TTA_TEST_SET/1_DSC5470.JPG \n", "10 ./TTA_TEST_SET/img_20240705_090326_out.jpg \n", "11 ./TTA_TEST_SET/img_20240620_064324_out.jpg \n", "12 ./TTA_TEST_SET/img_20240620_070604_out.jpg \n", "13 ./TTA_TEST_SET/1_DSC5400.JPG \n", "14 ./TTA_TEST_SET/_DSC1146.jpg \n", "15 ./TTA_TEST_SET/img_20240707_110503_out.jpg \n", "16 ./TTA_TEST_SET/img_20240701_071044_out.jpg \n", "17 ./TTA_TEST_SET/20240502_180948_12.jpg \n", "18 ./TTA_TEST_SET/img_20240708_141239_out.jpg \n", "19 ./TTA_TEST_SET/img_20240701_002229_out.jpg \n", "20 ./TTA_TEST_SET/img_20240712_125831_out.jpg \n", "21 ./TTA_TEST_SET/img_20240702_085738_out.jpg \n", "22 ./TTA_TEST_SET/img_20240715_031531_out.jpg \n", "23 ./TTA_TEST_SET/_DSC2034.jpg \n", "24 ./TTA_TEST_SET/img_20240710_090524_out.jpg \n", "25 ./TTA_TEST_SET/img_20240702_050024_out.jpg \n", "26 ./TTA_TEST_SET/img_20240705_075748_out.jpg \n", "27 ./TTA_TEST_SET/_DSC225606.jpg \n", "28 ./TTA_TEST_SET/img_20240621_074809_out.jpg \n", "29 ./TTA_TEST_SET/img_20240714_114651_out.jpg \n", "30 ./TTA_TEST_SET/img_20240708_120149_out.jpg \n", "31 ./TTA_TEST_SET/img_20240709_043019_out.jpg \n", "32 ./TTA_TEST_SET/img_20240705_084642_out.jpg \n", "33 ./TTA_TEST_SET/img_20240708_112929_out.jpg \n", "34 ./TTA_TEST_SET/KakaoTalk_20240501_224948237_12... \n", "35 ./TTA_TEST_SET/img_20240712_082028_out.jpg \n", "36 ./TTA_TEST_SET/img_20240714_124221_out.jpg \n", "37 ./TTA_TEST_SET/img_20240702_020103_out.jpg \n", "38 ./TTA_TEST_SET/_DSC5669_03.jpg \n", "\n", " unique_class predicted_unique_class \n", "0 COS_Arc_mark_level_1 COS_Arc_mark_level_1 \n", "1 COS_Arc_mark_level_2 COS_Arc_mark_level_2 \n", "2 COS_Arc_mark_level_3 COS_Arc_mark_level_3 \n", "3 COS_Crack_level_1 COS_Crack_level_1 \n", "4 COS_Crack_level_2 COS_Crack_level_2 \n", "5 COS_Crack_level_3 COS_Crack_level_3 \n", "6 COS_Fuse_link_corrosion_level_1 COS_Fuse_link_corrosion_level_1 \n", "7 COS_Fuse_link_corrosion_level_2 COS_Fuse_link_corrosion_level_2 \n", "8 COS_Fuse_link_corrosion_level_3 COS_Fuse_link_corrosion_level_3 \n", "9 COS_Normal_normal COS_Normal_normal \n", "10 GS_Arc_mark_level_1 GS_Arc_mark_level_1 \n", "11 GS_Arc_mark_level_2 GS_Arc_mark_level_2 \n", "12 GS_Arc_mark_level_3 GS_Arc_mark_level_3 \n", "13 GS_Normal_normal GS_Normal_normal \n", "14 Insulator_Installation_level_1 Insulator_Installation_level_1 \n", "15 Insulator_Installation_level_2 Insulator_Installation_level_2 \n", "16 Insulator_Installation_level_3 Insulator_Installation_level_3 \n", "17 Insulator_Normal_normal Insulator_Normal_normal \n", "18 LA_Arc_mark_level_1 LA_Arc_mark_level_1 \n", "19 LA_Arc_mark_level_2 LA_Arc_mark_level_2 \n", "20 LA_Arc_mark_level_3 LA_Arc_mark_level_3 \n", "21 LA_Crack_level_1 LA_Crack_level_1 \n", "22 LA_Crack_level_2 LA_Crack_level_2 \n", "23 LA_Crack_level_3 LA_Crack_level_3 \n", "24 LA_Installation_level_1 LA_Installation_level_1 \n", "25 LA_Installation_level_2 LA_Installation_level_2 \n", "26 LA_Installation_level_3 LA_Installation_level_3 \n", "27 LA_Normal_normal LA_Normal_normal \n", "28 LP_Crack_level_1 LP_Crack_level_1 \n", "29 LP_Crack_level_2 LP_Crack_level_2 \n", "30 LP_Crack_level_3 LP_Crack_level_3 \n", "31 LP_Installation_level_1 LP_Installation_level_1 \n", "32 LP_Installation_level_2 LP_Installation_level_2 \n", "33 LP_Installation_level_3 LP_Installation_level_3 \n", "34 LP_Normal_normal LP_Normal_normal \n", "35 Pole_Crack_level_1 Pole_Crack_level_1 \n", "36 Pole_Crack_level_2 Pole_Crack_level_2 \n", "37 Pole_Crack_level_3 Pole_Crack_level_3 \n", "38 Pole_Normal_normal Pole_Normal_normal " ] }, "execution_count": 12, "metadata": {}, "output_type": "execute_result" } ], "source": [ "# Print the mapping of each image to its class\n", "image_class_mapping = random_images[['image_path', 'unique_class']]\n", "image_class_mapping\n", "\n", "# Merge the ground truth and prediction dataframes on image_path\n", "merged_df = pd.merge(random_images, prediction_df, on='image_path', suffixes=('_gt', '_pred'))\n", "\n", "# Select relevant columns to display\n", "display_df = merged_df[['image_path', 'unique_class', 'predicted_unique_class']]\n", "\n", "display_df\n" ] }, { "cell_type": "code", "execution_count": 15, "metadata": {}, "outputs": [ { "data": { "text/plain": [ "'./image_path_with_gt_and_pred_classes.xlsx'" ] }, "execution_count": 15, "metadata": {}, "output_type": "execute_result" } ], "source": [ "# Save the display dataframe to an Excel file\n", "excel_file_path = './image_path_with_gt_and_pred_classes.xlsx'\n", "display_df.to_excel(excel_file_path, index=False)\n", "\n", "excel_file_path" ] }, { "cell_type": "code", "execution_count": null, "metadata": {}, "outputs": [], "source": [] } ], "metadata": { "kernelspec": { "display_name": "mmpretrain", "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.8.0" }, "orig_nbformat": 4 }, "nbformat": 4, "nbformat_minor": 2 }