pytorch 데이터셋 클래스

https://github.com/dohun-mat/dna_study_semina/blob/main/%EA%B3%BC%EC%A0%9C/week4_%EA%B3%BC%EC%A0%9C_2.ipynb

GitHub - dohun-mat/dna_study_semina

코드는 깃허브에 자세하게있다.

 

파이토치 데이터셋 클래스는 init, getitem, len의 특수 메소드3개를 요구하고 이 메소드는 전부 오버라이딩되어있다.

root_path, mode(train or validation or test), transform(전처리&Augmentation) 3개의 입력을 받아 data와 label을 출력하는 클래스를 만들도록하자.

 

pytorch dataset 클래스

 

이어서 전처리와 augmentation을 적용해보겠다.

https://dohun-0714.tistory.com/28

pytorch 데이터셋 augmentation