import torch 
import os
from tqdm import tqdm
import pandas as pd
import matplotlib.pyplot as plt
from pathlib import Path
import cv2
import numpy as np
import torch.nn as nn
import torch.nn.functional as F
%matplotlib inline

project_name = '04-assignment-mask_detector'

classes = ['Non-Masked', 'Masked']

#Dataset: https://www.kaggle.com/omkar1008/covid19-mask-detection
from google.colab import drive

USE_GDRIVE=False

root_path = 'drive/MyDrive/input/data'
if USE_GDRIVE:
  drive.mount('/content/gdrive')
  root_path = 'gdrive/My Drive/input/data/'  #change dir to your project folder

REBUILD_DATA = True
if REBUILD_DATA: #if we are running it for the first time
    data_path = Path(root_path)
    maskPath = data_path/'mask_with_coodinate'
    #maskPath2= data_path/'dataset2/webface_masked'
    nonMaskPath = data_path/'normal_with_coodinate'
    maskDF = pd.DataFrame() 
    path_dirs = [ [maskPath,1],[nonMaskPath,0] ] #path and label
    if not os.path.exists(data_path):
        raise Exception("The data path doesn't exist")