모델 결과 문제 about pill_classification HOT 13 CLOSED

모델의 input 으로 들어간 게 뭐야?
이미지만 들어간 거야?

from pill_classification.

이미지만 들어갔습니다

from pill_classification.

모델 구성 시 weight는 모두 초기화 한 뒤, 구조만 가져왔습니다.

from pill_classification.

기존 학습된 weight 가져와서 초기화한 후,
네 데이터를 사용해서 다시 학습(fine-tuning)해보도록 해.

from pill_classification.

R, G, B 3개 채널의 픽셀 값이 입력으로 들어간 거야?

from pill_classification.

네! 이미지 shape = (18170, 100, 100, 3) 입니다. 이미지 크기를 줄이는 것도 memery error때문에 고려해봐야할 거 같습니다.

from pill_classification.

기존 학습된 weight를 가져온 후에 초기화를 시키고, fine-tuning을 하는 것과, 구조만 가져온 후 tuning을 하는 거는 다른 방식인가요?

from pill_classification.

Weight 가져와서 전이학습 하는거야.

from pill_classification.

Weight 가져와야하니 이미지 크기는 바꾸지말고 배치사이즈만 조절해.

from pill_classification.

미디어랩스 컴퓨터로 진행중인데 batch size가 지금 32,16,8일때 안돌아가고, 4일때 하나는 돌아가는데 그 뒤의 모델을 하려하면 메모리 부족 에러가 뜹니다ㅠㅠ
모델이 원하는 데이터 shape는 (227,227,3) 크기입니다.

from pill_classification.

• 데이터가 너무 없는 레이블 없애기
• 227,227,3 크기로 조정하기
• 배치사이즈 낮추기

이 3가지 진행해본 다음에 모델 학습 진행해보겠습니다.

from pill_classification.

서버실 컴퓨터사용중? 몇번?

from pill_classification.

ML313 컴퓨터입니다.

from pill_classification.