root layout

패스트캠퍼스

  1. 강의 질문
  2. AI TECH

"01-09. (5) 영상 밝기 차이에 따른 분류 - 실습 (Resnet 활용, imgaug package)"

2026.01.08 18:24 작성

아래 Colab 소스코드 블록에서 "ModuleNotFoundError: No module named 'imgaug'" 오류 메시지가 발생하는데 오류 도우미를 통해 아래코드를 삽입하면 또 다른 오류가 발생됩니다.


해결코드 : "!pip install imgaug"


또 다른 오류, "AttributeError: `np.sctypes` was removed in the NumPy 2.0 release. Access dtypes explicitly instead."


해결할 수 있는 코드가 어떻게 되는 지 확인 부탁합니다.


---------------------------

from imgaug import augmenters as iaa

import imgaug as ia



# imgaug를 사용한 커스텀 데이터셋 정의

class ImgAugTransform:

    def __init__(self):

        self.aug = iaa.Sequential([

            iaa.Fliplr(0.5),  # 50% 확률로 좌우 반전

            iaa.Affine(rotate=(-20, 20), mode='symmetric'),  # -20도에서 20도 사이로 회전

        ])



    def __call__(self, img):

        img = np.array(img)

        return self.aug.augment_image(img)


답변 

연관 질문

커뮤니티 질문보기