TensorFlow2 내 데이터로 ResNet 돌려보기 📌 캡스톤 프로젝트 내 데이터로 pkl 파일 만들어서 적용하는 과정은 🤯내 데이터로 CNN 돌려보기 🧐 왜 ResNet? 현재 모델의 정확도가 좋지 않은 상태. 논문에 나오는 구조를 참고한 CNN 모델의 깊이가 얕아서 발생한 문제라고 판단. 교수님이 Resnet 모델을 추천해주셨다. 원래 만들었던 CNN 구조 super(AcousticSoundModel, self).__init__() self.conv1 = Conv2D(filters=64, kernel_size=[3, 3], padding='SAME', activation=tf.nn.relu) self.drop1 = Dropout(rate=0.2) self.pool1 = MaxPool2D(padding='SAME') self.conv2 = Conv2D(.. 2020. 9. 9. 내 데이터로 CNN 돌려보기 📌 캡스톤 프로젝트 데이터 준비하기 우리는 6개 클래스를 가지고 있고 어쩌구 근데 지금은 3개 클래스에 대한 데이터만 있어서 일단 세개로만 해보겠음. AWS Sagemaker를 이용할 계획인데, 그러려면 데이터셋을 pkl 파일로 만들어야 한다. 진짜 여기저기 삽질하다가 이 깃헙을 발견하고 따라해봤다! train, valid, test 폴더를 만들고 데이터를 7:2:1의 비율로 넣어주었다. 1. label이 들어간 csv 파일 만들어주기 csv 파일을 만들어주는 코드를 작성했다. import os import natsort import csv import re file_path = 'test/' file_lists = os.listdir(file_path) file_lists = natsort.natsor.. 2020. 9. 9. 이전 1 다음
