[P-stage] Image Classification Day 1

해당 포스팅은 네이버 부스트캠프 P-stage를 위해 개인 학습 내용을 정리했습니다.

1. 목표

• Dataloader 만들기
• 이미지를 불러오고 확인하기

 

2. 진행 내용

1) Dataloader 만들기

 Dataloader를 만들 때, split은 sklearn의 train_test_split을 적용하면 좋을 것 같습니다. 내일 강의를 듣고 추가 예정입니다.

2) 이미지 불러오고 확인하기

•  os.listdir() 명령어는 숨김파일도 다 불러오는 기능이 존재합니다. 그래서 이를 방지하려면 glob을 사용하는 것이 더 좋은 것 같습니다.
• 서버에서 import cv2 했을 때, ImportError: libGL.so.1: cannot open shared object file: No such file or directory 에러가 발생했습니다. !apt-get -y install libgl1-mesa-glx 명령어를 쳐주면 해결됩니다. PIL.image를 사용하는 것도 좋은 방법입니다.

 

3. 회고

 아직은 py파일로 구성하는 것이 어렵습니다. pytorch에 익숙하지 않아서 그런 것 같습니다. 틈틈히 pytorch에 대해 공부해야할 필요가 있을 것 같습니다.