반응형
pytorch로 연구할 때, data loader부분소스코드를 짜는것과 데이터준비하는것만 해도 절반은 했다는 말이 있다.
그 만큼 data를 잘 불러와서 처리하는게 중요하고 어렵다는 말인데 나중에 겪을 시행착오를 위해 미리미리 쉬운 케이스부터 기록해놓겠다.
아래는 Bulat의 "How far are we from solving the 2d and 3d face alignment problem" 논문에서 제공하는 코드를 실행하는 부분 중에서도 데이터를 가져오는 파일의 코드이다.
class facedb는 data를 load하는 기능의 클래스이다. 이 글에서 더 이상 자세하게 볼 필요는 없다.
오픈소스코드에서는 저렇게(위 사진처럼) 경로지정하는 부분까지 제공해주는데 이 때, 경로를 어떻게 지정해주어야 하며 다운받은 300W, 300W_LP폴더는 어떻게 놔야할까? 가공해야할까? 가 문제이다.
결론은 웹에서 다운받은 데이터폴더는 따로 정리하지말고 코드에서 그대로 가져다놓으면 된다.
나중에 참고를 위해 이 코드에서 내가 설정한 경로설정법을 아래에 명시해놓았다.
해당 프로젝트 경로 : <user path>/face
데이터 폴더 경로 :
300W : <user path>/face/300Ws/300W <----- 여기서 300W는 다운받은 그대로를 옮겨놓으면 된다.
300W-LP : <user path>/face/300Ws/300W_LP <------ 여기서 300W_LP는 다운받은 그대로를 옮겨놓으면 된다.
반응형