0616 - 실제로 적용된 모델 성능 향상 테크닉

zln·2022년 6월 16일
0

학습한 내용

실습 코드

1. train_pant.py - main code

2. utils_file.py - data augmentation

  • dataset을 train, test로 나눠서 각각의 data에 augmentation을 적용한다.

3. customdataset.py - CustomData

  • 전체 데이터 경로를 불러와서 데이터 가져옴
  • 경로를 split해 Yoga Poses마다 label을 정해줌
    ex) Downdog=0, Goddess=1, Plank=2, Tree=3, Warrior2=4
  • Image 열어줌

4. models_build


-> 여기서 resnet을 사용할 예정

5. configer.py - 하이퍼 파라미터 값 설정

학습한 내용 중 어려웠던 점

이해를 못할 정도는 아닌데 혼자하기에는 어려울 것 같다.

해결방법 작성

학습 소감

일단 train code를 만들어야되는데 . .
이 컴퓨터로 model 돌릴때 너무 오래걸려서 문제다.

0개의 댓글