import os
import numpy as np
import pandas as pd

!pip install fastai2 -q

from fastai2.vision import *
from fastai2.vision.all import *
from fastai2.vision.widgets import *

DATA_DIR = '../input/jovian-pytorch-z2g/Human protein atlas'

TRAIN_DIR = DATA_DIR + '/train'                           
TEST_DIR = DATA_DIR + '/test'                             

TRAIN_CSV = DATA_DIR + '/train.csv'                       
TEST_CSV = '../input/jovian-pytorch-z2g/submission.csv' 

train_df = pd.read_csv(TRAIN_CSV)
train_df['imgPath'] = train_df.apply(lambda x : os.path.join(TRAIN_DIR,str(x['Image'])+'.png'),axis=1)
train_df.head()