Tensorflow lite 관련 질문드립니다.

Midabo   
   조회 943   추천 0    

Tensorflow 2.0 을 이용해 모델을 학습하고 TF lite 로 추론을 하는 앱을 제작하려고 합니다. 


https://github.com/tensorflow/examples/blob/master/lite/examples/image_classification/android/README.md 

위의 링크를 따라서 제작해서 잘 작동은 하는데요,


문제는 레이블이 3개 이상인 모델은 클래스명과 확률이 잘 표시되는데,

이진분류 (레이블이 두개) 인 경우 클래스명, 확률 모두 표시가 안되고 있습니다.

andriod studio 를 이용해서 코드 디버깅을 하고 있는데, 자바가 처음이라 좀 난해합니다.


혹시 해결 혹은 고민하신 경험 있으신 분은 경험을 좀 나누어 주시기 바랍니다.


감사합니다.



김동규1 2019-12
2진분류일경우 sigmoid가 낫지싶은데
활성화 함수 softmax사용하신거아닌가요?
     
Midabo 2019-12
당연히 Sigmoid 썼는데요, 아마 프런트엔드 문제일 듯 싶습니다.




제목Page 1/7
2015-12   12943   백메가
2014-04   2632844   회원K
02-19   248   주니2
02-14   485   주니2
02-12   413   leejhkk
02-06   516   leejhkk
02-06   495   방방
01-15   1027   mozart
01-02   1304   영산회상
12-28   954   GPU사고싶다
2019-12   944   Midabo
2019-12   1157   앞짱구
2019-11   1462   Leo
2019-11   1193   색작업자
2019-11   1222   경기북
2019-11   1954   tpp52
2019-11   3807   tpp52
2019-11   1915   경기북
2019-10   1714   존쓴즈로션
2019-10   1939   MS77
2019-10   1351   MikroTik이진
2019-10   2086   leeplay