fiftyone_visualization.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 2,
   "id": "daf4b548-83a8-44fa-9da8-a11eea6128ac",
   "metadata": {},
   "outputs": [],
   "source": [
    "import numpy as np\n",
    "import PIL\n",
    "import umap\n",
    "import pandas as pd\n",
    "import json\n",
    "import glob\n",
    "from pytorch_metric_learning import distances, losses, miners, reducers\n",
    "import torch.nn as nn\n",
    "import os\n",
    "import matplotlib.pyplot as plt\n",
    "import torch\n",
    "import torchvision\n",
    "from torchvision import models\n",
    "import torch\n",
    "import torch.nn as nn\n",
    "import torch.nn.functional as F\n",
    "import torch.optim as optim\n",
    "from torchvision import datasets, transforms\n",
    "from torch.utils.data.sampler import SubsetRandomSampler\n",
    "from tqdm import tqdm\n",
    "from pytorch_metric_learning import losses, testers\n",
    "from pytorch_metric_learning.utils.accuracy_calculator import AccuracyCalculator\n",
    "from pytorch_metric_learning.distances import LpDistance,CosineSimilarity\n",
    "from pytorch_metric_learning.utils.inference import CustomKNN\n",
    "%matplotlib inline\n",
    "import torch\n",
    "import torch.nn as nn\n",
    "from matplotlib import pyplot as plt\n",
    "import numpy as np\n",
    "import torchvision\n",
    "import torchvision.datasets as datasets\n",
    "import torchvision.models as models\n",
    "from torchvision import transforms\n",
    "import torch.optim as optim\n",
    "import time\n",
    "import tqdm as tqdm\n",
    "from torch.autograd import Variable"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "id": "60399700-0034-4dea-ae5a-ef00ab63829f",
   "metadata": {},
   "outputs": [],
   "source": [
    "config = {\n",
    "    'name':'cub_triplet_loss_epshn_resnet50_sgd_aug',\n",
    "    'dataset':'CUB_200_2011',\n",
    "    'random_seed':42,\n",
    "    'model_architecture':'resnet50',\n",
    "    'embedding_dim':128,\n",
    "    'distance':'cosine',\n",
    "    'image_height':224,\n",
    "    'image_width':224,\n",
    "    'train_test_split':0.2,\n",
    "    'class_split':0.1,\n",
    "    'batch_size':128,\n",
    "    'optimizer':'sgd',\n",
    "    'learning_rate':0.001,\n",
    "    'num_epochs':100,\n",
    "    'loss':'NTXentLoss',\n",
    "    'miner':'epshn',\n",
    "    'reducer':0,\n",
    "    'metric':'precision_at_1',\n",
    "    'model_save_path':'models/cub_triplet_loss_epshn_resnet50_sgd_aug',\n",
    "    'temperature': 0.1\n",
    "}"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "id": "8ef21e33-9382-4ae7-85b3-5a78d77a0b79",
   "metadata": {},
   "outputs": [],
   "source": [
    "np.random.seed(config['random_seed'])\n",
    "torch.manual_seed(config['random_seed'])\n",
    "torch.cuda.manual_seed(config['random_seed'])\n",
    "torch.backends.cudnn.deterministic = False"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "id": "1dc45862-c9a3-4bca-8072-69f4f9807bd8",
   "metadata": {},
   "outputs": [],
   "source": [
    "# print(models.resnet18())\n",
    "class ResNetFeatrueExtractor50(nn.Module):\n",
    "    def __init__(self, pretrained = True):\n",
    "        super(ResNetFeatrueExtractor50, self).__init__()\n",
    "        self.model = models.resnet50(pretrained=pretrained)\n",
    "        self.model.fc = nn.Linear(2048, config['embedding_dim'])\n",
    "\n",
    "    def forward(self, x):\n",
    "        x = self.model(x)\n",
    "        return F.normalize(x)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 6,
   "id": "7a4263f7-7a14-4f7b-8061-abae2c145e99",
   "metadata": {},
   "outputs": [],
   "source": [
    "device = torch.device(\"cuda\")\n",
    "\n",
    "mean = [0.485, 0.456, 0.406]\n",
    "std = [0.229, 0.224, 0.225]\n",
    "\n",
    "# create train and test transforms\n",
    "transform = transforms.Compose(\n",
    "    [\n",
    "        transforms.Resize((config['image_height'], config['image_width'])),\n",
    "        transforms.ToTensor(),\n",
    "        transforms.Normalize(mean, std),\n",
    "    ]\n",
    ")\n",
    "\n",
    "batch_size = 128"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 7,
   "id": "cb69aa0d-895c-4a2c-a989-6a9f8baeb0eb",
   "metadata": {},
   "outputs": [],
   "source": [
    "def tra_transforms(imgsize, RGBmean, RGBstdv):\n",
    "    return transforms.Compose([transforms.Resize(int(imgsize*1.1)),\n",
    "                                 transforms.RandomCrop(imgsize),\n",
    "                                 transforms.RandomHorizontalFlip(),\n",
    "                                 transforms.ToTensor(),\n",
    "                                 transforms.Normalize(RGBmean, RGBstdv)])\n",
    "\n",
    "def eva_transforms(imgsize, RGBmean, RGBstdv):\n",
    "    return transforms.Compose([transforms.Resize(imgsize),\n",
    "                                 transforms.CenterCrop(imgsize),\n",
    "                                 transforms.ToTensor(),\n",
    "                                 transforms.Normalize(RGBmean, RGBstdv)])"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 19,
   "id": "6a75acc0-7b3f-4ba4-9a9c-bf4b6cf798d1",
   "metadata": {},
   "outputs": [],
   "source": [
    "train_transform = tra_transforms(224,mean,std)\n",
    "test_transform = eva_transforms(224,mean,std)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 9,
   "id": "47af7c40-278a-49b8-b6c2-135cd07d16a8",
   "metadata": {},
   "outputs": [],
   "source": [
    "with open('CUB_200_2011/classes.txt','r') as f:\n",
    "    classes = f.readlines()\n",
    "classes = [i.replace('\\n','') for i in classes]\n",
    "classes = [i.split(' ')[1] for i in classes]\n",
    "class_dict = {k:v for k,v in zip(classes,range(200))}"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 10,
   "id": "6a7707be-29d4-45ba-8cb6-ad4cc6d16251",
   "metadata": {},
   "outputs": [],
   "source": [
    "image_paths = []\n",
    "labels = []\n",
    "for folder_path,i in class_dict.items():\n",
    "    folder_images = glob.glob('CUB_200_2011/images/'+'/'+str(folder_path)+'/*')\n",
    "    image_paths.extend(folder_images)\n",
    "    labels.extend([i]*len(folder_images))"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 11,
   "id": "191c73af-78c3-46a3-ba21-74eb13774f95",
   "metadata": {},
   "outputs": [],
   "source": [
    "from sklearn.model_selection import train_test_split\n",
    "X_train, X_test, y_train, y_test = train_test_split(image_paths, labels, test_size=config['train_test_split'],\n",
    "                                                    stratify=labels,\n",
    "                                                    random_state=config['random_seed'])"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 12,
   "id": "e85b4f5e-085b-47b2-b870-d790d1c57700",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "9430 2358 9430 2358\n"
     ]
    }
   ],
   "source": [
    "print(len(X_train), len(X_test), len(y_train), len(y_test))"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 13,
   "id": "0f4d0e3c-a668-4e7e-a070-b26db152f474",
   "metadata": {},
   "outputs": [],
   "source": [
    "model = torch.load('models/cub_triplet_loss_epshn_resnet50_sgd_aug_180.pth',map_location='cuda')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 14,
   "id": "11bf15c7-4d79-43b8-a5b1-ba987cc36217",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "ResNetFeatrueExtractor50(\n",
       "  (model): ResNet(\n",
       "    (conv1): Conv2d(3, 64, kernel_size=(7, 7), stride=(2, 2), padding=(3, 3), bias=False)\n",
       "    (bn1): BatchNorm2d(64, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
       "    (relu): ReLU(inplace=True)\n",
       "    (maxpool): MaxPool2d(kernel_size=3, stride=2, padding=1, dilation=1, ceil_mode=False)\n",
       "    (layer1): Sequential(\n",
       "      (0): Bottleneck(\n",
       "        (conv1): Conv2d(64, 64, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
       "        (bn1): BatchNorm2d(64, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
       "        (conv2): Conv2d(64, 64, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False)\n",
       "        (bn2): BatchNorm2d(64, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
       "        (conv3): Conv2d(64, 256, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
       "        (bn3): BatchNorm2d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
       "        (relu): ReLU(inplace=True)\n",
       "        (downsample): Sequential(\n",
       "          (0): Conv2d(64, 256, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
       "          (1): BatchNorm2d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
       "        )\n",
       "      )\n",
       "      (1): Bottleneck(\n",
       "        (conv1): Conv2d(256, 64, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
       "        (bn1): BatchNorm2d(64, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
       "        (conv2): Conv2d(64, 64, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False)\n",
       "        (bn2): BatchNorm2d(64, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
       "        (conv3): Conv2d(64, 256, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
       "        (bn3): BatchNorm2d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
       "        (relu): ReLU(inplace=True)\n",
       "      )\n",
       "      (2): Bottleneck(\n",
       "        (conv1): Conv2d(256, 64, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
       "        (bn1): BatchNorm2d(64, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
       "        (conv2): Conv2d(64, 64, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False)\n",
       "        (bn2): BatchNorm2d(64, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
       "        (conv3): Conv2d(64, 256, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
       "        (bn3): BatchNorm2d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
       "        (relu): ReLU(inplace=True)\n",
       "      )\n",
       "    )\n",
       "    (layer2): Sequential(\n",
       "      (0): Bottleneck(\n",
       "        (conv1): Conv2d(256, 128, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
       "        (bn1): BatchNorm2d(128, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
       "        (conv2): Conv2d(128, 128, kernel_size=(3, 3), stride=(2, 2), padding=(1, 1), bias=False)\n",
       "        (bn2): BatchNorm2d(128, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
       "        (conv3): Conv2d(128, 512, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
       "        (bn3): BatchNorm2d(512, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
       "        (relu): ReLU(inplace=True)\n",
       "        (downsample): Sequential(\n",
       "          (0): Conv2d(256, 512, kernel_size=(1, 1), stride=(2, 2), bias=False)\n",
       "          (1): BatchNorm2d(512, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
       "        )\n",
       "      )\n",
       "      (1): Bottleneck(\n",
       "        (conv1): Conv2d(512, 128, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
       "        (bn1): BatchNorm2d(128, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
       "        (conv2): Conv2d(128, 128, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False)\n",
       "        (bn2): BatchNorm2d(128, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
       "        (conv3): Conv2d(128, 512, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
       "        (bn3): BatchNorm2d(512, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
       "        (relu): ReLU(inplace=True)\n",
       "      )\n",
       "      (2): Bottleneck(\n",
       "        (conv1): Conv2d(512, 128, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
       "        (bn1): BatchNorm2d(128, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
       "        (conv2): Conv2d(128, 128, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False)\n",
       "        (bn2): BatchNorm2d(128, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
       "        (conv3): Conv2d(128, 512, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
       "        (bn3): BatchNorm2d(512, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
       "        (relu): ReLU(inplace=True)\n",
       "      )\n",
       "      (3): Bottleneck(\n",
       "        (conv1): Conv2d(512, 128, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
       "        (bn1): BatchNorm2d(128, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
       "        (conv2): Conv2d(128, 128, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False)\n",
       "        (bn2): BatchNorm2d(128, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
       "        (conv3): Conv2d(128, 512, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
       "        (bn3): BatchNorm2d(512, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
       "        (relu): ReLU(inplace=True)\n",
       "      )\n",
       "    )\n",
       "    (layer3): Sequential(\n",
       "      (0): Bottleneck(\n",
       "        (conv1): Conv2d(512, 256, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
       "        (bn1): BatchNorm2d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
       "        (conv2): Conv2d(256, 256, kernel_size=(3, 3), stride=(2, 2), padding=(1, 1), bias=False)\n",
       "        (bn2): BatchNorm2d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
       "        (conv3): Conv2d(256, 1024, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
       "        (bn3): BatchNorm2d(1024, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
       "        (relu): ReLU(inplace=True)\n",
       "        (downsample): Sequential(\n",
       "          (0): Conv2d(512, 1024, kernel_size=(1, 1), stride=(2, 2), bias=False)\n",
       "          (1): BatchNorm2d(1024, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
       "        )\n",
       "      )\n",
       "      (1): Bottleneck(\n",
       "        (conv1): Conv2d(1024, 256, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
       "        (bn1): BatchNorm2d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
       "        (conv2): Conv2d(256, 256, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False)\n",
       "        (bn2): BatchNorm2d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
       "        (conv3): Conv2d(256, 1024, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
       "        (bn3): BatchNorm2d(1024, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
       "        (relu): ReLU(inplace=True)\n",
       "      )\n",
       "      (2): Bottleneck(\n",
       "        (conv1): Conv2d(1024, 256, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
       "        (bn1): BatchNorm2d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
       "        (conv2): Conv2d(256, 256, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False)\n",
       "        (bn2): BatchNorm2d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
       "        (conv3): Conv2d(256, 1024, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
       "        (bn3): BatchNorm2d(1024, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
       "        (relu): ReLU(inplace=True)\n",
       "      )\n",
       "      (3): Bottleneck(\n",
       "        (conv1): Conv2d(1024, 256, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
       "        (bn1): BatchNorm2d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
       "        (conv2): Conv2d(256, 256, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False)\n",
       "        (bn2): BatchNorm2d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
       "        (conv3): Conv2d(256, 1024, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
       "        (bn3): BatchNorm2d(1024, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
       "        (relu): ReLU(inplace=True)\n",
       "      )\n",
       "      (4): Bottleneck(\n",
       "        (conv1): Conv2d(1024, 256, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
       "        (bn1): BatchNorm2d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
       "        (conv2): Conv2d(256, 256, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False)\n",
       "        (bn2): BatchNorm2d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
       "        (conv3): Conv2d(256, 1024, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
       "        (bn3): BatchNorm2d(1024, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
       "        (relu): ReLU(inplace=True)\n",
       "      )\n",
       "      (5): Bottleneck(\n",
       "        (conv1): Conv2d(1024, 256, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
       "        (bn1): BatchNorm2d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
       "        (conv2): Conv2d(256, 256, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False)\n",
       "        (bn2): BatchNorm2d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
       "        (conv3): Conv2d(256, 1024, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
       "        (bn3): BatchNorm2d(1024, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
       "        (relu): ReLU(inplace=True)\n",
       "      )\n",
       "    )\n",
       "    (layer4): Sequential(\n",
       "      (0): Bottleneck(\n",
       "        (conv1): Conv2d(1024, 512, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
       "        (bn1): BatchNorm2d(512, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
       "        (conv2): Conv2d(512, 512, kernel_size=(3, 3), stride=(2, 2), padding=(1, 1), bias=False)\n",
       "        (bn2): BatchNorm2d(512, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
       "        (conv3): Conv2d(512, 2048, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
       "        (bn3): BatchNorm2d(2048, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
       "        (relu): ReLU(inplace=True)\n",
       "        (downsample): Sequential(\n",
       "          (0): Conv2d(1024, 2048, kernel_size=(1, 1), stride=(2, 2), bias=False)\n",
       "          (1): BatchNorm2d(2048, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
       "        )\n",
       "      )\n",
       "      (1): Bottleneck(\n",
       "        (conv1): Conv2d(2048, 512, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
       "        (bn1): BatchNorm2d(512, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
       "        (conv2): Conv2d(512, 512, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False)\n",
       "        (bn2): BatchNorm2d(512, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
       "        (conv3): Conv2d(512, 2048, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
       "        (bn3): BatchNorm2d(2048, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
       "        (relu): ReLU(inplace=True)\n",
       "      )\n",
       "      (2): Bottleneck(\n",
       "        (conv1): Conv2d(2048, 512, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
       "        (bn1): BatchNorm2d(512, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
       "        (conv2): Conv2d(512, 512, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False)\n",
       "        (bn2): BatchNorm2d(512, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
       "        (conv3): Conv2d(512, 2048, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
       "        (bn3): BatchNorm2d(2048, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
       "        (relu): ReLU(inplace=True)\n",
       "      )\n",
       "    )\n",
       "    (avgpool): AdaptiveAvgPool2d(output_size=(1, 1))\n",
       "    (fc): Linear(in_features=2048, out_features=128, bias=True)\n",
       "  )\n",
       ")"
      ]
     },
     "execution_count": 14,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "model.eval()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 16,
   "id": "601dd013-34cb-4869-9d1e-39647989c7d8",
   "metadata": {},
   "outputs": [],
   "source": [
    "def extract_category(path):\n",
    "    return path.split('/')[-2]\n",
    "    \n",
    "sorted_test_paths = sorted(X_test, key=extract_category)\n",
    "sorted_train_paths = sorted(X_train, key=extract_category)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 17,
   "id": "39c32286-4bd6-49dc-9b4a-5bd49a31d846",
   "metadata": {},
   "outputs": [],
   "source": [
    "rev_class_dict = {v:k for k,v in class_dict.items()}"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 1,
   "id": "cea25996-25e0-485b-8f11-53a86a08fd75",
   "metadata": {
    "scrolled": true
   },
   "outputs": [],
   "source": [
    "import fiftyone as fo\n",
    "import numpy as np\n",
    "import fiftyone.brain as fob\n",
    "import fiftyone.zoo as foz\n",
    "from fiftyone import ViewField as F"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "id": "c4938bd8-1a49-453f-a987-348f92987e7c",
   "metadata": {
    "scrolled": true
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": []
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      " 100% |█████████████| 11788/11788 [1.9s elapsed, 0s remaining, 6.4K samples/s]         \n",
      "Name:        CUB-200\n",
      "Media type:  image\n",
      "Num samples: 11788\n",
      "Persistent:  False\n",
      "Tags:        []\n",
      "Sample fields:\n",
      "    id:           fiftyone.core.fields.ObjectIdField\n",
      "    filepath:     fiftyone.core.fields.StringField\n",
      "    tags:         fiftyone.core.fields.ListField(fiftyone.core.fields.StringField)\n",
      "    metadata:     fiftyone.core.fields.EmbeddedDocumentField(fiftyone.core.metadata.ImageMetadata)\n",
      "    ground_truth: fiftyone.core.fields.EmbeddedDocumentField(fiftyone.core.labels.Classification)\n",
      "[<Sample: {\n",
      "    'id': '6639187385c213e6d9a3e5be',\n",
      "    'media_type': 'image',\n",
      "    'filepath': '/home/pranav/embedding_classification/CUB_200_2011/images/001.Black_footed_Albatross/Black_Footed_Albatross_0001_796111.jpg',\n",
      "    'tags': [],\n",
      "    'metadata': None,\n",
      "    'ground_truth': <Classification: {\n",
      "        'id': '6639187385c213e6d9a3e5bd',\n",
      "        'tags': [],\n",
      "        'label': '001.Black_footed_Albatross',\n",
      "        'confidence': None,\n",
      "        'logits': None,\n",
      "    }>,\n",
      "}>, <Sample: {\n",
      "    'id': '6639187385c213e6d9a3e5c7',\n",
      "    'media_type': 'image',\n",
      "    'filepath': '/home/pranav/embedding_classification/CUB_200_2011/images/001.Black_footed_Albatross/Black_Footed_Albatross_0002_55.jpg',\n",
      "    'tags': [],\n",
      "    'metadata': None,\n",
      "    'ground_truth': <Classification: {\n",
      "        'id': '6639187385c213e6d9a3e5bf',\n",
      "        'tags': [],\n",
      "        'label': '001.Black_footed_Albatross',\n",
      "        'confidence': None,\n",
      "        'logits': None,\n",
      "    }>,\n",
      "}>, <Sample: {\n",
      "    'id': '6639187385c213e6d9a3e5c8',\n",
      "    'media_type': 'image',\n",
      "    'filepath': '/home/pranav/embedding_classification/CUB_200_2011/images/001.Black_footed_Albatross/Black_Footed_Albatross_0003_796136.jpg',\n",
      "    'tags': [],\n",
      "    'metadata': None,\n",
      "    'ground_truth': <Classification: {\n",
      "        'id': '6639187385c213e6d9a3e5c0',\n",
      "        'tags': [],\n",
      "        'label': '001.Black_footed_Albatross',\n",
      "        'confidence': None,\n",
      "        'logits': None,\n",
      "    }>,\n",
      "}>]\n"
     ]
    }
   ],
   "source": [
    "name = \"CUB-200\"\n",
    "dataset_dir = \"CUB_200_2011/images/\"\n",
    "# Create the dataset\n",
    "dataset = fo.Dataset.from_dir(\n",
    "    dataset_dir=dataset_dir,\n",
    "    dataset_type=fo.types.ImageClassificationDirectoryTree,\n",
    "    name=name,\n",
    ")\n",
    "\n",
    "# View summary info about the dataset\n",
    "print(dataset)\n",
    "\n",
    "# Print the first few samples in the dataset\n",
    "print(dataset.head())"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "id": "21fccead-abfc-4f77-b4c1-a3c98784cf2d",
   "metadata": {},
   "outputs": [],
   "source": [
    "with open('CUB_200_2011/classes.txt','r') as f:\n",
    "    classes = f.readlines()\n",
    "classes = [i.replace('\\n','') for i in classes]\n",
    "classes = [i.split(' ')[1] for i in classes]\n",
    "class_dict = {k:v for k,v in zip(classes,range(200))}"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 6,
   "id": "9bc535e2-3952-49d3-a836-d89ee2b29911",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "['/home/pranav/embedding_classification/CUB_200_2011/images/001.Black_footed_Albatross/Black_Footed_Albatross_0001_796111.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/001.Black_footed_Albatross/Black_Footed_Albatross_0002_55.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/001.Black_footed_Albatross/Black_Footed_Albatross_0003_796136.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/001.Black_footed_Albatross/Black_Footed_Albatross_0005_796090.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/001.Black_footed_Albatross/Black_Footed_Albatross_0006_796065.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/001.Black_footed_Albatross/Black_Footed_Albatross_0007_796138.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/001.Black_footed_Albatross/Black_Footed_Albatross_0008_796083.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/001.Black_footed_Albatross/Black_Footed_Albatross_0009_34.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/001.Black_footed_Albatross/Black_Footed_Albatross_0010_796097.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/001.Black_footed_Albatross/Black_Footed_Albatross_0014_89.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/001.Black_footed_Albatross/Black_Footed_Albatross_0016_796067.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/001.Black_footed_Albatross/Black_Footed_Albatross_0017_796098.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/001.Black_footed_Albatross/Black_Footed_Albatross_0019_796104.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/001.Black_footed_Albatross/Black_Footed_Albatross_0023_796059.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/001.Black_footed_Albatross/Black_Footed_Albatross_0024_796089.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/001.Black_footed_Albatross/Black_Footed_Albatross_0025_796057.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/001.Black_footed_Albatross/Black_Footed_Albatross_0026_796095.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/001.Black_footed_Albatross/Black_Footed_Albatross_0031_100.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/001.Black_footed_Albatross/Black_Footed_Albatross_0032_796115.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/001.Black_footed_Albatross/Black_Footed_Albatross_0033_796086.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/001.Black_footed_Albatross/Black_Footed_Albatross_0035_796140.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/001.Black_footed_Albatross/Black_Footed_Albatross_0036_796127.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/001.Black_footed_Albatross/Black_Footed_Albatross_0037_796120.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/001.Black_footed_Albatross/Black_Footed_Albatross_0038_212.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/001.Black_footed_Albatross/Black_Footed_Albatross_0039_796132.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/001.Black_footed_Albatross/Black_Footed_Albatross_0040_796066.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/001.Black_footed_Albatross/Black_Footed_Albatross_0041_796108.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/001.Black_footed_Albatross/Black_Footed_Albatross_0042_796071.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/001.Black_footed_Albatross/Black_Footed_Albatross_0045_796129.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/001.Black_footed_Albatross/Black_Footed_Albatross_0046_18.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/001.Black_footed_Albatross/Black_Footed_Albatross_0047_796064.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/001.Black_footed_Albatross/Black_Footed_Albatross_0049_796063.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/001.Black_footed_Albatross/Black_Footed_Albatross_0050_796125.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/001.Black_footed_Albatross/Black_Footed_Albatross_0051_796103.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/001.Black_footed_Albatross/Black_Footed_Albatross_0053_796109.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/001.Black_footed_Albatross/Black_Footed_Albatross_0056_796078.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/001.Black_footed_Albatross/Black_Footed_Albatross_0057_796106.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/001.Black_footed_Albatross/Black_Footed_Albatross_0058_796074.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/001.Black_footed_Albatross/Black_Footed_Albatross_0060_796076.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/001.Black_footed_Albatross/Black_Footed_Albatross_0061_796082.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/001.Black_footed_Albatross/Black_Footed_Albatross_0063_796141.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/001.Black_footed_Albatross/Black_Footed_Albatross_0064_796101.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/001.Black_footed_Albatross/Black_Footed_Albatross_0065_796068.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/001.Black_footed_Albatross/Black_Footed_Albatross_0067_170.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/001.Black_footed_Albatross/Black_Footed_Albatross_0068_796135.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/001.Black_footed_Albatross/Black_Footed_Albatross_0069_796139.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/001.Black_footed_Albatross/Black_Footed_Albatross_0071_796113.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/001.Black_footed_Albatross/Black_Footed_Albatross_0074_59.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/001.Black_footed_Albatross/Black_Footed_Albatross_0076_417.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/001.Black_footed_Albatross/Black_Footed_Albatross_0077_796114.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/001.Black_footed_Albatross/Black_Footed_Albatross_0078_796126.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/001.Black_footed_Albatross/Black_Footed_Albatross_0079_796122.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/001.Black_footed_Albatross/Black_Footed_Albatross_0080_796096.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/001.Black_footed_Albatross/Black_Footed_Albatross_0081_426.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/001.Black_footed_Albatross/Black_Footed_Albatross_0082_796121.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/001.Black_footed_Albatross/Black_Footed_Albatross_0085_92.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/001.Black_footed_Albatross/Black_Footed_Albatross_0086_796062.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/001.Black_footed_Albatross/Black_Footed_Albatross_0088_796133.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/001.Black_footed_Albatross/Black_Footed_Albatross_0089_796069.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/001.Black_footed_Albatross/Black_Footed_Albatross_0090_796077.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/002.Laysan_Albatross/Laysan_Albatross_0001_545.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/002.Laysan_Albatross/Laysan_Albatross_0002_1027.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/002.Laysan_Albatross/Laysan_Albatross_0003_1033.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/002.Laysan_Albatross/Laysan_Albatross_0004_930.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/002.Laysan_Albatross/Laysan_Albatross_0005_565.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/002.Laysan_Albatross/Laysan_Albatross_0006_702.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/002.Laysan_Albatross/Laysan_Albatross_0012_696.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/002.Laysan_Albatross/Laysan_Albatross_0013_910.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/002.Laysan_Albatross/Laysan_Albatross_0017_614.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/002.Laysan_Albatross/Laysan_Albatross_0018_492.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/002.Laysan_Albatross/Laysan_Albatross_0021_737.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/002.Laysan_Albatross/Laysan_Albatross_0023_1059.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/002.Laysan_Albatross/Laysan_Albatross_0025_571.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/002.Laysan_Albatross/Laysan_Albatross_0028_643.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/002.Laysan_Albatross/Laysan_Albatross_0029_482.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/002.Laysan_Albatross/Laysan_Albatross_0033_658.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/002.Laysan_Albatross/Laysan_Albatross_0034_628.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/002.Laysan_Albatross/Laysan_Albatross_0035_876.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/002.Laysan_Albatross/Laysan_Albatross_0037_699.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/002.Laysan_Albatross/Laysan_Albatross_0039_924.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/002.Laysan_Albatross/Laysan_Albatross_0040_472.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/002.Laysan_Albatross/Laysan_Albatross_0042_801.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/002.Laysan_Albatross/Laysan_Albatross_0044_784.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/002.Laysan_Albatross/Laysan_Albatross_0047_619.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/002.Laysan_Albatross/Laysan_Albatross_0049_918.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/002.Laysan_Albatross/Laysan_Albatross_0050_870.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/002.Laysan_Albatross/Laysan_Albatross_0051_1020.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/002.Laysan_Albatross/Laysan_Albatross_0053_543.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/002.Laysan_Albatross/Laysan_Albatross_0055_570.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/002.Laysan_Albatross/Laysan_Albatross_0056_500.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/002.Laysan_Albatross/Laysan_Albatross_0058_637.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/002.Laysan_Albatross/Laysan_Albatross_0059_488.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/002.Laysan_Albatross/Laysan_Albatross_0060_777.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/002.Laysan_Albatross/Laysan_Albatross_0061_563.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/002.Laysan_Albatross/Laysan_Albatross_0064_674.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/002.Laysan_Albatross/Laysan_Albatross_0065_809.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/002.Laysan_Albatross/Laysan_Albatross_0068_726.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/002.Laysan_Albatross/Laysan_Albatross_0070_788.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/002.Laysan_Albatross/Laysan_Albatross_0071_792.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/002.Laysan_Albatross/Laysan_Albatross_0073_927.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/002.Laysan_Albatross/Laysan_Albatross_0075_668.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/002.Laysan_Albatross/Laysan_Albatross_0076_671.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/002.Laysan_Albatross/Laysan_Albatross_0079_506.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/002.Laysan_Albatross/Laysan_Albatross_0080_821.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/002.Laysan_Albatross/Laysan_Albatross_0081_800.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/002.Laysan_Albatross/Laysan_Albatross_0082_524.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/002.Laysan_Albatross/Laysan_Albatross_0083_756.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/002.Laysan_Albatross/Laysan_Albatross_0085_564.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/002.Laysan_Albatross/Laysan_Albatross_0088_883.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/002.Laysan_Albatross/Laysan_Albatross_0091_602.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/002.Laysan_Albatross/Laysan_Albatross_0092_834.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/002.Laysan_Albatross/Laysan_Albatross_0093_725.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/002.Laysan_Albatross/Laysan_Albatross_0094_1013.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/002.Laysan_Albatross/Laysan_Albatross_0096_673.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/002.Laysan_Albatross/Laysan_Albatross_0098_621.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/002.Laysan_Albatross/Laysan_Albatross_0099_869.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/002.Laysan_Albatross/Laysan_Albatross_0100_735.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/002.Laysan_Albatross/Laysan_Albatross_0102_611.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/002.Laysan_Albatross/Laysan_Albatross_0103_504.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/002.Laysan_Albatross/Laysan_Albatross_0104_630.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/003.Sooty_Albatross/Sooty_Albatross_0001_1071.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/003.Sooty_Albatross/Sooty_Albatross_0002_796395.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/003.Sooty_Albatross/Sooty_Albatross_0003_1078.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/003.Sooty_Albatross/Sooty_Albatross_0004_796366.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/003.Sooty_Albatross/Sooty_Albatross_0005_796342.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/003.Sooty_Albatross/Sooty_Albatross_0006_796390.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/003.Sooty_Albatross/Sooty_Albatross_0007_796372.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/003.Sooty_Albatross/Sooty_Albatross_0010_796355.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/003.Sooty_Albatross/Sooty_Albatross_0013_796402.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/003.Sooty_Albatross/Sooty_Albatross_0014_796373.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/003.Sooty_Albatross/Sooty_Albatross_0016_1075.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/003.Sooty_Albatross/Sooty_Albatross_0017_796349.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/003.Sooty_Albatross/Sooty_Albatross_0019_796391.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/003.Sooty_Albatross/Sooty_Albatross_0020_796359.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/003.Sooty_Albatross/Sooty_Albatross_0021_796339.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/003.Sooty_Albatross/Sooty_Albatross_0022_796398.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/003.Sooty_Albatross/Sooty_Albatross_0023_796401.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/003.Sooty_Albatross/Sooty_Albatross_0024_1161.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/003.Sooty_Albatross/Sooty_Albatross_0025_796361.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/003.Sooty_Albatross/Sooty_Albatross_0029_796357.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/003.Sooty_Albatross/Sooty_Albatross_0030_1122.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/003.Sooty_Albatross/Sooty_Albatross_0031_1066.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/003.Sooty_Albatross/Sooty_Albatross_0032_1149.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/003.Sooty_Albatross/Sooty_Albatross_0033_1128.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/003.Sooty_Albatross/Sooty_Albatross_0034_1154.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/003.Sooty_Albatross/Sooty_Albatross_0036_796387.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/003.Sooty_Albatross/Sooty_Albatross_0038_1065.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/003.Sooty_Albatross/Sooty_Albatross_0040_796375.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/003.Sooty_Albatross/Sooty_Albatross_0041_796364.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/003.Sooty_Albatross/Sooty_Albatross_0042_1210.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/003.Sooty_Albatross/Sooty_Albatross_0043_1076.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/003.Sooty_Albatross/Sooty_Albatross_0044_1105.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/003.Sooty_Albatross/Sooty_Albatross_0045_1162.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/003.Sooty_Albatross/Sooty_Albatross_0046_1211.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/003.Sooty_Albatross/Sooty_Albatross_0048_1130.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/003.Sooty_Albatross/Sooty_Albatross_0049_796350.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/003.Sooty_Albatross/Sooty_Albatross_0050_796351.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/003.Sooty_Albatross/Sooty_Albatross_0051_796374.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/003.Sooty_Albatross/Sooty_Albatross_0054_796347.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/003.Sooty_Albatross/Sooty_Albatross_0055_1160.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/003.Sooty_Albatross/Sooty_Albatross_0057_796354.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/003.Sooty_Albatross/Sooty_Albatross_0058_796360.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/003.Sooty_Albatross/Sooty_Albatross_0063_1101.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/003.Sooty_Albatross/Sooty_Albatross_0064_796343.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/003.Sooty_Albatross/Sooty_Albatross_0065_796367.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/003.Sooty_Albatross/Sooty_Albatross_0066_796382.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/003.Sooty_Albatross/Sooty_Albatross_0067_796376.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/003.Sooty_Albatross/Sooty_Albatross_0068_796353.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/003.Sooty_Albatross/Sooty_Albatross_0069_796358.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/003.Sooty_Albatross/Sooty_Albatross_0070_796346.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/003.Sooty_Albatross/Sooty_Albatross_0071_1116.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/003.Sooty_Albatross/Sooty_Albatross_0072_796371.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/003.Sooty_Albatross/Sooty_Albatross_0073_1171.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/003.Sooty_Albatross/Sooty_Albatross_0074_1221.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/003.Sooty_Albatross/Sooty_Albatross_0075_796352.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/003.Sooty_Albatross/Sooty_Albatross_0076_796365.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/003.Sooty_Albatross/Sooty_Albatross_0077_1080.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/003.Sooty_Albatross/Sooty_Albatross_0079_796389.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/004.Groove_billed_Ani/Groove_Billed_Ani_0002_1670.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/004.Groove_billed_Ani/Groove_Billed_Ani_0004_1528.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/004.Groove_billed_Ani/Groove_Billed_Ani_0005_1750.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/004.Groove_billed_Ani/Groove_Billed_Ani_0006_1763.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/004.Groove_billed_Ani/Groove_Billed_Ani_0007_1615.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/004.Groove_billed_Ani/Groove_Billed_Ani_0009_1522.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/004.Groove_billed_Ani/Groove_Billed_Ani_0010_1704.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/004.Groove_billed_Ani/Groove_Billed_Ani_0012_1784.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/004.Groove_billed_Ani/Groove_Billed_Ani_0014_1755.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/004.Groove_billed_Ani/Groove_Billed_Ani_0015_1653.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/004.Groove_billed_Ani/Groove_Billed_Ani_0017_1561.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/004.Groove_billed_Ani/Groove_Billed_Ani_0018_1613.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/004.Groove_billed_Ani/Groove_Billed_Ani_0019_1585.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/004.Groove_billed_Ani/Groove_Billed_Ani_0023_1485.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/004.Groove_billed_Ani/Groove_Billed_Ani_0027_1754.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/004.Groove_billed_Ani/Groove_Billed_Ani_0029_1620.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/004.Groove_billed_Ani/Groove_Billed_Ani_0031_1588.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/004.Groove_billed_Ani/Groove_Billed_Ani_0032_1776.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/004.Groove_billed_Ani/Groove_Billed_Ani_0033_1494.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/004.Groove_billed_Ani/Groove_Billed_Ani_0035_1591.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/004.Groove_billed_Ani/Groove_Billed_Ani_0036_1604.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/004.Groove_billed_Ani/Groove_Billed_Ani_0037_1560.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/004.Groove_billed_Ani/Groove_Billed_Ani_0040_1715.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/004.Groove_billed_Ani/Groove_Billed_Ani_0044_1731.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/004.Groove_billed_Ani/Groove_Billed_Ani_0046_1663.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/004.Groove_billed_Ani/Groove_Billed_Ani_0047_1706.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/004.Groove_billed_Ani/Groove_Billed_Ani_0051_1650.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/004.Groove_billed_Ani/Groove_Billed_Ani_0053_1672.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/004.Groove_billed_Ani/Groove_Billed_Ani_0055_1501.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/004.Groove_billed_Ani/Groove_Billed_Ani_0056_1493.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/004.Groove_billed_Ani/Groove_Billed_Ani_0058_1751.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/004.Groove_billed_Ani/Groove_Billed_Ani_0059_1480.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/004.Groove_billed_Ani/Groove_Billed_Ani_0060_1505.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/004.Groove_billed_Ani/Groove_Billed_Ani_0061_1510.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/004.Groove_billed_Ani/Groove_Billed_Ani_0062_1767.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/004.Groove_billed_Ani/Groove_Billed_Ani_0065_1502.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/004.Groove_billed_Ani/Groove_Billed_Ani_0068_1538.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/004.Groove_billed_Ani/Groove_Billed_Ani_0069_1546.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/004.Groove_billed_Ani/Groove_Billed_Ani_0071_1559.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/004.Groove_billed_Ani/Groove_Billed_Ani_0072_1696.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/004.Groove_billed_Ani/Groove_Billed_Ani_0074_1730.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/004.Groove_billed_Ani/Groove_Billed_Ani_0075_1617.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/004.Groove_billed_Ani/Groove_Billed_Ani_0076_1661.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/004.Groove_billed_Ani/Groove_Billed_Ani_0077_1724.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/004.Groove_billed_Ani/Groove_Billed_Ani_0078_1780.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/004.Groove_billed_Ani/Groove_Billed_Ani_0080_1549.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/004.Groove_billed_Ani/Groove_Billed_Ani_0082_1697.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/004.Groove_billed_Ani/Groove_Billed_Ani_0085_1612.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/004.Groove_billed_Ani/Groove_Billed_Ani_0087_1765.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/004.Groove_billed_Ani/Groove_Billed_Ani_0088_1678.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/004.Groove_billed_Ani/Groove_Billed_Ani_0090_1567.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/004.Groove_billed_Ani/Groove_Billed_Ani_0091_1728.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/004.Groove_billed_Ani/Groove_Billed_Ani_0092_1516.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/004.Groove_billed_Ani/Groove_Billed_Ani_0094_1540.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/004.Groove_billed_Ani/Groove_Billed_Ani_0100_1646.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/004.Groove_billed_Ani/Groove_Billed_Ani_0101_1700.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/004.Groove_billed_Ani/Groove_Billed_Ani_0105_1562.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/004.Groove_billed_Ani/Groove_Billed_Ani_0107_1590.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/004.Groove_billed_Ani/Groove_Billed_Ani_0108_1639.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/004.Groove_billed_Ani/Groove_Billed_Ani_0109_1592.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/005.Crested_Auklet/Crested_Auklet_0001_794941.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/005.Crested_Auklet/Crested_Auklet_0003_794962.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/005.Crested_Auklet/Crested_Auklet_0005_794922.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/005.Crested_Auklet/Crested_Auklet_0006_1813.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/005.Crested_Auklet/Crested_Auklet_0010_794907.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/005.Crested_Auklet/Crested_Auklet_0011_794927.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/005.Crested_Auklet/Crested_Auklet_0012_794960.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/005.Crested_Auklet/Crested_Auklet_0013_794914.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/005.Crested_Auklet/Crested_Auklet_0018_1817.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/005.Crested_Auklet/Crested_Auklet_0019_794925.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/005.Crested_Auklet/Crested_Auklet_0021_794938.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/005.Crested_Auklet/Crested_Auklet_0028_794951.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/005.Crested_Auklet/Crested_Auklet_0029_1824.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/005.Crested_Auklet/Crested_Auklet_0030_794937.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/005.Crested_Auklet/Crested_Auklet_0032_794931.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/005.Crested_Auklet/Crested_Auklet_0033_794964.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/005.Crested_Auklet/Crested_Auklet_0036_794905.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/005.Crested_Auklet/Crested_Auklet_0039_794944.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/005.Crested_Auklet/Crested_Auklet_0040_794912.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/005.Crested_Auklet/Crested_Auklet_0041_794910.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/005.Crested_Auklet/Crested_Auklet_0042_794902.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/005.Crested_Auklet/Crested_Auklet_0044_1825.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/005.Crested_Auklet/Crested_Auklet_0045_794940.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/005.Crested_Auklet/Crested_Auklet_0047_794918.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/005.Crested_Auklet/Crested_Auklet_0048_1791.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/005.Crested_Auklet/Crested_Auklet_0052_794952.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/005.Crested_Auklet/Crested_Auklet_0053_794966.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/005.Crested_Auklet/Crested_Auklet_0057_794932.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/005.Crested_Auklet/Crested_Auklet_0059_794929.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/005.Crested_Auklet/Crested_Auklet_0061_794904.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/005.Crested_Auklet/Crested_Auklet_0062_794968.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/005.Crested_Auklet/Crested_Auklet_0063_794901.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/005.Crested_Auklet/Crested_Auklet_0066_785251.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/005.Crested_Auklet/Crested_Auklet_0067_785249.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/005.Crested_Auklet/Crested_Auklet_0068_785253.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/005.Crested_Auklet/Crested_Auklet_0069_785258.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/005.Crested_Auklet/Crested_Auklet_0070_785261.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/005.Crested_Auklet/Crested_Auklet_0071_785255.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/005.Crested_Auklet/Crested_Auklet_0072_785260.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/005.Crested_Auklet/Crested_Auklet_0073_785248.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/005.Crested_Auklet/Crested_Auklet_0074_794949.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/005.Crested_Auklet/Crested_Auklet_0075_785259.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/005.Crested_Auklet/Crested_Auklet_0076_785252.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/005.Crested_Auklet/Crested_Auklet_0077_785257.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/006.Least_Auklet/Least_Auklet_0004_795112.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/006.Least_Auklet/Least_Auklet_0005_795119.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/006.Least_Auklet/Least_Auklet_0007_795123.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/006.Least_Auklet/Least_Auklet_0008_795071.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/006.Least_Auklet/Least_Auklet_0010_795092.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/006.Least_Auklet/Least_Auklet_0011_795109.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/006.Least_Auklet/Least_Auklet_0012_795074.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/006.Least_Auklet/Least_Auklet_0014_1901.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/006.Least_Auklet/Least_Auklet_0015_795065.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/006.Least_Auklet/Least_Auklet_0016_1903.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/006.Least_Auklet/Least_Auklet_0017_795084.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/006.Least_Auklet/Least_Auklet_0018_795077.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/006.Least_Auklet/Least_Auklet_0020_795080.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/006.Least_Auklet/Least_Auklet_0021_795127.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/006.Least_Auklet/Least_Auklet_0023_1898.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/006.Least_Auklet/Least_Auklet_0024_795120.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/006.Least_Auklet/Least_Auklet_0025_795087.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/006.Least_Auklet/Least_Auklet_0026_795066.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/006.Least_Auklet/Least_Auklet_0027_795091.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/006.Least_Auklet/Least_Auklet_0028_795094.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/006.Least_Auklet/Least_Auklet_0029_795122.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/006.Least_Auklet/Least_Auklet_0030_795116.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/006.Least_Auklet/Least_Auklet_0032_795068.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/006.Least_Auklet/Least_Auklet_0034_795096.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/006.Least_Auklet/Least_Auklet_0035_1888.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/006.Least_Auklet/Least_Auklet_0038_795132.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/006.Least_Auklet/Least_Auklet_0039_795081.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/006.Least_Auklet/Least_Auklet_0040_795100.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/006.Least_Auklet/Least_Auklet_0041_795107.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/006.Least_Auklet/Least_Auklet_0042_1874.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/006.Least_Auklet/Least_Auklet_0043_795067.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/006.Least_Auklet/Least_Auklet_0045_795069.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/006.Least_Auklet/Least_Auklet_0046_795118.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/006.Least_Auklet/Least_Auklet_0050_1924.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/006.Least_Auklet/Least_Auklet_0052_795088.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/006.Least_Auklet/Least_Auklet_0054_795130.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/006.Least_Auklet/Least_Auklet_0055_795106.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/006.Least_Auklet/Least_Auklet_0057_795126.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/006.Least_Auklet/Least_Auklet_0058_1931.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/006.Least_Auklet/Least_Auklet_0060_795095.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/006.Least_Auklet/Least_Auklet_0063_795134.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/007.Parakeet_Auklet/Parakeet_Auklet_0001_795972.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/007.Parakeet_Auklet/Parakeet_Auklet_0003_795982.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/007.Parakeet_Auklet/Parakeet_Auklet_0004_795921.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/007.Parakeet_Auklet/Parakeet_Auklet_0007_795932.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/007.Parakeet_Auklet/Parakeet_Auklet_0008_795992.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/007.Parakeet_Auklet/Parakeet_Auklet_0010_795995.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/007.Parakeet_Auklet/Parakeet_Auklet_0012_795927.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/007.Parakeet_Auklet/Parakeet_Auklet_0013_795970.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/007.Parakeet_Auklet/Parakeet_Auklet_0015_795966.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/007.Parakeet_Auklet/Parakeet_Auklet_0016_795977.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/007.Parakeet_Auklet/Parakeet_Auklet_0017_795924.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/007.Parakeet_Auklet/Parakeet_Auklet_0019_795987.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/007.Parakeet_Auklet/Parakeet_Auklet_0020_795947.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/007.Parakeet_Auklet/Parakeet_Auklet_0024_2045.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/007.Parakeet_Auklet/Parakeet_Auklet_0025_795975.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/007.Parakeet_Auklet/Parakeet_Auklet_0026_795962.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/007.Parakeet_Auklet/Parakeet_Auklet_0027_795925.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/007.Parakeet_Auklet/Parakeet_Auklet_0028_795944.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/007.Parakeet_Auklet/Parakeet_Auklet_0030_795930.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/007.Parakeet_Auklet/Parakeet_Auklet_0031_795960.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/007.Parakeet_Auklet/Parakeet_Auklet_0032_795986.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/007.Parakeet_Auklet/Parakeet_Auklet_0034_795994.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/007.Parakeet_Auklet/Parakeet_Auklet_0035_795934.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/007.Parakeet_Auklet/Parakeet_Auklet_0036_795943.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/007.Parakeet_Auklet/Parakeet_Auklet_0038_795935.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/007.Parakeet_Auklet/Parakeet_Auklet_0039_795955.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/007.Parakeet_Auklet/Parakeet_Auklet_0040_795974.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/007.Parakeet_Auklet/Parakeet_Auklet_0041_795933.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/007.Parakeet_Auklet/Parakeet_Auklet_0042_795961.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/007.Parakeet_Auklet/Parakeet_Auklet_0043_795948.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/007.Parakeet_Auklet/Parakeet_Auklet_0045_795953.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/007.Parakeet_Auklet/Parakeet_Auklet_0047_795956.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/007.Parakeet_Auklet/Parakeet_Auklet_0048_795980.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/007.Parakeet_Auklet/Parakeet_Auklet_0050_795957.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/007.Parakeet_Auklet/Parakeet_Auklet_0051_795971.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/007.Parakeet_Auklet/Parakeet_Auklet_0055_795963.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/007.Parakeet_Auklet/Parakeet_Auklet_0056_795926.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/007.Parakeet_Auklet/Parakeet_Auklet_0058_795942.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/007.Parakeet_Auklet/Parakeet_Auklet_0059_795973.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/007.Parakeet_Auklet/Parakeet_Auklet_0061_795990.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/007.Parakeet_Auklet/Parakeet_Auklet_0062_795958.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/007.Parakeet_Auklet/Parakeet_Auklet_0064_795954.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/007.Parakeet_Auklet/Parakeet_Auklet_0065_795969.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/007.Parakeet_Auklet/Parakeet_Auklet_0067_795964.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/007.Parakeet_Auklet/Parakeet_Auklet_0069_795936.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/007.Parakeet_Auklet/Parakeet_Auklet_0070_795998.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/007.Parakeet_Auklet/Parakeet_Auklet_0072_795929.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/007.Parakeet_Auklet/Parakeet_Auklet_0073_795939.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/007.Parakeet_Auklet/Parakeet_Auklet_0074_795976.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/007.Parakeet_Auklet/Parakeet_Auklet_0075_795981.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/007.Parakeet_Auklet/Parakeet_Auklet_0076_796001.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/007.Parakeet_Auklet/Parakeet_Auklet_0078_2004.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/007.Parakeet_Auklet/Parakeet_Auklet_0080_795965.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/008.Rhinoceros_Auklet/Rhinoceros_Auklet_0001_797538.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/008.Rhinoceros_Auklet/Rhinoceros_Auklet_0003_797535.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/008.Rhinoceros_Auklet/Rhinoceros_Auklet_0004_797541.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/008.Rhinoceros_Auklet/Rhinoceros_Auklet_0005_2111.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/008.Rhinoceros_Auklet/Rhinoceros_Auklet_0006_797512.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/008.Rhinoceros_Auklet/Rhinoceros_Auklet_0007_797521.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/008.Rhinoceros_Auklet/Rhinoceros_Auklet_0008_797531.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/008.Rhinoceros_Auklet/Rhinoceros_Auklet_0009_797539.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/008.Rhinoceros_Auklet/Rhinoceros_Auklet_0010_797545.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/008.Rhinoceros_Auklet/Rhinoceros_Auklet_0011_797530.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/008.Rhinoceros_Auklet/Rhinoceros_Auklet_0012_2161.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/008.Rhinoceros_Auklet/Rhinoceros_Auklet_0013_797537.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/008.Rhinoceros_Auklet/Rhinoceros_Auklet_0014_797522.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/008.Rhinoceros_Auklet/Rhinoceros_Auklet_0015_2160.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/008.Rhinoceros_Auklet/Rhinoceros_Auklet_0016_2114.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/008.Rhinoceros_Auklet/Rhinoceros_Auklet_0017_2155.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/008.Rhinoceros_Auklet/Rhinoceros_Auklet_0018_797517.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/008.Rhinoceros_Auklet/Rhinoceros_Auklet_0019_797513.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/008.Rhinoceros_Auklet/Rhinoceros_Auklet_0020_2195.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/008.Rhinoceros_Auklet/Rhinoceros_Auklet_0021_2089.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/008.Rhinoceros_Auklet/Rhinoceros_Auklet_0022_2170.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/008.Rhinoceros_Auklet/Rhinoceros_Auklet_0023_797501.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/008.Rhinoceros_Auklet/Rhinoceros_Auklet_0024_797529.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/008.Rhinoceros_Auklet/Rhinoceros_Auklet_0025_797508.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/008.Rhinoceros_Auklet/Rhinoceros_Auklet_0026_797519.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/008.Rhinoceros_Auklet/Rhinoceros_Auklet_0027_797496.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/008.Rhinoceros_Auklet/Rhinoceros_Auklet_0028_2056.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/008.Rhinoceros_Auklet/Rhinoceros_Auklet_0029_797520.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/008.Rhinoceros_Auklet/Rhinoceros_Auklet_0030_797509.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/008.Rhinoceros_Auklet/Rhinoceros_Auklet_0031_797518.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/008.Rhinoceros_Auklet/Rhinoceros_Auklet_0032_797516.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/008.Rhinoceros_Auklet/Rhinoceros_Auklet_0033_2169.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/008.Rhinoceros_Auklet/Rhinoceros_Auklet_0034_797497.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/008.Rhinoceros_Auklet/Rhinoceros_Auklet_0035_2166.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/008.Rhinoceros_Auklet/Rhinoceros_Auklet_0037_797499.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/008.Rhinoceros_Auklet/Rhinoceros_Auklet_0038_797544.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/008.Rhinoceros_Auklet/Rhinoceros_Auklet_0039_2174.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/008.Rhinoceros_Auklet/Rhinoceros_Auklet_0040_797503.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/008.Rhinoceros_Auklet/Rhinoceros_Auklet_0041_797515.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/008.Rhinoceros_Auklet/Rhinoceros_Auklet_0042_2101.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/008.Rhinoceros_Auklet/Rhinoceros_Auklet_0043_2096.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/008.Rhinoceros_Auklet/Rhinoceros_Auklet_0044_797502.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/008.Rhinoceros_Auklet/Rhinoceros_Auklet_0045_797536.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/008.Rhinoceros_Auklet/Rhinoceros_Auklet_0046_797532.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/008.Rhinoceros_Auklet/Rhinoceros_Auklet_0048_2162.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/008.Rhinoceros_Auklet/Rhinoceros_Auklet_0049_797543.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/008.Rhinoceros_Auklet/Rhinoceros_Auklet_0050_797534.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/008.Rhinoceros_Auklet/Rhinoceros_Auklet_0051_797510.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/009.Brewer_Blackbird/Brewer_Blackbird_0002_2278.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/009.Brewer_Blackbird/Brewer_Blackbird_0004_2345.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/009.Brewer_Blackbird/Brewer_Blackbird_0009_2616.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/009.Brewer_Blackbird/Brewer_Blackbird_0010_2269.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/009.Brewer_Blackbird/Brewer_Blackbird_0012_2691.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/009.Brewer_Blackbird/Brewer_Blackbird_0014_2679.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/009.Brewer_Blackbird/Brewer_Blackbird_0015_2286.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/009.Brewer_Blackbird/Brewer_Blackbird_0016_2225.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/009.Brewer_Blackbird/Brewer_Blackbird_0017_2668.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/009.Brewer_Blackbird/Brewer_Blackbird_0018_2261.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/009.Brewer_Blackbird/Brewer_Blackbird_0025_2231.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/009.Brewer_Blackbird/Brewer_Blackbird_0026_2625.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/009.Brewer_Blackbird/Brewer_Blackbird_0027_2329.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/009.Brewer_Blackbird/Brewer_Blackbird_0028_2682.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/009.Brewer_Blackbird/Brewer_Blackbird_0030_2268.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/009.Brewer_Blackbird/Brewer_Blackbird_0032_2214.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/009.Brewer_Blackbird/Brewer_Blackbird_0035_2611.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/009.Brewer_Blackbird/Brewer_Blackbird_0038_2294.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/009.Brewer_Blackbird/Brewer_Blackbird_0041_2653.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/009.Brewer_Blackbird/Brewer_Blackbird_0042_2302.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/009.Brewer_Blackbird/Brewer_Blackbird_0045_2303.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/009.Brewer_Blackbird/Brewer_Blackbird_0046_2688.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/009.Brewer_Blackbird/Brewer_Blackbird_0049_2258.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/009.Brewer_Blackbird/Brewer_Blackbird_0054_2631.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/009.Brewer_Blackbird/Brewer_Blackbird_0061_2270.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/009.Brewer_Blackbird/Brewer_Blackbird_0064_2290.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/009.Brewer_Blackbird/Brewer_Blackbird_0065_2310.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/009.Brewer_Blackbird/Brewer_Blackbird_0066_2693.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/009.Brewer_Blackbird/Brewer_Blackbird_0070_2325.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/009.Brewer_Blackbird/Brewer_Blackbird_0074_2277.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/009.Brewer_Blackbird/Brewer_Blackbird_0078_2659.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/009.Brewer_Blackbird/Brewer_Blackbird_0079_2343.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/009.Brewer_Blackbird/Brewer_Blackbird_0080_2234.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/009.Brewer_Blackbird/Brewer_Blackbird_0082_2593.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/009.Brewer_Blackbird/Brewer_Blackbird_0087_2622.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/009.Brewer_Blackbird/Brewer_Blackbird_0090_2658.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/009.Brewer_Blackbird/Brewer_Blackbird_0095_2610.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/009.Brewer_Blackbird/Brewer_Blackbird_0096_2634.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/009.Brewer_Blackbird/Brewer_Blackbird_0097_2322.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/009.Brewer_Blackbird/Brewer_Blackbird_0099_2560.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/009.Brewer_Blackbird/Brewer_Blackbird_0101_2630.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/009.Brewer_Blackbird/Brewer_Blackbird_0102_2620.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/009.Brewer_Blackbird/Brewer_Blackbird_0103_2273.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/009.Brewer_Blackbird/Brewer_Blackbird_0104_2230.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/009.Brewer_Blackbird/Brewer_Blackbird_0106_2608.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/009.Brewer_Blackbird/Brewer_Blackbird_0109_2232.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/009.Brewer_Blackbird/Brewer_Blackbird_0111_2613.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/009.Brewer_Blackbird/Brewer_Blackbird_0112_2340.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/009.Brewer_Blackbird/Brewer_Blackbird_0115_2279.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/009.Brewer_Blackbird/Brewer_Blackbird_0116_2327.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/009.Brewer_Blackbird/Brewer_Blackbird_0127_2235.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/009.Brewer_Blackbird/Brewer_Blackbird_0131_2289.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/009.Brewer_Blackbird/Brewer_Blackbird_0132_2293.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/009.Brewer_Blackbird/Brewer_Blackbird_0133_2324.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/009.Brewer_Blackbird/Brewer_Blackbird_0135_2607.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/009.Brewer_Blackbird/Brewer_Blackbird_0137_2680.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/009.Brewer_Blackbird/Brewer_Blackbird_0139_2567.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/009.Brewer_Blackbird/Brewer_Blackbird_0140_2586.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/009.Brewer_Blackbird/Brewer_Blackbird_0142_2636.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/010.Red_winged_Blackbird/Red_Winged_Blackbird_0001_3695.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/010.Red_winged_Blackbird/Red_Winged_Blackbird_0005_5636.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/010.Red_winged_Blackbird/Red_Winged_Blackbird_0006_6005.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/010.Red_winged_Blackbird/Red_Winged_Blackbird_0007_3706.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/010.Red_winged_Blackbird/Red_Winged_Blackbird_0009_5841.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/010.Red_winged_Blackbird/Red_Winged_Blackbird_0010_6386.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/010.Red_winged_Blackbird/Red_Winged_Blackbird_0011_5845.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/010.Red_winged_Blackbird/Red_Winged_Blackbird_0012_6015.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/010.Red_winged_Blackbird/Red_Winged_Blackbird_0013_5762.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/010.Red_winged_Blackbird/Red_Winged_Blackbird_0014_3761.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/010.Red_winged_Blackbird/Red_Winged_Blackbird_0017_4116.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/010.Red_winged_Blackbird/Red_Winged_Blackbird_0020_4050.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/010.Red_winged_Blackbird/Red_Winged_Blackbird_0021_3767.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/010.Red_winged_Blackbird/Red_Winged_Blackbird_0022_4483.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/010.Red_winged_Blackbird/Red_Winged_Blackbird_0023_5257.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/010.Red_winged_Blackbird/Red_Winged_Blackbird_0024_4180.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/010.Red_winged_Blackbird/Red_Winged_Blackbird_0025_5342.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/010.Red_winged_Blackbird/Red_Winged_Blackbird_0027_4123.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/010.Red_winged_Blackbird/Red_Winged_Blackbird_0028_4709.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/010.Red_winged_Blackbird/Red_Winged_Blackbird_0029_4804.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/010.Red_winged_Blackbird/Red_Winged_Blackbird_0032_4004.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/010.Red_winged_Blackbird/Red_Winged_Blackbird_0039_4285.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/010.Red_winged_Blackbird/Red_Winged_Blackbird_0040_4522.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/010.Red_winged_Blackbird/Red_Winged_Blackbird_0042_3635.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/010.Red_winged_Blackbird/Red_Winged_Blackbird_0044_5621.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/010.Red_winged_Blackbird/Red_Winged_Blackbird_0045_4526.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/010.Red_winged_Blackbird/Red_Winged_Blackbird_0046_4242.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/010.Red_winged_Blackbird/Red_Winged_Blackbird_0047_3802.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/010.Red_winged_Blackbird/Red_Winged_Blackbird_0049_5598.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/010.Red_winged_Blackbird/Red_Winged_Blackbird_0052_5575.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/010.Red_winged_Blackbird/Red_Winged_Blackbird_0053_4072.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/010.Red_winged_Blackbird/Red_Winged_Blackbird_0054_4625.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/010.Red_winged_Blackbird/Red_Winged_Blackbird_0055_4345.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/010.Red_winged_Blackbird/Red_Winged_Blackbird_0058_4141.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/010.Red_winged_Blackbird/Red_Winged_Blackbird_0059_4612.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/010.Red_winged_Blackbird/Red_Winged_Blackbird_0060_4688.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/010.Red_winged_Blackbird/Red_Winged_Blackbird_0061_4196.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/010.Red_winged_Blackbird/Red_Winged_Blackbird_0062_4233.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/010.Red_winged_Blackbird/Red_Winged_Blackbird_0064_4936.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/010.Red_winged_Blackbird/Red_Winged_Blackbird_0065_4026.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/010.Red_winged_Blackbird/Red_Winged_Blackbird_0066_5070.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/010.Red_winged_Blackbird/Red_Winged_Blackbird_0071_3988.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/010.Red_winged_Blackbird/Red_Winged_Blackbird_0072_4338.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/010.Red_winged_Blackbird/Red_Winged_Blackbird_0074_4146.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/010.Red_winged_Blackbird/Red_Winged_Blackbird_0075_4953.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/010.Red_winged_Blackbird/Red_Winged_Blackbird_0078_5372.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/010.Red_winged_Blackbird/Red_Winged_Blackbird_0079_4527.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/010.Red_winged_Blackbird/Red_Winged_Blackbird_0081_6081.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/010.Red_winged_Blackbird/Red_Winged_Blackbird_0085_5846.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/010.Red_winged_Blackbird/Red_Winged_Blackbird_0088_4007.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/010.Red_winged_Blackbird/Red_Winged_Blackbird_0089_4188.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/010.Red_winged_Blackbird/Red_Winged_Blackbird_0091_4096.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/010.Red_winged_Blackbird/Red_Winged_Blackbird_0093_5948.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/010.Red_winged_Blackbird/Red_Winged_Blackbird_0094_5856.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/010.Red_winged_Blackbird/Red_Winged_Blackbird_0096_5019.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/010.Red_winged_Blackbird/Red_Winged_Blackbird_0099_3985.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/010.Red_winged_Blackbird/Red_Winged_Blackbird_0101_6244.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/010.Red_winged_Blackbird/Red_Winged_Blackbird_0104_3918.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/010.Red_winged_Blackbird/Red_Winged_Blackbird_0105_4842.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/010.Red_winged_Blackbird/Red_Winged_Blackbird_0109_4454.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/011.Rusty_Blackbird/Rusty_Blackbird_0001_6548.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/011.Rusty_Blackbird/Rusty_Blackbird_0003_6749.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/011.Rusty_Blackbird/Rusty_Blackbird_0005_6771.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/011.Rusty_Blackbird/Rusty_Blackbird_0006_6633.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/011.Rusty_Blackbird/Rusty_Blackbird_0009_6853.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/011.Rusty_Blackbird/Rusty_Blackbird_0011_7028.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/011.Rusty_Blackbird/Rusty_Blackbird_0013_6902.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/011.Rusty_Blackbird/Rusty_Blackbird_0015_6885.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/011.Rusty_Blackbird/Rusty_Blackbird_0016_6684.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/011.Rusty_Blackbird/Rusty_Blackbird_0017_6755.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/011.Rusty_Blackbird/Rusty_Blackbird_0019_6704.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/011.Rusty_Blackbird/Rusty_Blackbird_0020_6679.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/011.Rusty_Blackbird/Rusty_Blackbird_0022_6808.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/011.Rusty_Blackbird/Rusty_Blackbird_0023_6752.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/011.Rusty_Blackbird/Rusty_Blackbird_0026_6768.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/011.Rusty_Blackbird/Rusty_Blackbird_0027_6593.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/011.Rusty_Blackbird/Rusty_Blackbird_0031_6699.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/011.Rusty_Blackbird/Rusty_Blackbird_0032_6611.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/011.Rusty_Blackbird/Rusty_Blackbird_0033_6879.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/011.Rusty_Blackbird/Rusty_Blackbird_0036_6550.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/011.Rusty_Blackbird/Rusty_Blackbird_0043_2597.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/011.Rusty_Blackbird/Rusty_Blackbird_0047_7009.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/011.Rusty_Blackbird/Rusty_Blackbird_0048_6632.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/011.Rusty_Blackbird/Rusty_Blackbird_0051_6715.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/011.Rusty_Blackbird/Rusty_Blackbird_0052_7035.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/011.Rusty_Blackbird/Rusty_Blackbird_0054_6676.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/011.Rusty_Blackbird/Rusty_Blackbird_0055_6923.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/011.Rusty_Blackbird/Rusty_Blackbird_0056_6856.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/011.Rusty_Blackbird/Rusty_Blackbird_0057_6935.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/011.Rusty_Blackbird/Rusty_Blackbird_0060_6756.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/011.Rusty_Blackbird/Rusty_Blackbird_0073_6744.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/011.Rusty_Blackbird/Rusty_Blackbird_0074_6585.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/011.Rusty_Blackbird/Rusty_Blackbird_0075_6717.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/011.Rusty_Blackbird/Rusty_Blackbird_0076_6716.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/011.Rusty_Blackbird/Rusty_Blackbird_0080_6877.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/011.Rusty_Blackbird/Rusty_Blackbird_0081_6967.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/011.Rusty_Blackbird/Rusty_Blackbird_0082_6906.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/011.Rusty_Blackbird/Rusty_Blackbird_0085_6713.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/011.Rusty_Blackbird/Rusty_Blackbird_0086_6658.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/011.Rusty_Blackbird/Rusty_Blackbird_0087_6727.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/011.Rusty_Blackbird/Rusty_Blackbird_0091_6695.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/011.Rusty_Blackbird/Rusty_Blackbird_0092_2727.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/011.Rusty_Blackbird/Rusty_Blackbird_0093_6628.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/011.Rusty_Blackbird/Rusty_Blackbird_0094_6582.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/011.Rusty_Blackbird/Rusty_Blackbird_0096_6846.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/011.Rusty_Blackbird/Rusty_Blackbird_0100_6597.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/011.Rusty_Blackbird/Rusty_Blackbird_0101_6880.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/011.Rusty_Blackbird/Rusty_Blackbird_0102_6590.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/011.Rusty_Blackbird/Rusty_Blackbird_0104_6685.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/011.Rusty_Blackbird/Rusty_Blackbird_0105_6937.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/011.Rusty_Blackbird/Rusty_Blackbird_0107_6839.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/011.Rusty_Blackbird/Rusty_Blackbird_0108_6867.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/011.Rusty_Blackbird/Rusty_Blackbird_0109_6698.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/011.Rusty_Blackbird/Rusty_Blackbird_0111_3220.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/011.Rusty_Blackbird/Rusty_Blackbird_0112_3415.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/011.Rusty_Blackbird/Rusty_Blackbird_0113_6664.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/011.Rusty_Blackbird/Rusty_Blackbird_0114_6760.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/011.Rusty_Blackbird/Rusty_Blackbird_0120_6762.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/011.Rusty_Blackbird/Rusty_Blackbird_0121_6637.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/011.Rusty_Blackbird/Rusty_Blackbird_0122_6736.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/012.Yellow_headed_Blackbird/Yellow_Headed_Blackbird_0003_8337.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/012.Yellow_headed_Blackbird/Yellow_Headed_Blackbird_0007_8366.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/012.Yellow_headed_Blackbird/Yellow_Headed_Blackbird_0008_8756.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/012.Yellow_headed_Blackbird/Yellow_Headed_Blackbird_0009_8248.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/012.Yellow_headed_Blackbird/Yellow_Headed_Blackbird_0012_8443.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/012.Yellow_headed_Blackbird/Yellow_Headed_Blackbird_0013_8362.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/012.Yellow_headed_Blackbird/Yellow_Headed_Blackbird_0015_8207.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/012.Yellow_headed_Blackbird/Yellow_Headed_Blackbird_0017_8511.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/012.Yellow_headed_Blackbird/Yellow_Headed_Blackbird_0018_8588.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/012.Yellow_headed_Blackbird/Yellow_Headed_Blackbird_0020_8549.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/012.Yellow_headed_Blackbird/Yellow_Headed_Blackbird_0023_7325.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/012.Yellow_headed_Blackbird/Yellow_Headed_Blackbird_0024_8586.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/012.Yellow_headed_Blackbird/Yellow_Headed_Blackbird_0025_8262.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/012.Yellow_headed_Blackbird/Yellow_Headed_Blackbird_0026_8545.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/012.Yellow_headed_Blackbird/Yellow_Headed_Blackbird_0031_8456.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/012.Yellow_headed_Blackbird/Yellow_Headed_Blackbird_0034_7736.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/012.Yellow_headed_Blackbird/Yellow_Headed_Blackbird_0035_8447.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/012.Yellow_headed_Blackbird/Yellow_Headed_Blackbird_0038_8689.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/012.Yellow_headed_Blackbird/Yellow_Headed_Blackbird_0040_7514.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/012.Yellow_headed_Blackbird/Yellow_Headed_Blackbird_0041_8264.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/012.Yellow_headed_Blackbird/Yellow_Headed_Blackbird_0042_8574.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/012.Yellow_headed_Blackbird/Yellow_Headed_Blackbird_0043_8250.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/012.Yellow_headed_Blackbird/Yellow_Headed_Blackbird_0047_7929.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/012.Yellow_headed_Blackbird/Yellow_Headed_Blackbird_0049_8548.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/012.Yellow_headed_Blackbird/Yellow_Headed_Blackbird_0051_8387.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/012.Yellow_headed_Blackbird/Yellow_Headed_Blackbird_0053_8410.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/012.Yellow_headed_Blackbird/Yellow_Headed_Blackbird_0055_8357.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/012.Yellow_headed_Blackbird/Yellow_Headed_Blackbird_0056_8455.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/012.Yellow_headed_Blackbird/Yellow_Headed_Blackbird_0057_8236.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/012.Yellow_headed_Blackbird/Yellow_Headed_Blackbird_0058_8350.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/012.Yellow_headed_Blackbird/Yellow_Headed_Blackbird_0059_8079.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/012.Yellow_headed_Blackbird/Yellow_Headed_Blackbird_0061_8208.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/012.Yellow_headed_Blackbird/Yellow_Headed_Blackbird_0062_8310.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/012.Yellow_headed_Blackbird/Yellow_Headed_Blackbird_0065_8481.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/012.Yellow_headed_Blackbird/Yellow_Headed_Blackbird_0070_8583.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/012.Yellow_headed_Blackbird/Yellow_Headed_Blackbird_0072_8606.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/012.Yellow_headed_Blackbird/Yellow_Headed_Blackbird_0073_8442.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/012.Yellow_headed_Blackbird/Yellow_Headed_Blackbird_0074_8452.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/012.Yellow_headed_Blackbird/Yellow_Headed_Blackbird_0077_8332.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/012.Yellow_headed_Blackbird/Yellow_Headed_Blackbird_0079_8535.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/012.Yellow_headed_Blackbird/Yellow_Headed_Blackbird_0080_8601.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/012.Yellow_headed_Blackbird/Yellow_Headed_Blackbird_0082_8577.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/012.Yellow_headed_Blackbird/Yellow_Headed_Blackbird_0083_8300.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/012.Yellow_headed_Blackbird/Yellow_Headed_Blackbird_0084_8435.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/012.Yellow_headed_Blackbird/Yellow_Headed_Blackbird_0085_8363.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/012.Yellow_headed_Blackbird/Yellow_Headed_Blackbird_0086_8487.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/012.Yellow_headed_Blackbird/Yellow_Headed_Blackbird_0087_8358.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/012.Yellow_headed_Blackbird/Yellow_Headed_Blackbird_0088_8257.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/012.Yellow_headed_Blackbird/Yellow_Headed_Blackbird_0089_8326.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/012.Yellow_headed_Blackbird/Yellow_Headed_Blackbird_0091_8555.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/012.Yellow_headed_Blackbird/Yellow_Headed_Blackbird_0095_8458.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/012.Yellow_headed_Blackbird/Yellow_Headed_Blackbird_0098_8367.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/012.Yellow_headed_Blackbird/Yellow_Headed_Blackbird_0100_8407.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/012.Yellow_headed_Blackbird/Yellow_Headed_Blackbird_0102_8441.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/012.Yellow_headed_Blackbird/Yellow_Headed_Blackbird_0108_7937.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/012.Yellow_headed_Blackbird/Yellow_Headed_Blackbird_0109_8271.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/013.Bobolink/Bobolink_0001_9261.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/013.Bobolink/Bobolink_0002_11085.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/013.Bobolink/Bobolink_0007_9246.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/013.Bobolink/Bobolink_0008_9289.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/013.Bobolink/Bobolink_0013_9367.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/013.Bobolink/Bobolink_0014_11055.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/013.Bobolink/Bobolink_0018_9402.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/013.Bobolink/Bobolink_0019_10552.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/013.Bobolink/Bobolink_0020_9194.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/013.Bobolink/Bobolink_0021_10623.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/013.Bobolink/Bobolink_0026_11057.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/013.Bobolink/Bobolink_0027_10569.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/013.Bobolink/Bobolink_0032_10217.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/013.Bobolink/Bobolink_0033_10809.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/013.Bobolink/Bobolink_0035_11117.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/013.Bobolink/Bobolink_0039_9779.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/013.Bobolink/Bobolink_0040_9681.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/013.Bobolink/Bobolink_0043_10607.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/013.Bobolink/Bobolink_0044_9990.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/013.Bobolink/Bobolink_0047_9204.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/013.Bobolink/Bobolink_0048_9988.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/013.Bobolink/Bobolink_0049_9540.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/013.Bobolink/Bobolink_0050_9821.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/013.Bobolink/Bobolink_0052_9423.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/013.Bobolink/Bobolink_0053_10166.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/013.Bobolink/Bobolink_0056_9080.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/013.Bobolink/Bobolink_0057_10051.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/013.Bobolink/Bobolink_0059_10041.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/013.Bobolink/Bobolink_0064_10092.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/013.Bobolink/Bobolink_0065_9375.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/013.Bobolink/Bobolink_0067_11533.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/013.Bobolink/Bobolink_0069_9085.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/013.Bobolink/Bobolink_0070_10624.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/013.Bobolink/Bobolink_0071_9503.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/013.Bobolink/Bobolink_0074_9311.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/013.Bobolink/Bobolink_0076_11093.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/013.Bobolink/Bobolink_0079_10736.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/013.Bobolink/Bobolink_0081_9439.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/013.Bobolink/Bobolink_0092_10026.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/013.Bobolink/Bobolink_0094_9823.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/013.Bobolink/Bobolink_0097_10861.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/013.Bobolink/Bobolink_0099_9314.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/013.Bobolink/Bobolink_0101_9811.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/013.Bobolink/Bobolink_0102_10807.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/013.Bobolink/Bobolink_0104_10273.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/013.Bobolink/Bobolink_0106_9126.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/013.Bobolink/Bobolink_0107_10252.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/013.Bobolink/Bobolink_0109_9869.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/013.Bobolink/Bobolink_0110_9496.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/013.Bobolink/Bobolink_0112_11073.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/013.Bobolink/Bobolink_0114_10627.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/013.Bobolink/Bobolink_0115_9265.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/013.Bobolink/Bobolink_0117_10215.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/013.Bobolink/Bobolink_0119_10430.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/013.Bobolink/Bobolink_0120_10859.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/013.Bobolink/Bobolink_0124_10182.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/013.Bobolink/Bobolink_0126_11458.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/013.Bobolink/Bobolink_0128_9947.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/013.Bobolink/Bobolink_0131_9578.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/013.Bobolink/Bobolink_0133_9618.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/014.Indigo_Bunting/Indigo_Bunting_0001_12469.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/014.Indigo_Bunting/Indigo_Bunting_0002_12163.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/014.Indigo_Bunting/Indigo_Bunting_0003_13049.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/014.Indigo_Bunting/Indigo_Bunting_0004_13195.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/014.Indigo_Bunting/Indigo_Bunting_0005_13832.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/014.Indigo_Bunting/Indigo_Bunting_0006_14317.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/014.Indigo_Bunting/Indigo_Bunting_0010_13000.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/014.Indigo_Bunting/Indigo_Bunting_0013_12949.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/014.Indigo_Bunting/Indigo_Bunting_0014_14167.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/014.Indigo_Bunting/Indigo_Bunting_0015_12632.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/014.Indigo_Bunting/Indigo_Bunting_0016_13661.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/014.Indigo_Bunting/Indigo_Bunting_0017_11574.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/014.Indigo_Bunting/Indigo_Bunting_0018_11883.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/014.Indigo_Bunting/Indigo_Bunting_0021_13979.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/014.Indigo_Bunting/Indigo_Bunting_0022_12781.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/014.Indigo_Bunting/Indigo_Bunting_0024_13523.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/014.Indigo_Bunting/Indigo_Bunting_0025_12532.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/014.Indigo_Bunting/Indigo_Bunting_0026_11964.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/014.Indigo_Bunting/Indigo_Bunting_0027_11579.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/014.Indigo_Bunting/Indigo_Bunting_0028_12335.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/014.Indigo_Bunting/Indigo_Bunting_0029_13761.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/014.Indigo_Bunting/Indigo_Bunting_0031_13300.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/014.Indigo_Bunting/Indigo_Bunting_0032_12215.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/014.Indigo_Bunting/Indigo_Bunting_0033_12777.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/014.Indigo_Bunting/Indigo_Bunting_0034_12464.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/014.Indigo_Bunting/Indigo_Bunting_0036_13716.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/014.Indigo_Bunting/Indigo_Bunting_0037_14128.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/014.Indigo_Bunting/Indigo_Bunting_0039_12756.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/014.Indigo_Bunting/Indigo_Bunting_0040_11805.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/014.Indigo_Bunting/Indigo_Bunting_0041_13987.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/014.Indigo_Bunting/Indigo_Bunting_0044_14389.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/014.Indigo_Bunting/Indigo_Bunting_0047_12966.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/014.Indigo_Bunting/Indigo_Bunting_0049_13641.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/014.Indigo_Bunting/Indigo_Bunting_0050_11811.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/014.Indigo_Bunting/Indigo_Bunting_0051_12837.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/014.Indigo_Bunting/Indigo_Bunting_0052_11893.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/014.Indigo_Bunting/Indigo_Bunting_0053_13391.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/014.Indigo_Bunting/Indigo_Bunting_0054_12213.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/014.Indigo_Bunting/Indigo_Bunting_0055_13473.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/014.Indigo_Bunting/Indigo_Bunting_0056_12637.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/014.Indigo_Bunting/Indigo_Bunting_0058_12207.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/014.Indigo_Bunting/Indigo_Bunting_0059_11596.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/014.Indigo_Bunting/Indigo_Bunting_0060_14495.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/014.Indigo_Bunting/Indigo_Bunting_0061_13259.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/014.Indigo_Bunting/Indigo_Bunting_0063_11820.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/014.Indigo_Bunting/Indigo_Bunting_0065_14558.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/014.Indigo_Bunting/Indigo_Bunting_0066_12869.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/014.Indigo_Bunting/Indigo_Bunting_0068_13081.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/014.Indigo_Bunting/Indigo_Bunting_0071_11639.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/014.Indigo_Bunting/Indigo_Bunting_0072_14197.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/014.Indigo_Bunting/Indigo_Bunting_0073_13933.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/014.Indigo_Bunting/Indigo_Bunting_0074_12829.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/014.Indigo_Bunting/Indigo_Bunting_0075_12835.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/014.Indigo_Bunting/Indigo_Bunting_0076_12950.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/014.Indigo_Bunting/Indigo_Bunting_0077_14060.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/014.Indigo_Bunting/Indigo_Bunting_0078_11852.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/014.Indigo_Bunting/Indigo_Bunting_0080_13416.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/014.Indigo_Bunting/Indigo_Bunting_0082_11907.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/014.Indigo_Bunting/Indigo_Bunting_0084_11848.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/014.Indigo_Bunting/Indigo_Bunting_0085_11991.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/015.Lazuli_Bunting/Lazuli_Bunting_0001_14916.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/015.Lazuli_Bunting/Lazuli_Bunting_0004_14887.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/015.Lazuli_Bunting/Lazuli_Bunting_0008_15195.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/015.Lazuli_Bunting/Lazuli_Bunting_0009_15163.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/015.Lazuli_Bunting/Lazuli_Bunting_0010_14915.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/015.Lazuli_Bunting/Lazuli_Bunting_0014_14824.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/015.Lazuli_Bunting/Lazuli_Bunting_0015_14690.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/015.Lazuli_Bunting/Lazuli_Bunting_0020_14837.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/015.Lazuli_Bunting/Lazuli_Bunting_0021_14686.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/015.Lazuli_Bunting/Lazuli_Bunting_0025_15079.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/015.Lazuli_Bunting/Lazuli_Bunting_0026_14669.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/015.Lazuli_Bunting/Lazuli_Bunting_0027_14895.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/015.Lazuli_Bunting/Lazuli_Bunting_0028_14950.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/015.Lazuli_Bunting/Lazuli_Bunting_0030_14986.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/015.Lazuli_Bunting/Lazuli_Bunting_0031_15018.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/015.Lazuli_Bunting/Lazuli_Bunting_0032_14778.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/015.Lazuli_Bunting/Lazuli_Bunting_0034_14864.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/015.Lazuli_Bunting/Lazuli_Bunting_0035_14920.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/015.Lazuli_Bunting/Lazuli_Bunting_0037_15021.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/015.Lazuli_Bunting/Lazuli_Bunting_0039_15081.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/015.Lazuli_Bunting/Lazuli_Bunting_0040_14923.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/015.Lazuli_Bunting/Lazuli_Bunting_0041_15152.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/015.Lazuli_Bunting/Lazuli_Bunting_0042_14820.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/015.Lazuli_Bunting/Lazuli_Bunting_0045_14954.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/015.Lazuli_Bunting/Lazuli_Bunting_0046_14787.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/015.Lazuli_Bunting/Lazuli_Bunting_0047_14863.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/015.Lazuli_Bunting/Lazuli_Bunting_0048_14844.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/015.Lazuli_Bunting/Lazuli_Bunting_0052_14618.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/015.Lazuli_Bunting/Lazuli_Bunting_0054_14714.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/015.Lazuli_Bunting/Lazuli_Bunting_0055_15043.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/015.Lazuli_Bunting/Lazuli_Bunting_0056_15032.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/015.Lazuli_Bunting/Lazuli_Bunting_0057_14775.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/015.Lazuli_Bunting/Lazuli_Bunting_0059_14749.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/015.Lazuli_Bunting/Lazuli_Bunting_0061_15155.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/015.Lazuli_Bunting/Lazuli_Bunting_0066_14914.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/015.Lazuli_Bunting/Lazuli_Bunting_0067_14672.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/015.Lazuli_Bunting/Lazuli_Bunting_0070_14665.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/015.Lazuli_Bunting/Lazuli_Bunting_0073_14594.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/015.Lazuli_Bunting/Lazuli_Bunting_0074_14854.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/015.Lazuli_Bunting/Lazuli_Bunting_0076_14662.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/015.Lazuli_Bunting/Lazuli_Bunting_0078_15164.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/015.Lazuli_Bunting/Lazuli_Bunting_0080_14893.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/015.Lazuli_Bunting/Lazuli_Bunting_0081_14709.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/015.Lazuli_Bunting/Lazuli_Bunting_0082_15047.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/015.Lazuli_Bunting/Lazuli_Bunting_0084_14815.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/015.Lazuli_Bunting/Lazuli_Bunting_0085_14627.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/015.Lazuli_Bunting/Lazuli_Bunting_0086_14992.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/015.Lazuli_Bunting/Lazuli_Bunting_0087_15096.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/015.Lazuli_Bunting/Lazuli_Bunting_0089_14598.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/015.Lazuli_Bunting/Lazuli_Bunting_0092_14656.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/015.Lazuli_Bunting/Lazuli_Bunting_0093_15030.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/015.Lazuli_Bunting/Lazuli_Bunting_0094_11894.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/015.Lazuli_Bunting/Lazuli_Bunting_0095_14919.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/015.Lazuli_Bunting/Lazuli_Bunting_0097_14617.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/015.Lazuli_Bunting/Lazuli_Bunting_0101_14873.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/015.Lazuli_Bunting/Lazuli_Bunting_0102_14605.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/015.Lazuli_Bunting/Lazuli_Bunting_0105_15017.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/015.Lazuli_Bunting/Lazuli_Bunting_0107_14705.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/016.Painted_Bunting/Painted_Bunting_0001_16585.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/016.Painted_Bunting/Painted_Bunting_0002_16887.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/016.Painted_Bunting/Painted_Bunting_0004_16641.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/016.Painted_Bunting/Painted_Bunting_0005_15202.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/016.Painted_Bunting/Painted_Bunting_0006_15249.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/016.Painted_Bunting/Painted_Bunting_0008_16416.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/016.Painted_Bunting/Painted_Bunting_0009_16674.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/016.Painted_Bunting/Painted_Bunting_0010_16948.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/016.Painted_Bunting/Painted_Bunting_0011_16690.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/016.Painted_Bunting/Painted_Bunting_0013_15294.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/016.Painted_Bunting/Painted_Bunting_0016_15200.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/016.Painted_Bunting/Painted_Bunting_0019_15231.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/016.Painted_Bunting/Painted_Bunting_0021_15295.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/016.Painted_Bunting/Painted_Bunting_0025_16722.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/016.Painted_Bunting/Painted_Bunting_0027_16536.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/016.Painted_Bunting/Painted_Bunting_0028_15205.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/016.Painted_Bunting/Painted_Bunting_0029_16530.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/016.Painted_Bunting/Painted_Bunting_0032_16605.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/016.Painted_Bunting/Painted_Bunting_0034_15207.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/016.Painted_Bunting/Painted_Bunting_0036_16563.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/016.Painted_Bunting/Painted_Bunting_0039_15235.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/016.Painted_Bunting/Painted_Bunting_0040_16691.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/016.Painted_Bunting/Painted_Bunting_0044_16557.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/016.Painted_Bunting/Painted_Bunting_0046_16535.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/016.Painted_Bunting/Painted_Bunting_0049_16869.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/016.Painted_Bunting/Painted_Bunting_0050_16670.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/016.Painted_Bunting/Painted_Bunting_0053_16404.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/016.Painted_Bunting/Painted_Bunting_0054_16711.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/016.Painted_Bunting/Painted_Bunting_0055_15208.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/016.Painted_Bunting/Painted_Bunting_0056_16599.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/016.Painted_Bunting/Painted_Bunting_0058_16719.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/016.Painted_Bunting/Painted_Bunting_0060_15224.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/016.Painted_Bunting/Painted_Bunting_0061_16930.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/016.Painted_Bunting/Painted_Bunting_0066_15241.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/016.Painted_Bunting/Painted_Bunting_0069_16462.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/016.Painted_Bunting/Painted_Bunting_0070_16515.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/016.Painted_Bunting/Painted_Bunting_0071_15209.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/016.Painted_Bunting/Painted_Bunting_0072_16697.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/016.Painted_Bunting/Painted_Bunting_0073_16737.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/016.Painted_Bunting/Painted_Bunting_0076_16765.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/016.Painted_Bunting/Painted_Bunting_0077_16819.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/016.Painted_Bunting/Painted_Bunting_0078_16565.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/016.Painted_Bunting/Painted_Bunting_0079_15197.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/016.Painted_Bunting/Painted_Bunting_0080_16534.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/016.Painted_Bunting/Painted_Bunting_0081_15230.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/016.Painted_Bunting/Painted_Bunting_0083_16587.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/016.Painted_Bunting/Painted_Bunting_0084_16531.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/016.Painted_Bunting/Painted_Bunting_0085_15282.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/016.Painted_Bunting/Painted_Bunting_0086_16540.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/016.Painted_Bunting/Painted_Bunting_0087_15232.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/016.Painted_Bunting/Painted_Bunting_0091_15198.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/016.Painted_Bunting/Painted_Bunting_0093_15212.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/016.Painted_Bunting/Painted_Bunting_0094_16467.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/016.Painted_Bunting/Painted_Bunting_0096_15233.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/016.Painted_Bunting/Painted_Bunting_0098_15226.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/016.Painted_Bunting/Painted_Bunting_0099_16525.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/016.Painted_Bunting/Painted_Bunting_0100_16735.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/016.Painted_Bunting/Painted_Bunting_0102_16642.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/017.Cardinal/Cardinal_0001_17057.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/017.Cardinal/Cardinal_0002_18424.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/017.Cardinal/Cardinal_0006_17684.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/017.Cardinal/Cardinal_0007_18537.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/017.Cardinal/Cardinal_0010_18894.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/017.Cardinal/Cardinal_0012_18638.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/017.Cardinal/Cardinal_0014_17389.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/017.Cardinal/Cardinal_0016_17862.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/017.Cardinal/Cardinal_0018_17071.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/017.Cardinal/Cardinal_0019_17368.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/017.Cardinal/Cardinal_0020_18664.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/017.Cardinal/Cardinal_0022_17233.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/017.Cardinal/Cardinal_0023_19026.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/017.Cardinal/Cardinal_0025_17239.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/017.Cardinal/Cardinal_0028_18054.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/017.Cardinal/Cardinal_0029_17297.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/017.Cardinal/Cardinal_0033_19215.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/017.Cardinal/Cardinal_0035_17678.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/017.Cardinal/Cardinal_0037_18092.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/017.Cardinal/Cardinal_0038_19203.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/017.Cardinal/Cardinal_0039_17597.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/017.Cardinal/Cardinal_0040_17477.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/017.Cardinal/Cardinal_0041_17189.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/017.Cardinal/Cardinal_0045_18021.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/017.Cardinal/Cardinal_0047_17673.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/017.Cardinal/Cardinal_0049_18258.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/017.Cardinal/Cardinal_0050_18035.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/017.Cardinal/Cardinal_0051_17223.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/017.Cardinal/Cardinal_0052_18334.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/017.Cardinal/Cardinal_0053_17541.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/017.Cardinal/Cardinal_0054_19334.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/017.Cardinal/Cardinal_0055_18898.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/017.Cardinal/Cardinal_0056_18352.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/017.Cardinal/Cardinal_0057_17128.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/017.Cardinal/Cardinal_0060_17139.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/017.Cardinal/Cardinal_0062_17334.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/017.Cardinal/Cardinal_0066_17803.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/017.Cardinal/Cardinal_0072_17159.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/017.Cardinal/Cardinal_0074_18339.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/017.Cardinal/Cardinal_0075_17946.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/017.Cardinal/Cardinal_0078_17181.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/017.Cardinal/Cardinal_0079_19044.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/017.Cardinal/Cardinal_0081_17291.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/017.Cardinal/Cardinal_0082_17875.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/017.Cardinal/Cardinal_0083_18042.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/017.Cardinal/Cardinal_0084_17576.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/017.Cardinal/Cardinal_0085_19162.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/017.Cardinal/Cardinal_0089_18005.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/017.Cardinal/Cardinal_0092_17591.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/017.Cardinal/Cardinal_0093_17676.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/017.Cardinal/Cardinal_0094_17165.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/017.Cardinal/Cardinal_0095_18108.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/017.Cardinal/Cardinal_0097_17396.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/017.Cardinal/Cardinal_0102_17808.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/017.Cardinal/Cardinal_0103_17425.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/017.Cardinal/Cardinal_0104_17122.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/017.Cardinal/Cardinal_0105_19045.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/018.Spotted_Catbird/Spotted_Catbird_0001_796797.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/018.Spotted_Catbird/Spotted_Catbird_0002_796827.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/018.Spotted_Catbird/Spotted_Catbird_0003_796822.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/018.Spotted_Catbird/Spotted_Catbird_0004_796804.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/018.Spotted_Catbird/Spotted_Catbird_0005_19411.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/018.Spotted_Catbird/Spotted_Catbird_0006_796823.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/018.Spotted_Catbird/Spotted_Catbird_0007_19424.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/018.Spotted_Catbird/Spotted_Catbird_0008_796789.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/018.Spotted_Catbird/Spotted_Catbird_0009_796795.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/018.Spotted_Catbird/Spotted_Catbird_0010_19436.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/018.Spotted_Catbird/Spotted_Catbird_0011_796815.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/018.Spotted_Catbird/Spotted_Catbird_0012_796802.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/018.Spotted_Catbird/Spotted_Catbird_0013_19428.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/018.Spotted_Catbird/Spotted_Catbird_0014_19425.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/018.Spotted_Catbird/Spotted_Catbird_0015_796808.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/018.Spotted_Catbird/Spotted_Catbird_0016_796803.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/018.Spotted_Catbird/Spotted_Catbird_0018_796800.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/018.Spotted_Catbird/Spotted_Catbird_0019_796788.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/018.Spotted_Catbird/Spotted_Catbird_0020_796794.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/018.Spotted_Catbird/Spotted_Catbird_0021_796798.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/018.Spotted_Catbird/Spotted_Catbird_0022_796821.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/018.Spotted_Catbird/Spotted_Catbird_0023_796793.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/018.Spotted_Catbird/Spotted_Catbird_0024_796791.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/018.Spotted_Catbird/Spotted_Catbird_0025_796819.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/018.Spotted_Catbird/Spotted_Catbird_0026_796818.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/018.Spotted_Catbird/Spotted_Catbird_0027_796796.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/018.Spotted_Catbird/Spotted_Catbird_0028_796813.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/018.Spotted_Catbird/Spotted_Catbird_0029_796825.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/018.Spotted_Catbird/Spotted_Catbird_0030_796801.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/018.Spotted_Catbird/Spotted_Catbird_0031_796806.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/018.Spotted_Catbird/Spotted_Catbird_0032_796828.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/018.Spotted_Catbird/Spotted_Catbird_0033_796805.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/018.Spotted_Catbird/Spotted_Catbird_0034_19437.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/018.Spotted_Catbird/Spotted_Catbird_0035_796799.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/018.Spotted_Catbird/Spotted_Catbird_0036_19406.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/018.Spotted_Catbird/Spotted_Catbird_0037_796810.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/018.Spotted_Catbird/Spotted_Catbird_0039_796811.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/018.Spotted_Catbird/Spotted_Catbird_0040_796820.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/018.Spotted_Catbird/Spotted_Catbird_0041_796807.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/018.Spotted_Catbird/Spotted_Catbird_0042_19430.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/018.Spotted_Catbird/Spotted_Catbird_0044_796809.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/018.Spotted_Catbird/Spotted_Catbird_0045_796814.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/018.Spotted_Catbird/Spotted_Catbird_0046_19399.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/018.Spotted_Catbird/Spotted_Catbird_0047_19400.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/018.Spotted_Catbird/Spotted_Catbird_0048_796817.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/019.Gray_Catbird/Gray_Catbird_0001_20695.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/019.Gray_Catbird/Gray_Catbird_0002_21395.jpg',\n",
       " '/home/pranav/embedding_classification/CUB_200_2011/images/019.Gray_Catbird/Gray_Catbird_0006_20867.jpg',\n",
       " ...]"
      ]
     },
     "execution_count": 6,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "filepaths = dataset.values('filepath')"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "324187a1-a23c-4d34-842c-e5ec95582de5",
   "metadata": {},
   "source": [
    "## filter classes to last 100, since fiftyone cant show more than 100 classes at once"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 7,
   "id": "e7031f93-ca9c-45c9-9c4e-b9bd54786bb6",
   "metadata": {},
   "outputs": [],
   "source": [
    "filepaths = dataset.values('filepath')\n",
    "with open('fiftyone_filepaths.txt','r') as f:\n",
    "    filepaths = f.readlines()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 8,
   "id": "acf3ab88-4752-4235-8bcb-1b63b1a98884",
   "metadata": {},
   "outputs": [],
   "source": [
    "filepaths = [i.replace('\\n','') for i in filepaths]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 9,
   "id": "4d4d181c-7749-424a-a60f-ca8c747285b3",
   "metadata": {
    "scrolled": true
   },
   "outputs": [],
   "source": [
    "pos_view = dataset.filter_labels(\"ground_truth\", F(\"label\").is_in(tuple(classes[100:])))"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 10,
   "id": "d04fb893-32fd-4d21-ae13-11740ab5e273",
   "metadata": {},
   "outputs": [],
   "source": [
    "filtered_paths = (np.array(filepaths)[:, None] == np.array(pos_view.values('filepath'))).argmax(axis=0)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 11,
   "id": "02b0e387-9e49-461d-9068-435cec63972a",
   "metadata": {},
   "outputs": [],
   "source": [
    "embeddings = np.load('fiftyone_embeddings.npy')\n",
    "# embeddings = []\n",
    "# for path in filepaths:\n",
    "#     img = PIL.Image.open(path.replace('/home/pranav/embedding_classification/',''))\n",
    "#     if len(img.getbands()) == 1:\n",
    "#         img = img.convert(\"RGB\")\n",
    "#     transformed_img = test_transform(img)\n",
    "#     embedding = model(torch.unsqueeze(transformed_img,dim=0).to(device))\n",
    "#     embeddings.append(embedding.detach().cpu().numpy()[0])"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 12,
   "id": "782f0bed-fc30-4494-b918-79a60c47ef27",
   "metadata": {},
   "outputs": [],
   "source": [
    "filtered_embeddings = np.take(embeddings,filtered_paths,axis=0)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 13,
   "id": "1b348f84-0126-4cbe-b764-ba608f5d518a",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Generating visualization...\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "/home/pranav/embedding_classification/fiftyone_env/lib/python3.8/site-packages/umap/umap_.py:1945: UserWarning: n_jobs value 1 overridden to 1 by setting random_state. Use no seed for parallelism.\n",
      "  warn(f\"n_jobs value {self.n_jobs} overridden to 1 by setting random_state. Use no seed for parallelism.\")\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "UMAP(n_jobs=1, random_state=51, verbose=True)\n",
      "Mon May  6 23:26:14 2024 Construct fuzzy simplicial set\n",
      "Mon May  6 23:26:14 2024 Finding Nearest Neighbors\n",
      "Mon May  6 23:26:14 2024 Building RP forest with 9 trees\n",
      "Mon May  6 23:26:16 2024 NN descent for 13 iterations\n",
      "\t 1  /  13\n",
      "\t 2  /  13\n",
      "\t 3  /  13\n",
      "\t 4  /  13\n",
      "\tStopping threshold met -- exiting after 4 iterations\n",
      "Mon May  6 23:26:21 2024 Finished Nearest Neighbor Search\n",
      "Mon May  6 23:26:22 2024 Construct embedding\n"
     ]
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "a2eff190e6ca485db1e3b36527312fb1",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "Epochs completed:   0%|            0/500 [00:00]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "\tcompleted  0  /  500 epochs\n",
      "\tcompleted  50  /  500 epochs\n",
      "\tcompleted  100  /  500 epochs\n",
      "\tcompleted  150  /  500 epochs\n",
      "\tcompleted  200  /  500 epochs\n",
      "\tcompleted  250  /  500 epochs\n",
      "\tcompleted  300  /  500 epochs\n",
      "\tcompleted  350  /  500 epochs\n",
      "\tcompleted  400  /  500 epochs\n",
      "\tcompleted  450  /  500 epochs\n",
      "Mon May  6 23:26:27 2024 Finished embedding\n"
     ]
    }
   ],
   "source": [
    "# Compute visualization\n",
    "results = fob.compute_visualization(\n",
    "    pos_view, embeddings=filtered_embeddings, seed=51, brain_key=\"resnet50\"\n",
    ")"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 14,
   "id": "905cbaf4-cfb4-4aab-afa6-384da73eed3d",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/html": [
       "\n",
       "        <iframe\n",
       "            width=\"100%\"\n",
       "            height=\"800\"\n",
       "            src=\"http://localhost:8051/?notebook=True&subscription=08af0a4a-67d1-4e4d-8b7b-4a92e1c1d220\"\n",
       "            frameborder=\"0\"\n",
       "            allowfullscreen\n",
       "            \n",
       "        ></iframe>\n",
       "        "
      ],
      "text/plain": [
       "<IPython.lib.display.IFrame at 0x7fe916309070>"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Notebook sessions cannot wait\n"
     ]
    }
   ],
   "source": [
    "session = fo.launch_app(pos_view,port=8051)\n",
    "session.wait()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 1,
   "id": "8be1be8e-82c8-46a8-bb8a-2903cc01d5a9",
   "metadata": {},
   "outputs": [],
   "source": [
    "import glob\n",
    "images = glob.glob('CUB_200_2011/images/**/*.jpg')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "id": "0dfcf291-970b-48eb-be31-2d8de75b823f",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "11788"
      ]
     },
     "execution_count": 3,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "len(images)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "8e05bab7-ff79-4af8-b0d5-2279c49cbe93",
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3 (ipykernel)",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.10.11"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 5
}