[Deep] 기초

CV 압축 해제 

 

!unzip -qq "/content/drive/MyDrive/data.zip"

 

데이터 경로 설정

train_dir = '/content/TRAIN'
test_dir = '/content/TEST'

 

데이터 로드

 def load_data(directory):
    images_path = []
    labels = []
    
    for label in os.listdir(directory):
            
        for file_name in os.listdir(directory + '/' + label):
            
            images_path.append(directory + '/' + label + '/' + file_name)
            labels.append(label)
                
    return images_path, labels

train_images, train_labels = load_data(train_dir)
test_images = glob.glob(test_dir+'/*')

Train셋

train = pd.DataFrame({'path':train_images, 'label':train_labels})
train

Test셋

test = pd.DataFrame({'path':test_images, 'label':np.NaN})
test