Programming17 [Pytorch] Segmentation task에서 Albumentation 이용하기 인공지능 학습할 때 overfitting을 막고 일반화 성능을 올리기 위한 다양한 방법들이 존재합니다. 그 중 쉽게 적용할 수 있는 방법 중 하나가 데이터 증강(Augmentation) 입니다. 오늘은 데이터 증강 라이브러리인 Albumentation를 이용하여 Segmentation task에 적용해본 방법을 정리하려고 합니다. Pytorch의 경우 TorchVision 라이브러리에 있는 torchvision.transforms을 이용할 수 있지만, Albumentation 만의 직관적인 사용법과 빠른 처리속도에 사용하지 않을 이유가 없습니다. 추천추천 1. Albumentation 설치하기 albumentation은 python 3.6 버전 이상을 지원합니다. pip를 이용하여 아주 쉽게 설치할 수 .. 2022. 6. 22. Ubuntu에서 Pycharm 실행 스크립트 만들기 및 가상환경 연동 다양한 Python IDE 중에 저는 Pycharm을 즐겨쓰고 있습니다. 큰 이유 중 하나는 코드 history를 시간대 별로 저장되고 볼 수 있는 기능입니다. 이건 git과는 별개로 좀 더 가볍게 과거의 코드를 확인하거나 지금 코드와 비교하기 위해서 애용하는 기능입니다. 사설은 여기까지 하고.. 우분투 20.04 환경에서 Pycharm을 설치해보도록 합시다. 1. Ubuntu에서 Pycharm 설치 및 실행 스크립트 만들기 우분투에서 Pycharm 설치 방법은 2가지가 있습니다. 첫번째는 설치 파일을 통해서 설치하는 것이고, 두번째는 터미널에서 명령어로 설치하는 방법입니다. 1) 설치 파일 이용하기 아래 링크에서 무료 버전인 Community 버전을 클릭하여 tar 압축 파일을 다운 받습니다. 👉 우분.. 2022. 6. 20. Ubuntu 20.04에서 Anaconda 가상환경 및 Pytorch 제대로 설치하기 세상 모든 개발자들은 공감할 것이다. 개발 환경이 한번 꼬이면 얼마나 주옥같은지를.. 나와 같은 삽질을 하는 사람이 생기지 않도록 기록을 남기고자 한다. 1. Anaconda 설치하기 제일 먼저 가상환경을 구성하기 위한 아나콘다(Anaconda) 부터 설치한다. 작은 프로젝트를 하더라도 프로젝트마다 의존하는 라이브러리들이 다르기 때문에 라이브러리 끼리 충돌되고 싸우는 일이 발생하는 대참사를 막으려면 가상환경 구성은 필수적이다. 아나콘다 다운로드 홈페이지에서 설치 파일을 다운받는다. 👉 아나콘다 다운로드 링크 (https://www.anaconda.com/products/distribution) 홈페이지의 맨 밑으로 가면 각 OS마다 설치 파일 목록들을 볼 수 있다. Python 버전이 적혀있는데 가상환경.. 2022. 6. 14. Overlay two transparent image with PIL import numpy as np from PIL import Image img = Image.open("background.png") # convert numpy array to PIL image mask_img = Image.fromarray(mask).convert('RGBA') # change mask color opacity_level = 130 # 0~255 new_data = [] for item in mask_img.getdata() if item[0] != 0 and item[1] != 0 and item[2] != 0: new_data.append((R, G, B, opacity_level)) else: new_data.append((0, 0, 0, 0)) # no-opacity. so.. 2022. 5. 22. 이전 1 2 3 4 5 다음
