{ "cells": [ { "cell_type": "code", "execution_count": 13, "id": "602ab916", "metadata": {}, "outputs": [], "source": [ "import pandas as pd\n", "import numpy as np\n", "import glob\n", "from os import walk" ] }, { "cell_type": "code", "execution_count": 2, "id": "a7492299", "metadata": {}, "outputs": [], "source": [ "TRAIN_DATA_DIR = \"./data/cropped_images\"\n", "TRAIN_LABELS_PATH = \"./data/labels.txt\"" ] }, { "cell_type": "code", "execution_count": 29, "id": "5239a585", "metadata": {}, "outputs": [ { "data": { "text/html": [ "
\n", " | label | \n", "
---|---|
0 | \n", "without_mask | \n", "
1 | \n", "with_mask | \n", "
2 | \n", "without_mask | \n", "
3 | \n", "with_mask | \n", "
4 | \n", "with_mask | \n", "
... | \n", "... | \n", "
4067 | \n", "with_mask | \n", "
4068 | \n", "with_mask | \n", "
4069 | \n", "mask_weared_incorrect | \n", "
4070 | \n", "with_mask | \n", "
4071 | \n", "with_mask | \n", "
4072 rows × 1 columns
\n", "\n", " | file_name | \n", "
---|---|
0 | \n", "0.png | \n", "
1 | \n", "1.png | \n", "
2 | \n", "2.png | \n", "
3 | \n", "3.png | \n", "
4 | \n", "4.png | \n", "
... | \n", "... | \n", "
4067 | \n", "4067.png | \n", "
4068 | \n", "4068.png | \n", "
4069 | \n", "4069.png | \n", "
4070 | \n", "4070.png | \n", "
4071 | \n", "4071.png | \n", "
4072 rows × 1 columns
\n", "\n", " | file_name | \n", "label | \n", "
---|---|---|
1660 | \n", "1660.png | \n", "with_mask | \n", "
2976 | \n", "2976.png | \n", "with_mask | \n", "
920 | \n", "920.png | \n", "with_mask | \n", "
3984 | \n", "3984.png | \n", "with_mask | \n", "
3867 | \n", "3867.png | \n", "with_mask | \n", "
... | \n", "... | \n", "... | \n", "
530 | \n", "530.png | \n", "with_mask | \n", "
1344 | \n", "1344.png | \n", "with_mask | \n", "
1756 | \n", "1756.png | \n", "with_mask | \n", "
1196 | \n", "1196.png | \n", "without_mask | \n", "
364 | \n", "364.png | \n", "with_mask | \n", "
815 rows × 2 columns
\n", "