지난 12.02~ 12.16 프로그래머스에서 진행하는 머신러닝 온라인 잡페어에 참가하였습니다.
흥미로운 점은 머신 러닝 관련 과제를 통해 지원자들의 역량을 확인하고 추후 채용과정에서
이 점을 준다는 점이 새로웠습니다.
과제는 추천시스템과 컴퓨터 비전이 있었는데 그 중 컴퓨터 비전에 관련 챌린지에 참가하였습니다.
과제를 진행하기 위해서 데이터를 다운을 받았는데 train, test 데이터가 각각 다른 폴더에 나누어 담아지지 않았고 하나의 폴더에 8000개의 이미지가 있었습니다.
그래도 csv 파일에 train, test 이미지의 이름이 있었습니다. 과제를 시작하기 전에 train, test 데이터를 디렉토리를 만들어 나누어야 한다는 생각에 하루정도 삽질을 했습니다...... 굳이 안그래도 됫었는데... 하하하
csv를 넣으면 이미지를 나누어주는 마법같은 package가 있을 거라 생각하고 계속 찾아보았지만 그런건 없었네요. 그래서 간단하게 만들었습니다.
혹시라도 csv파일을 기준으로 이미지를 옮기고 싶으면 부족하지만 참고바랍니다. 삽질한 김에 포스팅 ㅎㅎ..
먼저 package를 import 하고 train 파일 이름이 있는 csv파일을 읽습니다.
그 다음 train 파일 이름을 list로 저장합니다. 이는 for문을 이용해서 해당 경로에 있는 이미지 하나 하나를 옮기는 작업을 위한 준비과정입니다.
데이터가 있는 폴더와 옮길 폴더를 지정합니다. 그후 shutil의 move함수를 이용해서 이미지 파일을 옮깁니다.
성공적으로 train 폴더에 이미지를 옮겼습니다.!!