Main.cpp

// Copyright 2021 The WebNN-native Authors
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

#include "examples/MobileNetV2/MobileNetV2.h"

int main(int argc, const char* argv[]) {
    // Set input options for the example.
    MobileNetV2 mobilevetv2;
    if (!mobilevetv2.ParseAndCheckExampleOptions(argc, argv)) {
        return -1;
    }

    // Pre-process the input image.
    std::vector<float> processedPixels(mobilevetv2.mModelHeight * mobilevetv2.mModelWidth *
                                       mobilevetv2.mModelChannels);
    if (!utils::LoadAndPreprocessImage(&mobilevetv2, processedPixels)) {
        return -1;
    }

    // Create a graph with weights and biases from .npy files.
    const wnn::ContextOptions options =
        utils::CreateContextOptions(mobilevetv2.mDevicePreference, mobilevetv2.mPowerPreference);
    wnn::Context context = CreateCppContext(&options);
    context.SetUncapturedErrorCallback(
        [](WNNErrorType type, char const* message, void* userData) {
            if (type != WNNErrorType_NoError) {
                dawn::ErrorLog() << "Error type is " << type << ", message is " << message;
            }
        },
        &mobilevetv2);
    wnn::GraphBuilder builder = wnn::CreateGraphBuilder(context);
    wnn::Operand output = mobilevetv2.mLayout == "nchw" ? mobilevetv2.LoadNchw(builder)
                                                        : mobilevetv2.LoadNhwc(builder);

    // Build the graph.
    const std::chrono::time_point<std::chrono::high_resolution_clock> compilationStartTime =
        std::chrono::high_resolution_clock::now();
    wnn::Graph graph = utils::Build(builder, {{"output", output}});
    if (!graph) {
        dawn::ErrorLog() << "Failed to build graph.";
        return -1;
    }

    const TIME_TYPE compilationElapsedTime =
        std::chrono::high_resolution_clock::now() - compilationStartTime;
    dawn::InfoLog() << "Compilation Time: " << compilationElapsedTime.count() << " ms";

    // Compute the graph.
    std::vector<float> result(utils::SizeOfShape(mobilevetv2.mOutputShape));
    // Do the first inference for warming up if nIter > 1.
    if (mobilevetv2.mNIter > 1) {
        utils::Compute(context, graph, {{"input", processedPixels}}, {{"output", result}});
    }

    std::vector<TIME_TYPE> executionTime;
    for (int i = 0; i < mobilevetv2.mNIter; ++i) {
        std::chrono::time_point<std::chrono::high_resolution_clock> executionStartTime =
            std::chrono::high_resolution_clock::now();
        utils::Compute(context, graph, {{"input", processedPixels}}, {{"output", result}});
        executionTime.push_back(std::chrono::high_resolution_clock::now() - executionStartTime);
    }

    // Print the result.
    utils::PrintExexutionTime(executionTime);
    utils::PrintResult(result, mobilevetv2.mLabelPath);
    dawn::InfoLog() << "Done.";
}