251007 - Project (12)

시작하며

어제 학습시켜둔 모델이 오늘 아침에 끝나서 일어나서 바로 테스트해봤는데 아직 성능이 조금 아쉬웠다.

Model v2 샘플 테스트

모델 예측 분포 테스트
각 클래스당 20개 샘플 테스트
======================================================================
디바이스 : cuda:0

실제 클래스: 캔
예측 분포 :
    → 캔: 20개 (100.0%)

실제 클래스: 유리
예측 분포 :
    → 유리: 19개 (95.0%)
    → 비닐: 1개 (5.0%)

실제 클래스: 종이
예측 분포 :
    → 종이: 18개 (90.0%)
    → 스티로폼: 1개 (5.0%)
    → 비닐: 1개 (5.0%)

실제 클래스: 플라스틱
예측 분포 :
    → 캔: 3개 (15.0%)
    → 유리: 1개 (5.0%)
    → 종이: 1개 (5.0%)
    → 플라스틱: 10개 (50.0%)
    → 스티로폼: 2개 (10.0%)
    → 비닐: 3개 (15.0%)

실제 클래스: 스티로폼
예측 분포 :
    → 스티로폼: 20개 (100.0%)

실제 클래스: 비닐
예측 분포 :
    → 비닐: 20개 (100.0%)
======================================================================

플라스틱이 이상하게 계속 성능이 안 나와서 학습용 데이터에서 노이즈가 많이 들어간 데이터를 직접 추린 뒤 종이와 플라스틱에는 가중치를 조금씩 더 부여하였다. 그리고 데이터 변형도 조금 수정해서 데이터의 다양성을 더 증가시켜 학습 난이도를 올리고 20 Epoch로 Model v3 학습을 시작하였다.

마치며

요즘 하루에 최소 12시간씩 모델 학습을 시키고 있어서 3일 정도 컴퓨터를 계속 켜두고 있다. 내일 Model v3는 부디 만족스러운 결과가 나와서 다음 작업에 들어가고 싶다.