딥러닝 공부방

안녕하세요 ㅎㅎ 오늘은 Albumentations 모듈을 사용해서 이미지 transformation을 정의하고, 데이터셋에 적용하겠습니다. Albumentations 모듈은 torchvision.transformer 보다 빠르게 작동하며, object detection task에서 이미지를 transform 적용하면 바운딩 박스도 함께 transform 되도록 할 수 있습니다. 즉, 아주 아주 편리합니다 ㅎㅎ 우선 albumentations 모듈을 설치합니다. # install transformation package !pip install -U albumentations 커스텀 데이터셋을 생성합니다. 저의 경우에는 VOC dataset을 사용했습니다. class VOCDataset(Dataset): d..

이미지 불러오기 OpecCV 라이브러리를 이용하여 이미지를 불러오는 방법을 알아보겠습니다. import cv2 import sys img = cv2.imread('cat.bmp') # imread로 영상을 불러온다. if img is None: print('Imager load failed!') # 이미지가 없으면 출력 sys.exit() cv2.namedWindow('image') # OpenCV에서 지원하는 창을 생성하는 명령어 cv2.imshow('image', img) # 첫 번째는 어떤 창에 불러올 것이냐, 두 번째는 어떤 것을 불러올 것이냐 cv2.waitKey() # 키보드입력을 누를떄까지 보여줌 cv2.destroyAllWindows() # 모든 창을 닫음 위 명령어를 실행하면 윈도우 창에..