[📰논문] pytorch로 전이학습 이미지 분류하는데 질문 있습니다.

[익명(14.35)]

모바일넷 전이학습으로 실습중입니다.

이진분류로는 어느정도 나오는데 

class 하나 더 추가해서 세 개로 하려니 

하나만 accuracy 가 높고 나머지 두개는 거의 못맞추네요

전체 acc가 0.32 나오는 걸로 봐선 거의 전부 class 하나로 예측해버리는거 같은데

이럴때는 뭘 시도해 볼 수 있을까요

데이터가 문제일 가능성이 높겠죠?

[지잡컴공(wlwkqzjarhd)]

코드 깃헙에 올려서 링크로 달아주면 더 도움되는 답변이 달릴 것 같습니다

[익명(14.35)]

https://github.com/YOOyong/mobilenet
코드입니다

[익명(14.35)]

테스트에 쓰인 데이터셋이 너무 쓰레기였나봅니다. 다른 세트로 해보니 적당히 나오네요

[지잡컴공(wlwkqzjarhd)]

해결되셨다니 다행이네요~