[딥러닝] 1.데이터셋준비 - kaggle 파일 colab에서 다운, 압축해제

1. 코랩 실행

 

2. 캐글 로그인 후,   설정>API>Create New Token 클릭> 다운로드 된 json파일 colab에 업로드

 

3. 캐글 data탭에서 다운로드 코드 복사

- 예시 : 개, 고양이 데이터 셋 https://www.kaggle.com/competitions/dogs-vs-cats-redux-kernels-edition/data

 

4. 코랩에 해당 코드 입력 및 실행하여 다운로드 받기

import os
os.environ['KAGGLE_CONFIG_DIR'] = '/content/'
!kaggle competitions download -c dogs-vs-cats-redux-kernels-edition

 

5. 파일 압축 해제

!unzip -q dogs-vs-cats-redux-kernels-edition.zip -d .

 

6. 폴더 안에 갯수 확인

import os
print( len(os.listdir('/content/train/')))

 

7. 파일 폴더별로 나누기

import os
import shutil

# 저장할 폴더 생성
os.mkdir('/content/dataset')
os.mkdir('/content/dataset/cat')
os.mkdir('/content/dataset/dog')


# 파일 복사해오기
for i in os.listdir('/content/train/'):
  if 'cat' in i:
    shutil.copyfile('/content/train/'+ i,'/content/dataset/cat/'+i )
  if 'dog' in i:
    shutil.copyfile('/content/train/'+ i,'/content/dataset/dog/'+i )