How to distinguish results from empty image and image containing 1

potterhsu / SVHNClassifier-PyTorch

A PyTorch implementation of Multi-digit Number Recognition from Street View Imagery using Deep Convolutional Neural Networks (http://arxiv.org/pdf/1312.6082.pdf)

MIT License

198 stars 45 forks source link

How to distinguish results from empty image and image containing 1 #8

Open ghwn opened 3 years ago

ghwn commented 3 years ago

Hi,

I am using this classifier to recognize square images containing numbers.

In most cases this classifier performs very well but when there are no numbers in the image, it returns 1 (length: 1, digits: [1, 10, 10, 10, 10]).

It's confusing because it also returns same result when recognizing images containing only 1.

How can I distinguish them?