2023. 8. 15. 11:51ㆍ딥러닝 모델: 파이토치
ImageDataLoaders와 ImageDataBunch는 둘 다 fastai 라이브러리에서 이미지 데이터를 처리하는 데 사용하는 클래스임
그러나 두 클래스는 라이브러리의 다른 버전에서 각각 다른 역할과 사용법을 갖고 있음.
♣ ImageDataBunch
fastai 버전 1.x에서 훈련 및 검증용 이미지 데이터를 생성하고 관리하기 위한 클래스
훈련, 검증 및 테스트 데이터 로더를 하나의 객체로 묶는 방법 제공
예시
ds_tfms: batching이나 model training이 일어나기 전에 데이터셋 전체에 적용되는 변환. 주로 데이터 전처리 및 증강을 위해 사용됨. ds_tfms는 ImageDataBunch에서는 사용되지만 ImageDataLoaders에서는 잘 사용하지 않음.
♣ ImageDataLoaders
fastai 버전 2.x의 출범과 함께 ImageDataBunch는 ImageDataLoaders로 대체됨. DataLoaders는 더 유연하고 조합 가능한 방식으로 데이터로더를 구성함.
예시
item_tfms는 개별 이미지에 변형을 적용하는 것임. 개별 이미지 변형은 CPU에서 일어남. 개별 이미지 변형이 끝난 후에 이 이미지들을 batch로 묶어서 GPU에 넣을 수 있음.
참고 자료
https://dining-developer.tistory.com/26
Fastai - X-ray 폐렴 검출 CNN 10분 컷
FastAi 딥러닝 라이브러리를 이용해 이미지 분류해보기 Fastai는 내가 tensorflow 이후 두 번째로 접했었던 딥러닝 라이브러리이다. 직장 동료로부터 이걸 소개받고는 적잖이 충격을 받았었는데, 너~
dining-developer.tistory.com
'딥러닝 모델: 파이토치' 카테고리의 다른 글
| 파이토치 튜토리얼 (1) | 2023.10.07 |
|---|---|
| 밑바닥부터 Learner 클래스 만들기 (0) | 2023.09.20 |
| DataBlock와 DataLoader 비교 (0) | 2023.09.17 |
| 텐서(Tensor) (0) | 2023.08.07 |
| 옵티마이저(최적화)의 의미와 경사하강법(Gradient Descent) (0) | 2023.07.16 |