Function main

examples/tensorflow_wrapper/label_image/label_image.cpp:172–260

static void dummy_deallocator(void* data, size_t len, void* arg) {}

int main(int argc, char* argv[])
{
    const string root_path = get_root_path();
#ifdef MOBILE_NET
    const string model_file = root_path + "models/frozen_mobilenet_v1_224.pb";
    const string image_file = root_path + "tests/images/cat.jpg";
    const string label_file = root_path + "models/synset_words.txt";
    int input_height = 224;
    int input_width = 224;
    float input_mean = -127;
    float input_std = 127;
#elif RESNET50
    const string model_file = root_path + "models/frozen_resnet50v1.pb";
    const string image_file = root_path + "tests/images/bike.jpg";
    const string label_file = root_path + "models/synset_words.txt";
    int input_height = 224;
    int input_width = 224;
    float input_mean = 0;
    float input_std = 1;
#else
    const string model_file = root_path + "models/inception_v3_2016_08_28_frozen.pb";
    const string image_file = root_path + "tests/images/grace_hopper.jpg";
    const string label_file = root_path + "models/imagenet_slim_labels.txt";
    int input_height = 299;
    int input_width = 299;
    float input_mean = 0;
    float input_std = 255;
#endif

#ifdef MOBILE_NET
    string input_layer = "input";
    string output_layer = "MobilenetV1/Predictions/Softmax";
#elif RESNET50
    string input_layer = "input";
    string output_layer = "resnet_v1_50/predictions/Softmax";
#else
    string input_layer = "input";
    string output_layer = "InceptionV3/Predictions/Softmax";
#endif

    // Load and initialize the model
    TF_Graph* graph = TF_NewGraph();
    TF_Session* session = LoadGraph(model_file.c_str(), graph);
    if(!session)
        return -1;

    // Read image file
    cv::Mat img;
    float* input_data = ReadImageFile(image_file.c_str(), img, input_height, input_width, input_mean, input_std);
    if(!input_data)
        return -1;

    // Create input tensor
    vector<TF_Output> input_names;
    vector<TF_Tensor*> input_values;

    TF_Operation* input_name = TF_GraphOperationByName(graph, input_layer.c_str());
    input_names.push_back({input_name, 0});