Tensorflow 이미지 분류 스크립트

Question

저는 TensorFlow에서 꽤 새로 왔습니다.

나는 TensorFlow Poets의 모든 CodeLab을 따랐으며 데이지와 장미 만 사용하여 모델을 교육했습니다.

그런 다음 이미지를 테스트하고 분류하기 위해 label_image.py 스크립트를 사용했습니다. https://gist.githubusercontent.com/wolffg/541c97a74dfc0d77c4b8fd2a946a5b41/raw/578853dd26180dbf5bfc66eb40fdd13fb3aba4d6/TensorFlow%2520codelab

그러나 장미 나 데이지가없는 임의의 이미지로 시도했을 때 나는 여전히 높은 점수를 얻었습니다.
어떻게 그 스크립트를 수정하거나 장미 나 데이지 또는 아무것도없는 것처럼 말할 수있는 다른 스크립트를 아는 경우.

Answer 1

Softmax는 네트워크가 결정을 내릴 것을 권장하므로 대개 무언가로 끝날 것입니다.

Luis가 말했듯이 "알 수없는"클래스를 카테고리에 추가 할 수 있습니다. 이것은 아마도 가장 간단한 옵션 일 것입니다.

softmax 출력이 확률이기 때문에, 점수가 충분히 낮은 것이 아무것도없는 것으로 간주되는 임계 값을 설정해보십시오. 위와 같이 softmax는 좋은 결정입니다. 제작자가 너무 잘 작동하지 않을 수 있습니다.

2 클래스 예측을 사용하고 있으므로 다른 옵션은 softmax를 sigmoid 함수로 대체하여 0 (예 : 장미)과 1 (예 : 데이지) 사이의 출력을 얻는 것입니다. 그런 다음 불확실한 것으로 간주되는 범위 (예 : 0.4 ~ 0.6)를 선택하고이를 "알 수 없음"레이블로 사용할 수 있습니다.

Answer 2

당신이 필요로하는 것이 있다면, 당신의 훈련 세트를 수정하고 또 다른 클래스로서 "Nothing"의 이미지 예제를 추가하십시오.