대학 대항전 : 퍼즐 이미지 AI 경진대회

Algorithm | Dacon Monthly | University Challenge | Vision | Self-Supervised Learning | Technological Innovation | Classification

 

제가 사용한 이미지 증강용 코드

2024.01.24 18:11 1,630 Views language

아래의 데이터증강관련 게시물을 참고해서, 제가 사용하는 증강코드를 공유해드립니다.

증강속도 면에서 개선하여 pool의 multiprocessing을 활용하였습니다. 
(Computing resource에 따라 조정해주시고 써주세요)

이미지를 증강할때 90도, 180도, 270도로 퍼즐의 회전을 주었고, 재배열 시켰습니다.
Train 이미지를 그대로 사용하여 증강을 하려고 하였으나, 회전한 이미지의 로직을 만들어내는데에 가성비가 안나와서, 
원본을 만들고, 원본에서 재배열하는 방식으로 코드를 진행했습니다. 감사합니다

i7-14700k, 4090 기준으로 한 7만개 재배열하는데 3초걸린것 같습니다.

Code