▷ 머신러닝 프로젝트 시작(2)
어제 S.A를 작성하고 시작하기에 앞서 데이터셋을 찾아보았다.
https://www.kaggle.com/datasets/shanmukh05/agedetection
Age-Detection
Age,Sex,Ethnicity detection
www.kaggle.com
찾은 데이터셋에는 1살부터 116살까지의 이미지파일이 들어있었고, 이걸 토대로 코랩을 통해 분석해본 결과.
base model | batch size | drop out(rate) | activation function | optimizer | learning rate | epochs | loss function | loss | acc | val_loss | val_acc | size | seed |
InceptionV3 | 32 | 0.25 | relu | Adam | 0.1 | 10 | categorical_crossentropy | 4.108 | 0.0919 | 4.0965 | 0.0934 | 224, 224 | 2021 |
InceptionV3 | 32 | 0.25 | relu | Adam | 0.01 | 10 | categorical_crossentropy | 4.0736 | 0.0925 | 4.0555 | 0.0934 | 224, 224 | 2021 |
InceptionV3 | 32 | 0.25 | relu | Adam | 0.001 | 10 | categorical_crossentropy | 3.0454 | 0.1782 | 224, 224 | 2021 | ||
InceptionV3 | 32 | 0.25 | relu | Adam | 0.0001 | 10 | categorical_crossentropy | 5.3998 | 0.1249 | 224, 224 | 2021 | ||
InceptionV3 | 32 | 0.25 | relu | Adam | 0.01 | 10 | categorical_crossentropy | 3.2214 | 0.1615 | 3.5998 | 0.1228 | 128, 128 | 2021 |
VGG16 | 64 | 0.25 | relu | SGD | 0.1 | 10 | categorical_crossentropy | 3.0352 | 0.1765 | 3.3141 | 0.1522 | ||
InceptionV3 | 32 | 0.25 | relu | Adam | 0.002 | 10 | categorical_crossentropy | 2.7694 | 0.2572 | 3.4411 | 0.1377 | ||
InceptionV3 | 64 | 0.25 | relu | Adam | 0.01 | 10 | categorical_crossentropy | 4.0749 | 0.0925 | 4.0587 | 0.0934 | 224,224 | |
InceptionV3 | 32 | 0.25 | relu | Adam | 1 | 10 | categorical_crossentropy | 4.078 | 0.0924 | 4.0595 | 0.0934 | ||
InceptionV3 | 64 | 0.25 | relu | Adam | 0.1 | categorical_crossentropy | - | - | - | 0.0934 | |||
아침퀴즈 | 64 | 0.3 | relu | Adam | - | 30 | categorical_crossentropy | 2.8022 | 0.219 | 3.4366 | 0.1549 | ||
InceptionV3 | 32 | 0.2 | relu | Adam | 0.001 | 20 | categorical_crossentropy | 1.7573 | 0.4295 | 4.8816 | 0.1164 | 224,224 | |
InceptionV3 | 32 | 0.2 | relu | SGD | 0.001 | 10 | categorical_crossentropy | 3.0477 | 0.1878 | 3.375 | 0.1428 | 128,128 | |
InceptionV3 | 128 | 0.4 | relu | SGD | 0.01 | 10 | categorical_crossentropy | 2.2486 | 0.3533 | 3.8475 | 0.1264 | 128,128 | |
InceptionV3 | 128 | 0.2 | relu | SGD | 0.001 | 10 | categorical_crossentropy | 1.0095 | 0.7163 | 5.4321 | 0.1215 | 128,128 | 3000 |
ResNet50 | 32 | 0.25 | relu | Adam | 0.1 | 10 | categorical_crossentropy | 4.1067 | 0.0925 | 4.0885 | 0.0934 | 224, 224 | 2021 |
ResNet50 | 32 | 0.25 | relu | Adam | 0.1 | 10 | categorical_crossentropy | 4.1954 | 0.0922 | 4.0917 | 0.0934 | 224, 224 | 1021 |
ResNet50 | 32 | 0.25 | relu | Adam | 0.01 | 10 | categorical_crossentropy | 3.9598 | 0.1217 | 4.0912 | 0.099 | 224, 224 | 2021 |
InceptionResNetV2 | 32 | 0.25 | relu | Adam | 0.01 | 10 | categorical_crossentropy | 4.0618 | 0.092 | 4.0326 | 0.0934 | 299, 299 | 2021 |
InceptionV3 | 32 | 0.3 | relu | Adam | 0.1 | 10 | categorical_crossentropy | 4.1073 | 0.0925 | 4.0878 | 0.0934 | 128, 128 | 2022 |
결과는 처참했다..모델 / 배치사이즈 / 드랍아웃/ 옵티마이저 / IR 등등 변경하면서 분석을 해보았지만, val_acc는
나아질 기미가 보이지 않았다. 또, 코랩을 무료로 이용하다보니 GPU 사용량 제한으로 인해서 1 계정당 GPU를 통해
분석하는 것이 쉽지않았다.
다른 데이터셋을 찾아보면서 진행방향을 찾아봐야 할 듯 싶다.
'개발_TIL' 카테고리의 다른 글
개발_TIL | 2022-05-23(25) (0) | 2022.05.23 |
---|---|
개발_TIL | 2022-05-20(24) (0) | 2022.05.20 |
개발_TIL | 2022-05-18(22) (0) | 2022.05.18 |
개발_TIL | 2022-05-17(21) (0) | 2022.05.17 |
개발_TIL | 2022-05-16(20) (0) | 2022.05.16 |