profile
그래프와 이미지를 연구하는 데이터사이언티스트입니다.
post-thumbnail

torchvision의 데이터 augmentation 기법들 살펴보기

다량의 양질의 데이터는 좋은 딥러닝 모델을 만드는데 필수 조건입니다. 그런데 실제 산업 현장에서는 데이터가 충분히 확보 되어있지 않거나 명확히 분류되어있지 않은 경우가 많습니다. 특히, 공정과정에서 발생하는 이미지는 이런 경우가 비일비재합니다. 그러므로, 모델에 학습 시키기전 데이터 augmentation 과정은 필수입니다. torchvision.transforms은 이미지의 다양한 전처리 기능을 제공하며 이를 통해 데이터 augmentation도 손쉽게 구현할 수 있습니다. 이에 본 포스팅에서는 torchvision의 transforms 메써드에서 제공하는 다양한 데이터 증강용 함수를 기능 중점적으로 소개드리고자 합니다. 더 자세한 내용은 pytorch에서 제공하는 [공식 doc](https://pytorch.org/vision/stable/autoexamples/plottransforms.html#sphx-glr-auto-examples-plot-transf

2022년 4월 8일
·
0개의 댓글
·