dataset.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 3,
   "id": "c303cb31",
   "metadata": {},
   "outputs": [],
   "source": [
    "import imageio as iio\n",
    "import numpy as np\n",
    "from tqdm import tqdm\n",
    "import os"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 135,
   "id": "83050e1a",
   "metadata": {},
   "outputs": [],
   "source": [
    "class Dataset:\n",
    "    \n",
    "    path_to_dataset = \"./flower/\"\n",
    "    files_list = []\n",
    "    dataset = []\n",
    "    image_shape = [0, 0]\n",
    "    total_images = 0\n",
    "    \n",
    "    def __init__(self, path = None):\n",
    "        if path != None:\n",
    "            self.path_to_dataset = path\n",
    "        self.file_list()\n",
    "    \n",
    "    def file_list(self):\n",
    "        self.files_list = os.listdir(self.path_to_dataset)\n",
    "        self.total_images = len(self.files_list)\n",
    "        return self.files_list\n",
    "    \n",
    "    def read_image(self, image_name):\n",
    "        return iio.imread(self.path_to_dataset + image_name)\n",
    "    \n",
    "    def load_data(self):\n",
    "        for file in self.files_list:\n",
    "            self.dataset.append([self.read_image(file)])\n",
    "        self.dataset = np.array(self.dataset)\n",
    "        self.image_shape = np.array(file).shape\n",
    "        \n",
    "        self.dataset = np.squeeze(self.dataset)\n",
    "        \n",
    "        return self.dataset"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 143,
   "id": "bfc24024",
   "metadata": {},
   "outputs": [],
   "source": [
    "data = Dataset()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 144,
   "id": "684db4a8",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "['1000_64.jpg',\n",
       " '1001_64.jpg',\n",
       " '1002_64.jpg',\n",
       " '1003_64.jpg',\n",
       " '1004_64.jpg',\n",
       " '1005_64.jpg',\n",
       " '1006_64.jpg',\n",
       " '1007_64.jpg',\n",
       " '1008_64.jpg',\n",
       " '1009_64.jpg',\n",
       " '100_64.jpg',\n",
       " '100_64_1.jpg',\n",
       " '1010_64.jpg',\n",
       " '1011_64.jpg',\n",
       " '1012_64.jpg',\n",
       " '1013_64.jpg',\n",
       " '1014_64.jpg',\n",
       " '1015_64.jpg',\n",
       " '1016_64.jpg',\n",
       " '1017_64.jpg',\n",
       " '1018_64.jpg',\n",
       " '1019_64.jpg',\n",
       " '101_64.jpg',\n",
       " '101_64_1.jpg',\n",
       " '1020_64.jpg',\n",
       " '1021_64.jpg',\n",
       " '1022_64.jpg',\n",
       " '1023_64.jpg',\n",
       " '1024_64.jpg',\n",
       " '1025_64.jpg',\n",
       " '1026_64.jpg',\n",
       " '1027_64.jpg',\n",
       " '1028_64.jpg',\n",
       " '1029_64.jpg',\n",
       " '102_64.jpg',\n",
       " '102_64_1.jpg',\n",
       " '1030_64.jpg',\n",
       " '1031_64.jpg',\n",
       " '1032_64.jpg',\n",
       " '1033_64.jpg',\n",
       " '1034_64.jpg',\n",
       " '1035_64.jpg',\n",
       " '1036_64.jpg',\n",
       " '1037_64.jpg',\n",
       " '1038_64.jpg',\n",
       " '1039_64.jpg',\n",
       " '103_64.jpg',\n",
       " '103_64_1.jpg',\n",
       " '1040_64.jpg',\n",
       " '1041_64.jpg',\n",
       " '1042_64.jpg',\n",
       " '1043_64.jpg',\n",
       " '1044_64.jpg',\n",
       " '1045_64.jpg',\n",
       " '1046_64.jpg',\n",
       " '1047_64.jpg',\n",
       " '1048_64.jpg',\n",
       " '1049_64.jpg',\n",
       " '104_64.jpg',\n",
       " '104_64_1.jpg',\n",
       " '1050_64.jpg',\n",
       " '1051_64.jpg',\n",
       " '1052_64.jpg',\n",
       " '1053_64.jpg',\n",
       " '1054_64.jpg',\n",
       " '1055_64.jpg',\n",
       " '1056_64.jpg',\n",
       " '1057_64.jpg',\n",
       " '1058_64.jpg',\n",
       " '1059_64.jpg',\n",
       " '105_64.jpg',\n",
       " '105_64_1.jpg',\n",
       " '1060_64.jpg',\n",
       " '1061_64.jpg',\n",
       " '1062_64.jpg',\n",
       " '1063_64.jpg',\n",
       " '1064_64.jpg',\n",
       " '1065_64.jpg',\n",
       " '1066_64.jpg',\n",
       " '1067_64.jpg',\n",
       " '1068_64.jpg',\n",
       " '1069_64.jpg',\n",
       " '106_64.jpg',\n",
       " '106_64_1.jpg',\n",
       " '1070_64.jpg',\n",
       " '1071_64.jpg',\n",
       " '1072_64.jpg',\n",
       " '1073_64.jpg',\n",
       " '1074_64.jpg',\n",
       " '1075_64.jpg',\n",
       " '1076_64.jpg',\n",
       " '1077_64.jpg',\n",
       " '1078_64.jpg',\n",
       " '1079_64.jpg',\n",
       " '107_64.jpg',\n",
       " '107_64_1.jpg',\n",
       " '1080_64.jpg',\n",
       " '1081_64.jpg',\n",
       " '1082_64.jpg',\n",
       " '1083_64.jpg',\n",
       " '1084_64.jpg',\n",
       " '1085_64.jpg',\n",
       " '1086_64.jpg',\n",
       " '1087_64.jpg',\n",
       " '1088_64.jpg',\n",
       " '1089_64.jpg',\n",
       " '108_64.jpg',\n",
       " '108_64_1.jpg',\n",
       " '1090_64.jpg',\n",
       " '1091_64.jpg',\n",
       " '1092_64.jpg',\n",
       " '1093_64.jpg',\n",
       " '1094_64.jpg',\n",
       " '1095_64.jpg',\n",
       " '1096_64.jpg',\n",
       " '1097_64.jpg',\n",
       " '1098_64.jpg',\n",
       " '1099_64.jpg',\n",
       " '109_64.jpg',\n",
       " '109_64_1.jpg',\n",
       " '10_64.jpg',\n",
       " '10_64_1.jpg',\n",
       " '1100_64.jpg',\n",
       " '1101_64.jpg',\n",
       " '1102_64.jpg',\n",
       " '1103_64.jpg',\n",
       " '1104_64.jpg',\n",
       " '1105_64.jpg',\n",
       " '1106_64.jpg',\n",
       " '1107_64.jpg',\n",
       " '1108_64.jpg',\n",
       " '1109_64.jpg',\n",
       " '110_64.jpg',\n",
       " '110_64_1.jpg',\n",
       " '1110_64.jpg',\n",
       " '1111_64.jpg',\n",
       " '1112_64.jpg',\n",
       " '1113_64.jpg',\n",
       " '1114_64.jpg',\n",
       " '1115_64.jpg',\n",
       " '1116_64.jpg',\n",
       " '1117_64.jpg',\n",
       " '1118_64.jpg',\n",
       " '1119_64.jpg',\n",
       " '111_64.jpg',\n",
       " '111_64_1.jpg',\n",
       " '1120_64.jpg',\n",
       " '1121_64.jpg',\n",
       " '1122_64.jpg',\n",
       " '1123_64.jpg',\n",
       " '1124_64.jpg',\n",
       " '1125_64.jpg',\n",
       " '1126_64.jpg',\n",
       " '1127_64.jpg',\n",
       " '1128_64.jpg',\n",
       " '1129_64.jpg',\n",
       " '112_64.jpg',\n",
       " '112_64_1.jpg',\n",
       " '1130_64.jpg',\n",
       " '1131_64.jpg',\n",
       " '1132_64.jpg',\n",
       " '1133_64.jpg',\n",
       " '1134_64.jpg',\n",
       " '1135_64.jpg',\n",
       " '1136_64.jpg',\n",
       " '1137_64.jpg',\n",
       " '1138_64.jpg',\n",
       " '1139_64.jpg',\n",
       " '113_64.jpg',\n",
       " '113_64_1.jpg',\n",
       " '1140_64.jpg',\n",
       " '1141_64.jpg',\n",
       " '1142_64.jpg',\n",
       " '1143_64.jpg',\n",
       " '1144_64.jpg',\n",
       " '1145_64.jpg',\n",
       " '1146_64.jpg',\n",
       " '1147_64.jpg',\n",
       " '1148_64.jpg',\n",
       " '1149_64.jpg',\n",
       " '114_64.jpg',\n",
       " '114_64_1.jpg',\n",
       " '1150_64.jpg',\n",
       " '1151_64.jpg',\n",
       " '1152_64.jpg',\n",
       " '1153_64.jpg',\n",
       " '1154_64.jpg',\n",
       " '1155_64.jpg',\n",
       " '1156_64.jpg',\n",
       " '1157_64.jpg',\n",
       " '1158_64.jpg',\n",
       " '1159_64.jpg',\n",
       " '115_64.jpg',\n",
       " '115_64_1.jpg',\n",
       " '1160_64.jpg',\n",
       " '1161_64.jpg',\n",
       " '1162_64.jpg',\n",
       " '1163_64.jpg',\n",
       " '1164_64.jpg',\n",
       " '1165_64.jpg',\n",
       " '1166_64.jpg',\n",
       " '1167_64.jpg',\n",
       " '1168_64.jpg',\n",
       " '1169_64.jpg',\n",
       " '116_64.jpg',\n",
       " '116_64_1.jpg',\n",
       " '1170_64.jpg',\n",
       " '1171_64.jpg',\n",
       " '1172_64.jpg',\n",
       " '1173_64.jpg',\n",
       " '1174_64.jpg',\n",
       " '1175_64.jpg',\n",
       " '1176_64.jpg',\n",
       " '1177_64.jpg',\n",
       " '1178_64.jpg',\n",
       " '1179_64.jpg',\n",
       " '117_64.jpg',\n",
       " '117_64_1.jpg',\n",
       " '1180_64.jpg',\n",
       " '1181_64.jpg',\n",
       " '1182_64.jpg',\n",
       " '1183_64.jpg',\n",
       " '1184_64.jpg',\n",
       " '1185_64.jpg',\n",
       " '1186_64.jpg',\n",
       " '1187_64.jpg',\n",
       " '1188_64.jpg',\n",
       " '1189_64.jpg',\n",
       " '118_64.jpg',\n",
       " '118_64_1.jpg',\n",
       " '1190_64.jpg',\n",
       " '1191_64.jpg',\n",
       " '1192_64.jpg',\n",
       " '1193_64.jpg',\n",
       " '1194_64.jpg',\n",
       " '1195_64.jpg',\n",
       " '1196_64.jpg',\n",
       " '1197_64.jpg',\n",
       " '1198_64.jpg',\n",
       " '1199_64.jpg',\n",
       " '119_64.jpg',\n",
       " '119_64_1.jpg',\n",
       " '11_64.jpg',\n",
       " '11_64_1.jpg',\n",
       " '1200_64.jpg',\n",
       " '120_64.jpg',\n",
       " '120_64_1.jpg',\n",
       " '121_64.jpg',\n",
       " '121_64_1.jpg',\n",
       " '122_64.jpg',\n",
       " '122_64_1.jpg',\n",
       " '123_64.jpg',\n",
       " '123_64_1.jpg',\n",
       " '124_64.jpg',\n",
       " '124_64_1.jpg',\n",
       " '125_64.jpg',\n",
       " '125_64_1.jpg',\n",
       " '126_64.jpg',\n",
       " '126_64_1.jpg',\n",
       " '127_64.jpg',\n",
       " '127_64_1.jpg',\n",
       " '128_64.jpg',\n",
       " '128_64_1.jpg',\n",
       " '129_64.jpg',\n",
       " '129_64_1.jpg',\n",
       " '12_64.jpg',\n",
       " '12_64_1.jpg',\n",
       " '130_64.jpg',\n",
       " '130_64_1.jpg',\n",
       " '131_64.jpg',\n",
       " '131_64_1.jpg',\n",
       " '132_64.jpg',\n",
       " '132_64_1.jpg',\n",
       " '133_64.jpg',\n",
       " '133_64_1.jpg',\n",
       " '134_64.jpg',\n",
       " '134_64_1.jpg',\n",
       " '135_64.jpg',\n",
       " '135_64_1.jpg',\n",
       " '136_64.jpg',\n",
       " '136_64_1.jpg',\n",
       " '137_64.jpg',\n",
       " '137_64_1.jpg',\n",
       " '138_64.jpg',\n",
       " '138_64_1.jpg',\n",
       " '139_64.jpg',\n",
       " '139_64_1.jpg',\n",
       " '13_64.jpg',\n",
       " '13_64_1.jpg',\n",
       " '140_64.jpg',\n",
       " '140_64_1.jpg',\n",
       " '141_64.jpg',\n",
       " '141_64_1.jpg',\n",
       " '142_64.jpg',\n",
       " '142_64_1.jpg',\n",
       " '143_64.jpg',\n",
       " '143_64_1.jpg',\n",
       " '144_64.jpg',\n",
       " '144_64_1.jpg',\n",
       " '145_64.jpg',\n",
       " '145_64_1.jpg',\n",
       " '146_64.jpg',\n",
       " '146_64_1.jpg',\n",
       " '147_64.jpg',\n",
       " '147_64_1.jpg',\n",
       " '148_64.jpg',\n",
       " '148_64_1.jpg',\n",
       " '149_64.jpg',\n",
       " '149_64_1.jpg',\n",
       " '14_64.jpg',\n",
       " '14_64_1.jpg',\n",
       " '150_64.jpg',\n",
       " '150_64_1.jpg',\n",
       " '151_64.jpg',\n",
       " '151_64_1.jpg',\n",
       " '152_64.jpg',\n",
       " '152_64_1.jpg',\n",
       " '153_64.jpg',\n",
       " '153_64_1.jpg',\n",
       " '154_64.jpg',\n",
       " '154_64_1.jpg',\n",
       " '155_64.jpg',\n",
       " '155_64_1.jpg',\n",
       " '156_64.jpg',\n",
       " '156_64_1.jpg',\n",
       " '157_64.jpg',\n",
       " '157_64_1.jpg',\n",
       " '158_64.jpg',\n",
       " '158_64_1.jpg',\n",
       " '159_64.jpg',\n",
       " '159_64_1.jpg',\n",
       " '15_64.jpg',\n",
       " '15_64_1.jpg',\n",
       " '160_64.jpg',\n",
       " '160_64_1.jpg',\n",
       " '161_64.jpg',\n",
       " '161_64_1.jpg',\n",
       " '162_64.jpg',\n",
       " '162_64_1.jpg',\n",
       " '163_64.jpg',\n",
       " '163_64_1.jpg',\n",
       " '164_64.jpg',\n",
       " '164_64_1.jpg',\n",
       " '165_64.jpg',\n",
       " '165_64_1.jpg',\n",
       " '166_64.jpg',\n",
       " '166_64_1.jpg',\n",
       " '167_64.jpg',\n",
       " '167_64_1.jpg',\n",
       " '168_64.jpg',\n",
       " '168_64_1.jpg',\n",
       " '169_64.jpg',\n",
       " '169_64_1.jpg',\n",
       " '16_64.jpg',\n",
       " '16_64_1.jpg',\n",
       " '170_64.jpg',\n",
       " '170_64_1.jpg',\n",
       " '171_64.jpg',\n",
       " '171_64_1.jpg',\n",
       " '172_64.jpg',\n",
       " '172_64_1.jpg',\n",
       " '173_64.jpg',\n",
       " '173_64_1.jpg',\n",
       " '174_64.jpg',\n",
       " '174_64_1.jpg',\n",
       " '175_64.jpg',\n",
       " '175_64_1.jpg',\n",
       " '176_64.jpg',\n",
       " '176_64_1.jpg',\n",
       " '177_64.jpg',\n",
       " '177_64_1.jpg',\n",
       " '178_64.jpg',\n",
       " '178_64_1.jpg',\n",
       " '179_64.jpg',\n",
       " '179_64_1.jpg',\n",
       " '17_64.jpg',\n",
       " '17_64_1.jpg',\n",
       " '180_64.jpg',\n",
       " '180_64_1.jpg',\n",
       " '181_64.jpg',\n",
       " '181_64_1.jpg',\n",
       " '182_64.jpg',\n",
       " '182_64_1.jpg',\n",
       " '183_64.jpg',\n",
       " '183_64_1.jpg',\n",
       " '184_64.jpg',\n",
       " '184_64_1.jpg',\n",
       " '185_64.jpg',\n",
       " '185_64_1.jpg',\n",
       " '186_64.jpg',\n",
       " '186_64_1.jpg',\n",
       " '187_64.jpg',\n",
       " '187_64_1.jpg',\n",
       " '188_64.jpg',\n",
       " '188_64_1.jpg',\n",
       " '189_64.jpg',\n",
       " '189_64_1.jpg',\n",
       " '18_64.jpg',\n",
       " '18_64_1.jpg',\n",
       " '190_64.jpg',\n",
       " '190_64_1.jpg',\n",
       " '191_64.jpg',\n",
       " '191_64_1.jpg',\n",
       " '192_64.jpg',\n",
       " '192_64_1.jpg',\n",
       " '193_64.jpg',\n",
       " '193_64_1.jpg',\n",
       " '194_64.jpg',\n",
       " '194_64_1.jpg',\n",
       " '195_64.jpg',\n",
       " '195_64_1.jpg',\n",
       " '196_64.jpg',\n",
       " '196_64_1.jpg',\n",
       " '197_64.jpg',\n",
       " '197_64_1.jpg',\n",
       " '198_64.jpg',\n",
       " '198_64_1.jpg',\n",
       " '199_64.jpg',\n",
       " '199_64_1.jpg',\n",
       " '19_64.jpg',\n",
       " '19_64_1.jpg',\n",
       " '1_64.jpg',\n",
       " '1_64_1.jpg',\n",
       " '200_64.jpg',\n",
       " '200_64_1.jpg',\n",
       " '201_64.jpg',\n",
       " '201_64_1.jpg',\n",
       " '202_64.jpg',\n",
       " '202_64_1.jpg',\n",
       " '203_64.jpg',\n",
       " '203_64_1.jpg',\n",
       " '204_64.jpg',\n",
       " '204_64_1.jpg',\n",
       " '205_64.jpg',\n",
       " '205_64_1.jpg',\n",
       " '206_64.jpg',\n",
       " '206_64_1.jpg',\n",
       " '207_64.jpg',\n",
       " '207_64_1.jpg',\n",
       " '208_64.jpg',\n",
       " '208_64_1.jpg',\n",
       " '209_64.jpg',\n",
       " '209_64_1.jpg',\n",
       " '20_64.jpg',\n",
       " '20_64_1.jpg',\n",
       " '210_64.jpg',\n",
       " '210_64_1.jpg',\n",
       " '211_64.jpg',\n",
       " '211_64_1.jpg',\n",
       " '212_64.jpg',\n",
       " '212_64_1.jpg',\n",
       " '213_64.jpg',\n",
       " '213_64_1.jpg',\n",
       " '214_64.jpg',\n",
       " '214_64_1.jpg',\n",
       " '215_64.jpg',\n",
       " '215_64_1.jpg',\n",
       " '216_64.jpg',\n",
       " '216_64_1.jpg',\n",
       " '217_64.jpg',\n",
       " '217_64_1.jpg',\n",
       " '218_64.jpg',\n",
       " '218_64_1.jpg',\n",
       " '219_64.jpg',\n",
       " '219_64_1.jpg',\n",
       " '21_64.jpg',\n",
       " '21_64_1.jpg',\n",
       " '220_64.jpg',\n",
       " '220_64_1.jpg',\n",
       " '221_64.jpg',\n",
       " '221_64_1.jpg',\n",
       " '222_64.jpg',\n",
       " '222_64_1.jpg',\n",
       " '223_64.jpg',\n",
       " '223_64_1.jpg',\n",
       " '224_64.jpg',\n",
       " '224_64_1.jpg',\n",
       " '225_64.jpg',\n",
       " '225_64_1.jpg',\n",
       " '226_64.jpg',\n",
       " '226_64_1.jpg',\n",
       " '227_64.jpg',\n",
       " '227_64_1.jpg',\n",
       " '228_64.jpg',\n",
       " '228_64_1.jpg',\n",
       " '229_64.jpg',\n",
       " '229_64_1.jpg',\n",
       " '22_64.jpg',\n",
       " '22_64_1.jpg',\n",
       " '230_64.jpg',\n",
       " '230_64_1.jpg',\n",
       " '231_64.jpg',\n",
       " '231_64_1.jpg',\n",
       " '232_64.jpg',\n",
       " '232_64_1.jpg',\n",
       " '233_64.jpg',\n",
       " '233_64_1.jpg',\n",
       " '234_64.jpg',\n",
       " '234_64_1.jpg',\n",
       " '235_64.jpg',\n",
       " '235_64_1.jpg',\n",
       " '236_64.jpg',\n",
       " '236_64_1.jpg',\n",
       " '237_64.jpg',\n",
       " '237_64_1.jpg',\n",
       " '238_64.jpg',\n",
       " '238_64_1.jpg',\n",
       " '239_64.jpg',\n",
       " '239_64_1.jpg',\n",
       " '23_64.jpg',\n",
       " '23_64_1.jpg',\n",
       " '240_64.jpg',\n",
       " '240_64_1.jpg',\n",
       " '241_64.jpg',\n",
       " '241_64_1.jpg',\n",
       " '242_64.jpg',\n",
       " '242_64_1.jpg',\n",
       " '243_64.jpg',\n",
       " '243_64_1.jpg',\n",
       " '244_64.jpg',\n",
       " '244_64_1.jpg',\n",
       " '245_64.jpg',\n",
       " '245_64_1.jpg',\n",
       " '246_64.jpg',\n",
       " '246_64_1.jpg',\n",
       " '247_64.jpg',\n",
       " '247_64_1.jpg',\n",
       " '248_64.jpg',\n",
       " '248_64_1.jpg',\n",
       " '249_64.jpg',\n",
       " '249_64_1.jpg',\n",
       " '24_64.jpg',\n",
       " '24_64_1.jpg',\n",
       " '250_64.jpg',\n",
       " '250_64_1.jpg',\n",
       " '251_64.jpg',\n",
       " '251_64_1.jpg',\n",
       " '252_64.jpg',\n",
       " '252_64_1.jpg',\n",
       " '253_64.jpg',\n",
       " '253_64_1.jpg',\n",
       " '254_64.jpg',\n",
       " '254_64_1.jpg',\n",
       " '255_64.jpg',\n",
       " '255_64_1.jpg',\n",
       " '256_64.jpg',\n",
       " '256_64_1.jpg',\n",
       " '257_64.jpg',\n",
       " '257_64_1.jpg',\n",
       " '258_64.jpg',\n",
       " '258_64_1.jpg',\n",
       " '259_64.jpg',\n",
       " '259_64_1.jpg',\n",
       " '25_64.jpg',\n",
       " '25_64_1.jpg',\n",
       " '260_64.jpg',\n",
       " '260_64_1.jpg',\n",
       " '261_64.jpg',\n",
       " '261_64_1.jpg',\n",
       " '262_64.jpg',\n",
       " '262_64_1.jpg',\n",
       " '263_64.jpg',\n",
       " '263_64_1.jpg',\n",
       " '264_64.jpg',\n",
       " '264_64_1.jpg',\n",
       " '265_64.jpg',\n",
       " '265_64_1.jpg',\n",
       " '266_64.jpg',\n",
       " '266_64_1.jpg',\n",
       " '267_64.jpg',\n",
       " '267_64_1.jpg',\n",
       " '268_64.jpg',\n",
       " '268_64_1.jpg',\n",
       " '269_64.jpg',\n",
       " '269_64_1.jpg',\n",
       " '26_64.jpg',\n",
       " '26_64_1.jpg',\n",
       " '270_64.jpg',\n",
       " '270_64_1.jpg',\n",
       " '271_64.jpg',\n",
       " '271_64_1.jpg',\n",
       " '272_64.jpg',\n",
       " '272_64_1.jpg',\n",
       " '273_64.jpg',\n",
       " '273_64_1.jpg',\n",
       " '274_64.jpg',\n",
       " '274_64_1.jpg',\n",
       " '275_64.jpg',\n",
       " '275_64_1.jpg',\n",
       " '276_64.jpg',\n",
       " '276_64_1.jpg',\n",
       " '277_64.jpg',\n",
       " '277_64_1.jpg',\n",
       " '278_64.jpg',\n",
       " '278_64_1.jpg',\n",
       " '279_64.jpg',\n",
       " '279_64_1.jpg',\n",
       " '27_64.jpg',\n",
       " '27_64_1.jpg',\n",
       " '280_64.jpg',\n",
       " '280_64_1.jpg',\n",
       " '281_64.jpg',\n",
       " '281_64_1.jpg',\n",
       " '282_64.jpg',\n",
       " '282_64_1.jpg',\n",
       " '283_64.jpg',\n",
       " '283_64_1.jpg',\n",
       " '284_64.jpg',\n",
       " '284_64_1.jpg',\n",
       " '285_64.jpg',\n",
       " '285_64_1.jpg',\n",
       " '286_64.jpg',\n",
       " '286_64_1.jpg',\n",
       " '287_64.jpg',\n",
       " '287_64_1.jpg',\n",
       " '288_64.jpg',\n",
       " '288_64_1.jpg',\n",
       " '289_64.jpg',\n",
       " '289_64_1.jpg',\n",
       " '28_64.jpg',\n",
       " '28_64_1.jpg',\n",
       " '290_64.jpg',\n",
       " '290_64_1.jpg',\n",
       " '291_64.jpg',\n",
       " '291_64_1.jpg',\n",
       " '292_64.jpg',\n",
       " '292_64_1.jpg',\n",
       " '293_64.jpg',\n",
       " '293_64_1.jpg',\n",
       " '294_64.jpg',\n",
       " '294_64_1.jpg',\n",
       " '295_64.jpg',\n",
       " '295_64_1.jpg',\n",
       " '296_64.jpg',\n",
       " '296_64_1.jpg',\n",
       " '297_64.jpg',\n",
       " '297_64_1.jpg',\n",
       " '298_64.jpg',\n",
       " '298_64_1.jpg',\n",
       " '299_64.jpg',\n",
       " '299_64_1.jpg',\n",
       " '29_64.jpg',\n",
       " '29_64_1.jpg',\n",
       " '2_64.jpg',\n",
       " '2_64_1.jpg',\n",
       " '300_64.jpg',\n",
       " '300_64_1.jpg',\n",
       " '301_64.jpg',\n",
       " '301_64_1.jpg',\n",
       " '302_64.jpg',\n",
       " '302_64_1.jpg',\n",
       " '303_64.jpg',\n",
       " '303_64_1.jpg',\n",
       " '304_64.jpg',\n",
       " '304_64_1.jpg',\n",
       " '305_64.jpg',\n",
       " '305_64_1.jpg',\n",
       " '306_64.jpg',\n",
       " '306_64_1.jpg',\n",
       " '307_64.jpg',\n",
       " '307_64_1.jpg',\n",
       " '308_64.jpg',\n",
       " '308_64_1.jpg',\n",
       " '309_64.jpg',\n",
       " '309_64_1.jpg',\n",
       " '30_64.jpg',\n",
       " '30_64_1.jpg',\n",
       " '310_64.jpg',\n",
       " '310_64_1.jpg',\n",
       " '311_64.jpg',\n",
       " '311_64_1.jpg',\n",
       " '312_64.jpg',\n",
       " '312_64_1.jpg',\n",
       " '313_64.jpg',\n",
       " '313_64_1.jpg',\n",
       " '314_64.jpg',\n",
       " '314_64_1.jpg',\n",
       " '315_64.jpg',\n",
       " '315_64_1.jpg',\n",
       " '316_64.jpg',\n",
       " '316_64_1.jpg',\n",
       " '317_64.jpg',\n",
       " '317_64_1.jpg',\n",
       " '318_64.jpg',\n",
       " '318_64_1.jpg',\n",
       " '319_64.jpg',\n",
       " '319_64_1.jpg',\n",
       " '31_64.jpg',\n",
       " '31_64_1.jpg',\n",
       " '320_64.jpg',\n",
       " '320_64_1.jpg',\n",
       " '321_64.jpg',\n",
       " '321_64_1.jpg',\n",
       " '322_64.jpg',\n",
       " '322_64_1.jpg',\n",
       " '323_64.jpg',\n",
       " '323_64_1.jpg',\n",
       " '324_64.jpg',\n",
       " '324_64_1.jpg',\n",
       " '325_64.jpg',\n",
       " '325_64_1.jpg',\n",
       " '326_64.jpg',\n",
       " '326_64_1.jpg',\n",
       " '327_64.jpg',\n",
       " '327_64_1.jpg',\n",
       " '328_64.jpg',\n",
       " '328_64_1.jpg',\n",
       " '329_64.jpg',\n",
       " '329_64_1.jpg',\n",
       " '32_64.jpg',\n",
       " '32_64_1.jpg',\n",
       " '330_64.jpg',\n",
       " '330_64_1.jpg',\n",
       " '331_64.jpg',\n",
       " '331_64_1.jpg',\n",
       " '332_64.jpg',\n",
       " '332_64_1.jpg',\n",
       " '333_64.jpg',\n",
       " '333_64_1.jpg',\n",
       " '334_64.jpg',\n",
       " '334_64_1.jpg',\n",
       " '335_64.jpg',\n",
       " '335_64_1.jpg',\n",
       " '336_64.jpg',\n",
       " '336_64_1.jpg',\n",
       " '337_64.jpg',\n",
       " '337_64_1.jpg',\n",
       " '338_64.jpg',\n",
       " '338_64_1.jpg',\n",
       " '339_64.jpg',\n",
       " '339_64_1.jpg',\n",
       " '33_64.jpg',\n",
       " '33_64_1.jpg',\n",
       " '340_64.jpg',\n",
       " '340_64_1.jpg',\n",
       " '341_64.jpg',\n",
       " '341_64_1.jpg',\n",
       " '342_64.jpg',\n",
       " '342_64_1.jpg',\n",
       " '343_64.jpg',\n",
       " '343_64_1.jpg',\n",
       " '344_64.jpg',\n",
       " '344_64_1.jpg',\n",
       " '345_64.jpg',\n",
       " '345_64_1.jpg',\n",
       " '346_64.jpg',\n",
       " '346_64_1.jpg',\n",
       " '347_64.jpg',\n",
       " '347_64_1.jpg',\n",
       " '348_64.jpg',\n",
       " '348_64_1.jpg',\n",
       " '349_64.jpg',\n",
       " '349_64_1.jpg',\n",
       " '34_64.jpg',\n",
       " '34_64_1.jpg',\n",
       " '350_64.jpg',\n",
       " '350_64_1.jpg',\n",
       " '351_64.jpg',\n",
       " '351_64_1.jpg',\n",
       " '352_64.jpg',\n",
       " '352_64_1.jpg',\n",
       " '353_64.jpg',\n",
       " '353_64_1.jpg',\n",
       " '354_64.jpg',\n",
       " '354_64_1.jpg',\n",
       " '355_64.jpg',\n",
       " '355_64_1.jpg',\n",
       " '356_64.jpg',\n",
       " '356_64_1.jpg',\n",
       " '357_64.jpg',\n",
       " '357_64_1.jpg',\n",
       " '358_64.jpg',\n",
       " '358_64_1.jpg',\n",
       " '359_64.jpg',\n",
       " '359_64_1.jpg',\n",
       " '35_64.jpg',\n",
       " '35_64_1.jpg',\n",
       " '360_64.jpg',\n",
       " '360_64_1.jpg',\n",
       " '361_64.jpg',\n",
       " '361_64_1.jpg',\n",
       " '362_64.jpg',\n",
       " '362_64_1.jpg',\n",
       " '363_64.jpg',\n",
       " '363_64_1.jpg',\n",
       " '364_64.jpg',\n",
       " '364_64_1.jpg',\n",
       " '365_64.jpg',\n",
       " '365_64_1.jpg',\n",
       " '366_64.jpg',\n",
       " '366_64_1.jpg',\n",
       " '367_64.jpg',\n",
       " '367_64_1.jpg',\n",
       " '368_64.jpg',\n",
       " '368_64_1.jpg',\n",
       " '369_64.jpg',\n",
       " '369_64_1.jpg',\n",
       " '36_64.jpg',\n",
       " '36_64_1.jpg',\n",
       " '370_64.jpg',\n",
       " '370_64_1.jpg',\n",
       " '371_64.jpg',\n",
       " '371_64_1.jpg',\n",
       " '372_64.jpg',\n",
       " '372_64_1.jpg',\n",
       " '373_64.jpg',\n",
       " '373_64_1.jpg',\n",
       " '374_64.jpg',\n",
       " '374_64_1.jpg',\n",
       " '375_64.jpg',\n",
       " '375_64_1.jpg',\n",
       " '376_64.jpg',\n",
       " '376_64_1.jpg',\n",
       " '377_64.jpg',\n",
       " '377_64_1.jpg',\n",
       " '378_64.jpg',\n",
       " '378_64_1.jpg',\n",
       " '379_64.jpg',\n",
       " '379_64_1.jpg',\n",
       " '37_64.jpg',\n",
       " '37_64_1.jpg',\n",
       " '380_64.jpg',\n",
       " '380_64_1.jpg',\n",
       " '381_64.jpg',\n",
       " '381_64_1.jpg',\n",
       " '382_64.jpg',\n",
       " '382_64_1.jpg',\n",
       " '383_64.jpg',\n",
       " '383_64_1.jpg',\n",
       " '384_64.jpg',\n",
       " '384_64_1.jpg',\n",
       " '385_64.jpg',\n",
       " '385_64_1.jpg',\n",
       " '386_64.jpg',\n",
       " '386_64_1.jpg',\n",
       " '387_64.jpg',\n",
       " '387_64_1.jpg',\n",
       " '388_64.jpg',\n",
       " '388_64_1.jpg',\n",
       " '389_64.jpg',\n",
       " '389_64_1.jpg',\n",
       " '38_64.jpg',\n",
       " '38_64_1.jpg',\n",
       " '390_64.jpg',\n",
       " '390_64_1.jpg',\n",
       " '391_64.jpg',\n",
       " '391_64_1.jpg',\n",
       " '392_64.jpg',\n",
       " '392_64_1.jpg',\n",
       " '393_64.jpg',\n",
       " '393_64_1.jpg',\n",
       " '394_64.jpg',\n",
       " '394_64_1.jpg',\n",
       " '395_64.jpg',\n",
       " '395_64_1.jpg',\n",
       " '396_64.jpg',\n",
       " '396_64_1.jpg',\n",
       " '397_64.jpg',\n",
       " '397_64_1.jpg',\n",
       " '398_64.jpg',\n",
       " '398_64_1.jpg',\n",
       " '399_64.jpg',\n",
       " '399_64_1.jpg',\n",
       " '39_64.jpg',\n",
       " '39_64_1.jpg',\n",
       " '3_64.jpg',\n",
       " '3_64_1.jpg',\n",
       " '400_64.jpg',\n",
       " '400_64_1.jpg',\n",
       " '401_64.jpg',\n",
       " '402_64.jpg',\n",
       " '403_64.jpg',\n",
       " '404_64.jpg',\n",
       " '405_64.jpg',\n",
       " '406_64.jpg',\n",
       " '407_64.jpg',\n",
       " '408_64.jpg',\n",
       " '409_64.jpg',\n",
       " '40_64.jpg',\n",
       " '40_64_1.jpg',\n",
       " '410_64.jpg',\n",
       " '411_64.jpg',\n",
       " '412_64.jpg',\n",
       " '413_64.jpg',\n",
       " '414_64.jpg',\n",
       " '415_64.jpg',\n",
       " '416_64.jpg',\n",
       " '417_64.jpg',\n",
       " '418_64.jpg',\n",
       " '419_64.jpg',\n",
       " '41_64.jpg',\n",
       " '41_64_1.jpg',\n",
       " '420_64.jpg',\n",
       " '421_64.jpg',\n",
       " '422_64.jpg',\n",
       " '423_64.jpg',\n",
       " '424_64.jpg',\n",
       " '425_64.jpg',\n",
       " '426_64.jpg',\n",
       " '427_64.jpg',\n",
       " '428_64.jpg',\n",
       " '429_64.jpg',\n",
       " '42_64.jpg',\n",
       " '42_64_1.jpg',\n",
       " '430_64.jpg',\n",
       " '431_64.jpg',\n",
       " '432_64.jpg',\n",
       " '433_64.jpg',\n",
       " '434_64.jpg',\n",
       " '435_64.jpg',\n",
       " '436_64.jpg',\n",
       " '437_64.jpg',\n",
       " '438_64.jpg',\n",
       " '439_64.jpg',\n",
       " '43_64.jpg',\n",
       " '43_64_1.jpg',\n",
       " '440_64.jpg',\n",
       " '441_64.jpg',\n",
       " '442_64.jpg',\n",
       " '443_64.jpg',\n",
       " '444_64.jpg',\n",
       " '445_64.jpg',\n",
       " '446_64.jpg',\n",
       " '447_64.jpg',\n",
       " '448_64.jpg',\n",
       " '449_64.jpg',\n",
       " '44_64.jpg',\n",
       " '44_64_1.jpg',\n",
       " '450_64.jpg',\n",
       " '451_64.jpg',\n",
       " '452_64.jpg',\n",
       " '453_64.jpg',\n",
       " '454_64.jpg',\n",
       " '455_64.jpg',\n",
       " '456_64.jpg',\n",
       " '457_64.jpg',\n",
       " '458_64.jpg',\n",
       " '459_64.jpg',\n",
       " '45_64.jpg',\n",
       " '45_64_1.jpg',\n",
       " '460_64.jpg',\n",
       " '461_64.jpg',\n",
       " '462_64.jpg',\n",
       " '463_64.jpg',\n",
       " '464_64.jpg',\n",
       " '465_64.jpg',\n",
       " '466_64.jpg',\n",
       " '467_64.jpg',\n",
       " '468_64.jpg',\n",
       " '469_64.jpg',\n",
       " '46_64.jpg',\n",
       " '46_64_1.jpg',\n",
       " '470_64.jpg',\n",
       " '471_64.jpg',\n",
       " '472_64.jpg',\n",
       " '473_64.jpg',\n",
       " '474_64.jpg',\n",
       " '475_64.jpg',\n",
       " '476_64.jpg',\n",
       " '477_64.jpg',\n",
       " '478_64.jpg',\n",
       " '479_64.jpg',\n",
       " '47_64.jpg',\n",
       " '47_64_1.jpg',\n",
       " '480_64.jpg',\n",
       " '481_64.jpg',\n",
       " '482_64.jpg',\n",
       " '483_64.jpg',\n",
       " '484_64.jpg',\n",
       " '485_64.jpg',\n",
       " '486_64.jpg',\n",
       " '487_64.jpg',\n",
       " '488_64.jpg',\n",
       " '489_64.jpg',\n",
       " '48_64.jpg',\n",
       " '48_64_1.jpg',\n",
       " '490_64.jpg',\n",
       " '491_64.jpg',\n",
       " '492_64.jpg',\n",
       " '493_64.jpg',\n",
       " '494_64.jpg',\n",
       " '495_64.jpg',\n",
       " '496_64.jpg',\n",
       " '497_64.jpg',\n",
       " '498_64.jpg',\n",
       " '499_64.jpg',\n",
       " '49_64.jpg',\n",
       " '49_64_1.jpg',\n",
       " '4_64.jpg',\n",
       " '4_64_1.jpg',\n",
       " '500_64.jpg',\n",
       " '501_64.jpg',\n",
       " '502_64.jpg',\n",
       " '503_64.jpg',\n",
       " '504_64.jpg',\n",
       " '505_64.jpg',\n",
       " '506_64.jpg',\n",
       " '507_64.jpg',\n",
       " '508_64.jpg',\n",
       " '509_64.jpg',\n",
       " ...]"
      ]
     },
     "execution_count": 144,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "data.file_list()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 145,
   "id": "580e3031",
   "metadata": {},
   "outputs": [
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "C:\\Users\\VAIO\\AppData\\Local\\Temp\\ipykernel_13216\\2183177044.py:20: DeprecationWarning: Starting with ImageIO v3 the behavior of this function will switch to that of iio.v3.imread. To keep the current behavior (and make this warning dissapear) use `import imageio.v2 as imageio` or call `imageio.v2.imread` directly.\n",
      "  return iio.imread(self.path_to_dataset + image_name)\n"
     ]
    },
    {
     "data": {
      "text/plain": [
       "(64, 64, 3)"
      ]
     },
     "execution_count": 145,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "data.read_image(data.file_list()[0]).shape"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 146,
   "id": "9700ba79",
   "metadata": {},
   "outputs": [
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "C:\\Users\\VAIO\\AppData\\Local\\Temp\\ipykernel_13216\\2183177044.py:20: DeprecationWarning: Starting with ImageIO v3 the behavior of this function will switch to that of iio.v3.imread. To keep the current behavior (and make this warning dissapear) use `import imageio.v2 as imageio` or call `imageio.v2.imread` directly.\n",
      "  return iio.imread(self.path_to_dataset + image_name)\n"
     ]
    },
    {
     "data": {
      "text/plain": [
       "array([[[[113,  98,   0],\n",
       "         [128, 112,  16],\n",
       "         [126, 109,  19],\n",
       "         ...,\n",
       "         [180, 129,   2],\n",
       "         [179, 127,   5],\n",
       "         [174, 124,   1]],\n",
       "\n",
       "        [[115,  99,   3],\n",
       "         [117, 102,   9],\n",
       "         [105,  89,   4],\n",
       "         ...,\n",
       "         [184, 130,   4],\n",
       "         [179, 127,   5],\n",
       "         [171, 121,   0]],\n",
       "\n",
       "        [[115,  99,  11],\n",
       "         [105,  90,   5],\n",
       "         [ 84,  71,   0],\n",
       "         ...,\n",
       "         [185, 131,   5],\n",
       "         [175, 123,   5],\n",
       "         [166, 113,   0]],\n",
       "\n",
       "        ...,\n",
       "\n",
       "        [[ 78, 109, 103],\n",
       "         [121, 150, 154],\n",
       "         [140, 166, 191],\n",
       "         ...,\n",
       "         [175, 131,  10],\n",
       "         [159, 117,   0],\n",
       "         [171, 129,  11]],\n",
       "\n",
       "        [[ 96, 126, 118],\n",
       "         [124, 154, 154],\n",
       "         [136, 163, 180],\n",
       "         ...,\n",
       "         [180, 134,  12],\n",
       "         [161, 117,   0],\n",
       "         [169, 128,   4]],\n",
       "\n",
       "        [[ 87, 117, 107],\n",
       "         [100, 129, 125],\n",
       "         [103, 131, 143],\n",
       "         ...,\n",
       "         [179, 134,   9],\n",
       "         [179, 135,  10],\n",
       "         [192, 151,  23]]],\n",
       "\n",
       "\n",
       "       [[[  2,   2,   2],\n",
       "         [  1,   1,   1],\n",
       "         [  2,   2,   4],\n",
       "         ...,\n",
       "         [ 41,  22,  15],\n",
       "         [182, 140, 128],\n",
       "         [187, 136, 119]],\n",
       "\n",
       "        [[  3,   3,   3],\n",
       "         [  2,   2,   2],\n",
       "         [  2,   2,   4],\n",
       "         ...,\n",
       "         [ 83,  60,  54],\n",
       "         [162, 118, 107],\n",
       "         [183, 130, 116]],\n",
       "\n",
       "        [[  2,   4,   3],\n",
       "         [  2,   4,   3],\n",
       "         [  2,   2,   2],\n",
       "         ...,\n",
       "         [128,  99,  95],\n",
       "         [176, 127, 120],\n",
       "         [184, 127, 118]],\n",
       "\n",
       "        ...,\n",
       "\n",
       "        [[  2,   2,   2],\n",
       "         [  6,   0,   0],\n",
       "         [ 19,   0,   0],\n",
       "         ...,\n",
       "         [125,  39,  48],\n",
       "         [ 97,  14,  22],\n",
       "         [101,  20,  27]],\n",
       "\n",
       "        [[  0,   2,   1],\n",
       "         [  6,   0,   0],\n",
       "         [ 22,   0,   2],\n",
       "         ...,\n",
       "         [103,  24,  30],\n",
       "         [ 91,   8,  18],\n",
       "         [130,  47,  57]],\n",
       "\n",
       "        [[  1,   3,   2],\n",
       "         [  7,   1,   1],\n",
       "         [ 75,  51,  51],\n",
       "         ...,\n",
       "         [ 77,   2,   7],\n",
       "         [ 80,   0,   9],\n",
       "         [136,  50,  63]]],\n",
       "\n",
       "\n",
       "       [[[ 45,  83,   6],\n",
       "         [122, 160,  87],\n",
       "         [156, 190, 130],\n",
       "         ...,\n",
       "         [120, 114, 210],\n",
       "         [120, 116, 211],\n",
       "         [128, 124, 219]],\n",
       "\n",
       "        [[ 45,  78,  23],\n",
       "         [ 54,  83,  35],\n",
       "         [140, 165, 133],\n",
       "         ...,\n",
       "         [112, 107, 201],\n",
       "         [107, 102, 196],\n",
       "         [118, 114, 208]],\n",
       "\n",
       "        [[ 50,  70,  59],\n",
       "         [ 59,  75,  75],\n",
       "         [ 93, 104, 124],\n",
       "         ...,\n",
       "         [ 96,  90, 178],\n",
       "         [ 94,  88, 178],\n",
       "         [ 97,  92, 184]],\n",
       "\n",
       "        ...,\n",
       "\n",
       "        [[ 29,  49,  12],\n",
       "         [ 45,  65,  30],\n",
       "         [ 59,  75,  49],\n",
       "         ...,\n",
       "         [ 73,  64, 111],\n",
       "         [ 61,  44, 115],\n",
       "         [ 63,  41, 124]],\n",
       "\n",
       "        [[ 64,  98,  38],\n",
       "         [ 74, 104,  52],\n",
       "         [ 78, 105,  62],\n",
       "         ...,\n",
       "         [ 66,  63,  90],\n",
       "         [ 46,  32,  93],\n",
       "         [ 84,  63, 140]],\n",
       "\n",
       "        [[ 70, 110,  40],\n",
       "         [ 70, 106,  44],\n",
       "         [ 73, 105,  55],\n",
       "         ...,\n",
       "         [ 37,  39,  54],\n",
       "         [ 74,  61, 115],\n",
       "         [ 77,  57, 129]]],\n",
       "\n",
       "\n",
       "       ...,\n",
       "\n",
       "\n",
       "       [[[182, 187, 164],\n",
       "         [184, 189, 166],\n",
       "         [185, 190, 168],\n",
       "         ...,\n",
       "         [ 47,  64,  48],\n",
       "         [ 50,  71,  56],\n",
       "         [ 45,  67,  54]],\n",
       "\n",
       "        [[176, 180, 143],\n",
       "         [178, 182, 147],\n",
       "         [181, 185, 152],\n",
       "         ...,\n",
       "         [ 36,  53,  37],\n",
       "         [ 46,  64,  50],\n",
       "         [ 48,  69,  54]],\n",
       "\n",
       "        [[159, 159,  97],\n",
       "         [156, 156,  96],\n",
       "         [162, 163, 106],\n",
       "         ...,\n",
       "         [ 28,  42,  25],\n",
       "         [ 36,  53,  37],\n",
       "         [ 47,  65,  49]],\n",
       "\n",
       "        ...,\n",
       "\n",
       "        [[ 47,  71,  39],\n",
       "         [ 37,  61,  29],\n",
       "         [ 34,  54,  27],\n",
       "         ...,\n",
       "         [ 34,  52,  30],\n",
       "         [ 51,  73,  50],\n",
       "         [ 42,  65,  39]],\n",
       "\n",
       "        [[ 45,  61,  34],\n",
       "         [ 32,  51,  23],\n",
       "         [ 31,  51,  24],\n",
       "         ...,\n",
       "         [ 29,  47,  25],\n",
       "         [ 51,  74,  48],\n",
       "         [ 46,  71,  42]],\n",
       "\n",
       "        [[ 36,  50,  24],\n",
       "         [ 27,  43,  16],\n",
       "         [ 28,  46,  22],\n",
       "         ...,\n",
       "         [ 22,  37,  16],\n",
       "         [ 49,  72,  46],\n",
       "         [ 51,  76,  47]]],\n",
       "\n",
       "\n",
       "       [[[ 82,  56, 155],\n",
       "         [ 74,  46, 148],\n",
       "         [ 64,  33, 137],\n",
       "         ...,\n",
       "         [189, 153, 241],\n",
       "         [184, 150, 237],\n",
       "         [191, 157, 243]],\n",
       "\n",
       "        [[ 84,  58, 157],\n",
       "         [ 76,  48, 150],\n",
       "         [ 65,  32, 139],\n",
       "         ...,\n",
       "         [181, 145, 235],\n",
       "         [189, 154, 244],\n",
       "         [189, 155, 242]],\n",
       "\n",
       "        [[ 83,  55, 157],\n",
       "         [ 73,  45, 148],\n",
       "         [ 63,  29, 139],\n",
       "         ...,\n",
       "         [181, 146, 240],\n",
       "         [183, 146, 241],\n",
       "         [181, 144, 237]],\n",
       "\n",
       "        ...,\n",
       "\n",
       "        [[102,  86, 133],\n",
       "         [102,  86, 131],\n",
       "         [112,  99, 143],\n",
       "         ...,\n",
       "         [ 68,  44,  40],\n",
       "         [ 73,  51,  40],\n",
       "         [ 88,  69,  54]],\n",
       "\n",
       "        [[106,  90, 135],\n",
       "         [ 95,  79, 124],\n",
       "         [107,  94, 140],\n",
       "         ...,\n",
       "         [ 59,  40,  33],\n",
       "         [ 78,  61,  41],\n",
       "         [158, 142, 116]],\n",
       "\n",
       "        [[ 96,  80, 125],\n",
       "         [ 83,  67, 112],\n",
       "         [104,  90, 139],\n",
       "         ...,\n",
       "         [ 56,  38,  28],\n",
       "         [157, 143, 117],\n",
       "         [187, 174, 142]]],\n",
       "\n",
       "\n",
       "       [[[151,  87,  75],\n",
       "         [151,  87,  75],\n",
       "         [151,  87,  75],\n",
       "         ...,\n",
       "         [151,  87,  75],\n",
       "         [151,  87,  75],\n",
       "         [151,  87,  75]],\n",
       "\n",
       "        [[151,  87,  75],\n",
       "         [151,  87,  75],\n",
       "         [151,  87,  75],\n",
       "         ...,\n",
       "         [151,  87,  75],\n",
       "         [151,  87,  75],\n",
       "         [151,  87,  75]],\n",
       "\n",
       "        [[151,  87,  75],\n",
       "         [151,  87,  75],\n",
       "         [151,  87,  75],\n",
       "         ...,\n",
       "         [151,  87,  75],\n",
       "         [151,  87,  75],\n",
       "         [151,  87,  75]],\n",
       "\n",
       "        ...,\n",
       "\n",
       "        [[151,  87,  75],\n",
       "         [151,  87,  75],\n",
       "         [151,  87,  75],\n",
       "         ...,\n",
       "         [151,  87,  75],\n",
       "         [151,  87,  75],\n",
       "         [151,  87,  75]],\n",
       "\n",
       "        [[151,  87,  75],\n",
       "         [151,  87,  75],\n",
       "         [151,  87,  75],\n",
       "         ...,\n",
       "         [151,  87,  75],\n",
       "         [151,  87,  75],\n",
       "         [151,  87,  75]],\n",
       "\n",
       "        [[151,  87,  75],\n",
       "         [151,  87,  75],\n",
       "         [151,  87,  75],\n",
       "         ...,\n",
       "         [151,  87,  75],\n",
       "         [151,  87,  75],\n",
       "         [151,  87,  75]]]], dtype=uint8)"
      ]
     },
     "execution_count": 146,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "data.load_data()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 147,
   "id": "9f2cc322",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "array([[[[113,  98,   0],\n",
       "         [128, 112,  16],\n",
       "         [126, 109,  19],\n",
       "         ...,\n",
       "         [180, 129,   2],\n",
       "         [179, 127,   5],\n",
       "         [174, 124,   1]],\n",
       "\n",
       "        [[115,  99,   3],\n",
       "         [117, 102,   9],\n",
       "         [105,  89,   4],\n",
       "         ...,\n",
       "         [184, 130,   4],\n",
       "         [179, 127,   5],\n",
       "         [171, 121,   0]],\n",
       "\n",
       "        [[115,  99,  11],\n",
       "         [105,  90,   5],\n",
       "         [ 84,  71,   0],\n",
       "         ...,\n",
       "         [185, 131,   5],\n",
       "         [175, 123,   5],\n",
       "         [166, 113,   0]],\n",
       "\n",
       "        ...,\n",
       "\n",
       "        [[ 78, 109, 103],\n",
       "         [121, 150, 154],\n",
       "         [140, 166, 191],\n",
       "         ...,\n",
       "         [175, 131,  10],\n",
       "         [159, 117,   0],\n",
       "         [171, 129,  11]],\n",
       "\n",
       "        [[ 96, 126, 118],\n",
       "         [124, 154, 154],\n",
       "         [136, 163, 180],\n",
       "         ...,\n",
       "         [180, 134,  12],\n",
       "         [161, 117,   0],\n",
       "         [169, 128,   4]],\n",
       "\n",
       "        [[ 87, 117, 107],\n",
       "         [100, 129, 125],\n",
       "         [103, 131, 143],\n",
       "         ...,\n",
       "         [179, 134,   9],\n",
       "         [179, 135,  10],\n",
       "         [192, 151,  23]]],\n",
       "\n",
       "\n",
       "       [[[  2,   2,   2],\n",
       "         [  1,   1,   1],\n",
       "         [  2,   2,   4],\n",
       "         ...,\n",
       "         [ 41,  22,  15],\n",
       "         [182, 140, 128],\n",
       "         [187, 136, 119]],\n",
       "\n",
       "        [[  3,   3,   3],\n",
       "         [  2,   2,   2],\n",
       "         [  2,   2,   4],\n",
       "         ...,\n",
       "         [ 83,  60,  54],\n",
       "         [162, 118, 107],\n",
       "         [183, 130, 116]],\n",
       "\n",
       "        [[  2,   4,   3],\n",
       "         [  2,   4,   3],\n",
       "         [  2,   2,   2],\n",
       "         ...,\n",
       "         [128,  99,  95],\n",
       "         [176, 127, 120],\n",
       "         [184, 127, 118]],\n",
       "\n",
       "        ...,\n",
       "\n",
       "        [[  2,   2,   2],\n",
       "         [  6,   0,   0],\n",
       "         [ 19,   0,   0],\n",
       "         ...,\n",
       "         [125,  39,  48],\n",
       "         [ 97,  14,  22],\n",
       "         [101,  20,  27]],\n",
       "\n",
       "        [[  0,   2,   1],\n",
       "         [  6,   0,   0],\n",
       "         [ 22,   0,   2],\n",
       "         ...,\n",
       "         [103,  24,  30],\n",
       "         [ 91,   8,  18],\n",
       "         [130,  47,  57]],\n",
       "\n",
       "        [[  1,   3,   2],\n",
       "         [  7,   1,   1],\n",
       "         [ 75,  51,  51],\n",
       "         ...,\n",
       "         [ 77,   2,   7],\n",
       "         [ 80,   0,   9],\n",
       "         [136,  50,  63]]],\n",
       "\n",
       "\n",
       "       [[[ 45,  83,   6],\n",
       "         [122, 160,  87],\n",
       "         [156, 190, 130],\n",
       "         ...,\n",
       "         [120, 114, 210],\n",
       "         [120, 116, 211],\n",
       "         [128, 124, 219]],\n",
       "\n",
       "        [[ 45,  78,  23],\n",
       "         [ 54,  83,  35],\n",
       "         [140, 165, 133],\n",
       "         ...,\n",
       "         [112, 107, 201],\n",
       "         [107, 102, 196],\n",
       "         [118, 114, 208]],\n",
       "\n",
       "        [[ 50,  70,  59],\n",
       "         [ 59,  75,  75],\n",
       "         [ 93, 104, 124],\n",
       "         ...,\n",
       "         [ 96,  90, 178],\n",
       "         [ 94,  88, 178],\n",
       "         [ 97,  92, 184]],\n",
       "\n",
       "        ...,\n",
       "\n",
       "        [[ 29,  49,  12],\n",
       "         [ 45,  65,  30],\n",
       "         [ 59,  75,  49],\n",
       "         ...,\n",
       "         [ 73,  64, 111],\n",
       "         [ 61,  44, 115],\n",
       "         [ 63,  41, 124]],\n",
       "\n",
       "        [[ 64,  98,  38],\n",
       "         [ 74, 104,  52],\n",
       "         [ 78, 105,  62],\n",
       "         ...,\n",
       "         [ 66,  63,  90],\n",
       "         [ 46,  32,  93],\n",
       "         [ 84,  63, 140]],\n",
       "\n",
       "        [[ 70, 110,  40],\n",
       "         [ 70, 106,  44],\n",
       "         [ 73, 105,  55],\n",
       "         ...,\n",
       "         [ 37,  39,  54],\n",
       "         [ 74,  61, 115],\n",
       "         [ 77,  57, 129]]],\n",
       "\n",
       "\n",
       "       ...,\n",
       "\n",
       "\n",
       "       [[[182, 187, 164],\n",
       "         [184, 189, 166],\n",
       "         [185, 190, 168],\n",
       "         ...,\n",
       "         [ 47,  64,  48],\n",
       "         [ 50,  71,  56],\n",
       "         [ 45,  67,  54]],\n",
       "\n",
       "        [[176, 180, 143],\n",
       "         [178, 182, 147],\n",
       "         [181, 185, 152],\n",
       "         ...,\n",
       "         [ 36,  53,  37],\n",
       "         [ 46,  64,  50],\n",
       "         [ 48,  69,  54]],\n",
       "\n",
       "        [[159, 159,  97],\n",
       "         [156, 156,  96],\n",
       "         [162, 163, 106],\n",
       "         ...,\n",
       "         [ 28,  42,  25],\n",
       "         [ 36,  53,  37],\n",
       "         [ 47,  65,  49]],\n",
       "\n",
       "        ...,\n",
       "\n",
       "        [[ 47,  71,  39],\n",
       "         [ 37,  61,  29],\n",
       "         [ 34,  54,  27],\n",
       "         ...,\n",
       "         [ 34,  52,  30],\n",
       "         [ 51,  73,  50],\n",
       "         [ 42,  65,  39]],\n",
       "\n",
       "        [[ 45,  61,  34],\n",
       "         [ 32,  51,  23],\n",
       "         [ 31,  51,  24],\n",
       "         ...,\n",
       "         [ 29,  47,  25],\n",
       "         [ 51,  74,  48],\n",
       "         [ 46,  71,  42]],\n",
       "\n",
       "        [[ 36,  50,  24],\n",
       "         [ 27,  43,  16],\n",
       "         [ 28,  46,  22],\n",
       "         ...,\n",
       "         [ 22,  37,  16],\n",
       "         [ 49,  72,  46],\n",
       "         [ 51,  76,  47]]],\n",
       "\n",
       "\n",
       "       [[[ 82,  56, 155],\n",
       "         [ 74,  46, 148],\n",
       "         [ 64,  33, 137],\n",
       "         ...,\n",
       "         [189, 153, 241],\n",
       "         [184, 150, 237],\n",
       "         [191, 157, 243]],\n",
       "\n",
       "        [[ 84,  58, 157],\n",
       "         [ 76,  48, 150],\n",
       "         [ 65,  32, 139],\n",
       "         ...,\n",
       "         [181, 145, 235],\n",
       "         [189, 154, 244],\n",
       "         [189, 155, 242]],\n",
       "\n",
       "        [[ 83,  55, 157],\n",
       "         [ 73,  45, 148],\n",
       "         [ 63,  29, 139],\n",
       "         ...,\n",
       "         [181, 146, 240],\n",
       "         [183, 146, 241],\n",
       "         [181, 144, 237]],\n",
       "\n",
       "        ...,\n",
       "\n",
       "        [[102,  86, 133],\n",
       "         [102,  86, 131],\n",
       "         [112,  99, 143],\n",
       "         ...,\n",
       "         [ 68,  44,  40],\n",
       "         [ 73,  51,  40],\n",
       "         [ 88,  69,  54]],\n",
       "\n",
       "        [[106,  90, 135],\n",
       "         [ 95,  79, 124],\n",
       "         [107,  94, 140],\n",
       "         ...,\n",
       "         [ 59,  40,  33],\n",
       "         [ 78,  61,  41],\n",
       "         [158, 142, 116]],\n",
       "\n",
       "        [[ 96,  80, 125],\n",
       "         [ 83,  67, 112],\n",
       "         [104,  90, 139],\n",
       "         ...,\n",
       "         [ 56,  38,  28],\n",
       "         [157, 143, 117],\n",
       "         [187, 174, 142]]],\n",
       "\n",
       "\n",
       "       [[[151,  87,  75],\n",
       "         [151,  87,  75],\n",
       "         [151,  87,  75],\n",
       "         ...,\n",
       "         [151,  87,  75],\n",
       "         [151,  87,  75],\n",
       "         [151,  87,  75]],\n",
       "\n",
       "        [[151,  87,  75],\n",
       "         [151,  87,  75],\n",
       "         [151,  87,  75],\n",
       "         ...,\n",
       "         [151,  87,  75],\n",
       "         [151,  87,  75],\n",
       "         [151,  87,  75]],\n",
       "\n",
       "        [[151,  87,  75],\n",
       "         [151,  87,  75],\n",
       "         [151,  87,  75],\n",
       "         ...,\n",
       "         [151,  87,  75],\n",
       "         [151,  87,  75],\n",
       "         [151,  87,  75]],\n",
       "\n",
       "        ...,\n",
       "\n",
       "        [[151,  87,  75],\n",
       "         [151,  87,  75],\n",
       "         [151,  87,  75],\n",
       "         ...,\n",
       "         [151,  87,  75],\n",
       "         [151,  87,  75],\n",
       "         [151,  87,  75]],\n",
       "\n",
       "        [[151,  87,  75],\n",
       "         [151,  87,  75],\n",
       "         [151,  87,  75],\n",
       "         ...,\n",
       "         [151,  87,  75],\n",
       "         [151,  87,  75],\n",
       "         [151,  87,  75]],\n",
       "\n",
       "        [[151,  87,  75],\n",
       "         [151,  87,  75],\n",
       "         [151,  87,  75],\n",
       "         ...,\n",
       "         [151,  87,  75],\n",
       "         [151,  87,  75],\n",
       "         [151,  87,  75]]]], dtype=uint8)"
      ]
     },
     "execution_count": 147,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "data.dataset"
   ]
  }
 ],
 "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.9.5"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 5
}