albumentations 사용시 하나의 이미지에 이미지가9개(?)로 나오는 현상은 왜그런가요?(사진첨부)

포디블록 구조 추출 AI 경진대회

다냐니라

2023.01.26 01:32 1,842 조회

사진이 어떻게 변하는지 확인하고 있는데 질문이 있습니다.

이와같이 한장의 사진이 있는데, albumentations 의 ToTensorV2(), 즉 tensor로 바꾸면..

이런식으로 한장의 사진에 9장이 들어간듯하게 나오는데.. 아무리 이유를 찾아도 찾을 수 없네요.. 지식이 필요합니다!

댓글 4개

로그인이 필요합니다

comment

0 / 1000

Redix6

2023.01.26 12:26

ToTensorV2는 단순히 HWC 를 CHW로 바꿔주면서 Tensor로 만들어주는 역할입니다.
https://albumentations.ai/docs/api_reference/pytorch/transforms/#albumentations.pytorch.transforms.ToTensorV2
해당 부분은 다른 augmentation이 들어가서 그런것으로 보입니다.
아마도 scale out하면서 grid가 들어가는 것 같은데, scale out 시 border가 grid로 들어가는 옵션이 있을 확률이 높습니다.
그게 아니라면, augmentation시 grid가 들어가는 부분이 있거나 할거구요.

다냐니라

2023.01.26 14:15

위에 사진 1,2의 차이점은 ToTensorV2를 사진1에는 안넣고 사진 2에는 넣어서 테스트를 한건데, 위와같은 결과가 나와서요.. 한번 다시 확인해 봐야겠네요! 감사합니다 :)

Redix6

2023.01.26 12:27

아래 코드로 보시면 확실히 아실 수 있습니다.
import albumentations as A
from albumentations.pytorch.transforms import ToTensorV2
import matplotlib.pyplot as plt

img = cv2.imread("train/TRAIN_00000.jpg")
imgTensor = ToTensorV2()(image=img)["image"]
plt.imshow(imgTensor.permute(1,2,0).numpy())
plt.show()

다냐니라

2023.01.26 15:00

아 맞네요 ToTensorV2는 상관이 없네요.. 제가 이미지를 뽑아보려고 .reshape을 하면서 그렇게 된 것같습니다. permutation을 사용해야 하는군요... 하나 배웠습니다. 감사합니다.

📣 댓글 작성 창의 위치가 댓글 리스트 상단으로 이동하였습니다!