[코드 공유] rotate 연산시 발생하는 문제 해결

[shjas94]

def rotate_img(self, img):
    '''
    input : PIL image
    output : rotated image
    '''
    img = np.array(img)
    prob = np.random.rand(1)
    if prob < 0.5:
        img_rotated = cv2.rotate(img, cv2.ROTATE_90_CLOCKWISE)
    else:
        img_rotated = cv2.rotate(img, cv2.ROTATE_90_COUNTERCLOCKWISE)
    return Image.fromarray(img_rotated)

이미지를 불러온 후, 세로로 긴 이미지는 해당 코드로 변환시켜주시면 됩니다.
50% 확률로 시계방향으로 회전
50% 확률로 반시계방향으로 회전합니다.

[remaindere]

혹시 getitem에서 img ratio를 계산하고, ratio에 따라 rotate 할지에 관한 code가 있나요? 없으면 작성하러 가려구요

[shjas94]

def __getitem__(self, i):
        item = self.data[i]
        image = Image.open(item["path"])
        if self.rgb == 3:
            image = image.convert("RGB")
        elif self.rgb == 1:
            image = image.convert("L")
        else:
            raise NotImplementedError

        if image.size[0] / image.size[1] <= 0.6:
            image = self.rotate_img(image)

[remaindere]

if image.size[0] / image.size[1] <= 0.6:
이부분에서
왜 < 1이 아닌지 궁금합니다!

[shjas94]

aspect ratio 0.7 정도까지는 rotate된 이미지와 정상적인 이미지가 애매하게 섞여있습니다.