Modified image_inference.py

[PythonImageDeveloper]

This for multi-scale images, that is if in the test images there are any scale image this code resize and don't need to have all images greater than 224 / 331 .

from __future__ import division
from __future__ import print_function
from __future__ import absolute_import

import re
import os
import sys
import tarfile
import argparse
#import datetime
import numpy as np
import tensorflow as tf
from six.moves import urllib
import time

# For NasNet-Large must be 331 and NasNet-mobile 224
Image_Size=224

os.environ['TF_CPP_MIN_LOG_LEVEL']='2'
model = "./inference/frozen_nasnet_mobile_plants_seedling.pb"
model_graph = tf.Graph()
with model_graph.as_default():
    with tf.gfile.FastGFile(model, 'rb') as f:
        graph_def = tf.GraphDef()
        graph_def.ParseFromString(f.read())
        _ = tf.import_graph_def(graph_def, name='')
        input_layer = model_graph.get_tensor_by_name("input:0")
        output_layer = model_graph.get_tensor_by_name('final_layer/predictions:0')

config = tf.ConfigProto()
config.gpu_options.allow_growth = True
inference_session = tf.Session(graph = model_graph, config=config)


def decode_image(jpeg_file):
    with tf.device('/cpu:0'):
        decoder_graph = tf.Graph()
        with decoder_graph.as_default():
            decoded_image = tf.image.decode_jpeg(jpeg_file)
            image_resized=tf.image.resize_images(decoded_image,[Image_Size,Image_Size])
            normalized_image = tf.divide(image_resized, 255)
            #reshaped_image = tf.reshape(normalized_image, [224, 224])
            #expande_image=tf.expand_dims(reshaped_image,axis=0)
        with tf.Session(graph = decoder_graph) as image_session:
        # image_session = tf.Session(graph = decoder_graph)
            input_0 = image_session.run(normalized_image)
    return input_0


def diagnose_image(inference_session, input_image):
    with tf.device('/gpu:0'):
        predictions = inference_session.run(output_layer, feed_dict={input_layer: input_image})
    predictions = np.squeeze(predictions)
    return predictions


def main(arguments):
    """ Inference the whole src root directory """
    src_root = "./test"
    dst_root = "./result"
    label_map_path = "./flowers/labels.txt"
    if not os.path.isdir(dst_root):
        os.mkdir(dst_root)

    images = os.listdir(src_root)
    output_file = os.path.join(dst_root, "output_result.txt")
    result_file = open(output_file, "a")

    label_map_file = open(label_map_path)
    label_map = {}
    for line_number, label in enumerate(label_map_file.readlines()):
        label_map[line_number] = label[:-1]
        line_number += 1
    label_map_file.close()


    for image in images:
        image_path = os.path.join(src_root, image)
        #start = datetime.datetime.now()
        T1=time.time()
        with tf.gfile.FastGFile(image_path, 'rb') as jpeg_file_raw:
            jpeg_file = jpeg_file_raw.read()
            input_0 = decode_image(jpeg_file)

            image_height = input_0.shape[0]
            image_width = input_0.shape[1]
            image_height_center = int(image_height/2)
            image_width_center = int(image_width/2)

            tl_crop = input_0[0:Image_Size, 0:Image_Size]
            tr_crop = input_0[0:Image_Size, image_width-Image_Size:image_width]
            bl_crop = input_0[image_height-Image_Size:image_height, 0:Image_Size]
            br_crop = input_0[image_height-Image_Size:image_height, image_width-Image_Size:image_width]
            center_crop = input_0[image_height_center - (Image_Size//2): image_height_center + (Image_Size//2), image_width_center - (Image_Size//2): image_width_center + (Image_Size//2)]

            input_concat = np.asarray([tl_crop, tr_crop, bl_crop, br_crop, center_crop])
            input_batch = input_concat.reshape(-1, Image_Size, Image_Size, 3)

        predictions = diagnose_image(inference_session, input_batch)
        overall_result = np.argmax(np.sum(predictions, axis=0))

        result_file.write(image_path + "\n")
        result_file.write(str(overall_result) + "\n")

        #end = datetime.datetime.now()
        T2=time.time()
        print(image_path)
        print(overall_result, label_map[overall_result])
        #print("Time cost: ", end - start, "\n")
        print("Time Prediction: ", T2-T1, "\n")

    result_file.close()


if __name__ == '__main__':
    sys.exit(main(sys.argv[1:]))

i have question about this part: ```

        tl_crop = input_0[0:Image_Size, 0:Image_Size]
        tr_crop = input_0[0:Image_Size, image_width-Image_Size:image_width]
        bl_crop = input_0[image_height-Image_Size:image_height, 0:Image_Size]
        br_crop = input_0[image_height-Image_Size:image_height, image_width-Image_Size:image_width]
        center_crop = input_0[image_height_center - (Image_Size//2): image_height_center + (Image_Size//2), image_width_center - (Image_Size//2): image_width_center + (Image_Size//2)]

        input_concat = np.asarray([tl_crop, tr_crop, bl_crop, br_crop, center_crop])
        input_batch = input_concat.reshape(-1, Image_Size, Image_Size, 3)

what is the purpose of these? tl_crop , tr_crop , bl_crop , br_crop , center_crop
I think you use this as batch_size=5 , right ? why you don't send image directly ? 
why this shape must be [5,image_size,image_size,3] , why 5 ? when i use shape  [1,image_size,image_size,3] , this predict all images only 0 label, why?

[huiofficial]

Come up with the same question.