본문 바로가기
DL

Pytorch Dataset - cv2.imread 메모리 사용

by YGSEO 2021. 1. 31.
728x90

imread 라는 것이 메모리를 어느 정도 차지함.(어느정도?)

 

메모리 충분 ==> init 메서드에 imgs 리스트 생성해서 mread로 미리 읽어놔서 빠른속도로 로딩(대신 메모리 사용량 증가) 후 getitem 메서드에서 나머지 augmentation 진행

 

메모리 부족 ==> getitem에서 이미지를 하나씩 imread로 읽어서 augmentation 적용

 

 

728x90

댓글