import os
import torch
import pandas as pd
import numpy as np
from torch.utils.data import Dataset, random_split, DataLoader
from PIL import Image
import torchvision.models as models
import matplotlib.pyplot as plt
from tqdm.notebook import tqdm
import torchvision.transforms as T
from sklearn.metrics import f1_score
import torch.nn.functional as F
import torch.nn as nn
from torchvision.utils import make_grid
from torchvision.datasets import ImageFolder
%matplotlib inline

DATA_DIR = './data'

TRAIN_DIR = DATA_DIR + '/seg_train'                           
TEST_DIR = DATA_DIR + '/seg_test'                             
PRED_DIR= DATA_DIR + '/seg_pred'                             

#data_df = pd.read_csv(TRAIN_CSV)
#data_df.head()