predicting.py 文件源码

def evaluate_frozen(self, filename):
        # Load image
        with open(UPLOAD_FOLDER + '/' + filename, 'rb') as f:
            image_string = f.read()

        # Session
        with tf.Session() as sess:
            prob_values = sess.run(self.probs, feed_dict={
                self.input_image_string: image_string
                })

            pred_idx = prob_values[0].argsort()[-5:][::-1]
            pred_class = self.classes[pred_idx - 1] # from 1001 to 1000 classes
            pred_score = np.around(100*prob_values[0][pred_idx], decimals=2) # two decimals

            return list(pred_class), list(pred_score)