부캠 TIL 0901 - Albumentation라이브러리

기린이·2021년 9월 1일

부캠 TIL 🦆

목록 보기
22/53
post-thumbnail

Albumentation 주의점

        if self.transform:
            image = np.array(image)
            image = self.transform(image=image)
            image = image['image']

albumentation에 들어가는 이미지는 np.array형태이다.
transform에 넣어줄때 image인자에 지정해서 넣어야한다.
transform된 결과가 dict로 나오므로 여기서 image를 인덱싱해주어야한다.

왜 이렇게 만들어놨지...? 웨.....?

profile
중요한 것은 속력이 아니라 방향성, 공부하며 메모를 남기는 공간입니다.

0개의 댓글