Upload first version

examples/mobilenet/cpp/build-android.sh

@@ -0,0 +1,61 @@
+#!/bin/bash
+set -e
+
+usage() {
+    echo "Usage: $0 [-a <target_abi>]"
+    echo "  -a <target_abi> : Target ABI (default: arm64-v8a)"
+    echo "  -h              : Show this help message"
+    exit 1
+}
+
+# Default values
+TARGET_ABI=arm64-v8a
+
+# Parse arguments
+while getopts 'a:h' opt; do
+  case "$opt" in
+    a)
+      TARGET_ABI=$OPTARG
+      ;;
+    h)
+      usage
+      ;;
+    *)
+      usage
+      ;;
+  esac
+done
+
+if [ -z "${ANDROID_NDK_PATH}" ]; then
+    if [ -n "${ANDROID_NDK}" ]; then
+        ANDROID_NDK_PATH=${ANDROID_NDK}
+    elif [ -n "${ANDROID_NDK_HOME}" ]; then
+        ANDROID_NDK_PATH=${ANDROID_NDK_HOME}
+    else
+        echo "Error: ANDROID_NDK_PATH is not set."
+        echo "Please set ANDROID_NDK_PATH to your Android NDK directory."
+        exit 1
+    fi
+fi
+
+ROOT_PWD=$(cd "$(dirname $0)" && pwd)
+BUILD_DIR=${ROOT_PWD}/build/android
+
+echo "Building for Android..."
+echo "NDK_PATH: ${ANDROID_NDK_PATH}"
+echo "TARGET_ABI: ${TARGET_ABI}"
+echo "BUILD_DIR: ${BUILD_DIR}"
+
+mkdir -p ${BUILD_DIR}
+cd ${BUILD_DIR}
+
+cmake ../../src \
+    -DCMAKE_TOOLCHAIN_FILE=${ANDROID_NDK_PATH}/build/cmake/android.toolchain.cmake \
+    -DANDROID_ABI=${TARGET_ABI} \
+    -DANDROID_PLATFORM=android-24 \
+    -DCMAKE_BUILD_TYPE=Release \
+    -DOpenCV_DIR=${ROOT_PWD}/../../../dependency/opencv/opencv-android-sdk-build/sdk/native/jni/abi-${TARGET_ABI}
+
+make -j$(nproc)
+
+echo "Build complete. Executable in ${BUILD_DIR}/mobilenet_v2_demo"

examples/mobilenet/cpp/build-linux.sh

@@ -0,0 +1,19 @@
+#!/bin/bash
+set -e
+
+#
+# Copyright (C) 2024–2025 Amlogic, Inc. All rights reserved.
+#
+# 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.
+#
+### TO DO

examples/mobilenet/cpp/src/CMakeLists.txt

@@ -0,0 +1,39 @@
+cmake_minimum_required(VERSION 3.5)
+project(mobilenet_v2_demo)
+
+set(CMAKE_CXX_STANDARD 17)
+
+# Set NNSDK path
+set(NNSDK_ROOT "${CMAKE_SOURCE_DIR}/../../../../dependency/nnsdk")
+include_directories(${NNSDK_ROOT}/include)
+include_directories(${CMAKE_SOURCE_DIR}/../../../../common)
+
+# Set dependency path
+set(3RDPARTY_DIR "${CMAKE_SOURCE_DIR}/../../../../dependency")
+
+if(CMAKE_SYSTEM_NAME STREQUAL "Android")
+    if (ANDROID_ABI STREQUAL "arm64-v8a")
+        link_directories(${NNSDK_ROOT}/lib/android/arm64-v8a)
+    else()
+        link_directories(${NNSDK_ROOT}/lib/android/armeabi-v7a)
+    endif()
+    # Android needs log
+    link_libraries(log)
+elseif(CMAKE_SYSTEM_NAME STREQUAL "Linux")
+    link_directories(${NNSDK_ROOT}/lib/linux/lib64_yocto)
+endif()
+
+# Find OpenCV
+message(STATUS "OpenCV_DIR: ${OpenCV_DIR}")
+find_package(OpenCV REQUIRED)
+include_directories(${OpenCV_INCLUDE_DIRS})
+
+add_executable(mobilenet_v2_demo
+    main.cpp
+    ${CMAKE_SOURCE_DIR}/../../../../common/model_loader.cpp
+)
+
+target_link_libraries(mobilenet_v2_demo
+    ${OpenCV_LIBS}
+    nnsdk
+)

examples/mobilenet/cpp/src/main.cpp

@@ -0,0 +1,153 @@
+/*
+ * Copyright (C) 2024–2025 Amlogic, Inc. All rights reserved.
+ *
+ * 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 <iostream>
+#include <fstream>
+#include <string>
+#include <vector>
+#include <algorithm>
+#include <opencv2/opencv.hpp>
+#include <opencv2/opencv.hpp>
+#include "nn_sdk.h"
+#include "model_loader.h"
+
+const int MODEL_INPUT_WIDTH = 224;
+const int MODEL_INPUT_HEIGHT = 224;
+const int TOP_K = 5;
+
+typedef struct {
+    int index;
+    float score;
+} ClassScore;
+
+bool compare_scores(const ClassScore& a, const ClassScore& b) {
+    return a.score > b.score;
+}
+
+std::vector<std::string> load_labels(const std::string& path) {
+    std::vector<std::string> labels;
+    std::ifstream file(path);
+    std::string line;
+    if (file.is_open()) {
+        while (std::getline(file, line)) {
+            // Trim whitespace
+            line.erase(line.find_last_not_of(" \n\r\t")+1);
+            labels.push_back(line);
+        }
+        file.close();
+    } else {
+        std::cerr << "Warning: Could not open labels file: " << path << std::endl;
+    }
+    return labels;
+}
+
+
+
+int main(int argc, char** argv) {
+    std::string model_path = argv[1];
+    std::string image_path = argv[2];
+    std::string labels_path = argv[3];
+
+    std::cout << "MobileNetV2 Demo" << std::endl;
+    std::cout << "Model: " << model_path << std::endl;
+    std::cout << "Image: " << image_path << std::endl;
+    std::cout << "Labels: " << labels_path << std::endl;
+
+    // 1. Initialize Network
+    void* context = init_network(model_path.c_str());
+    if (!context) {
+        std::cerr << "Failed to initialize network." << std::endl;
+        return -1;
+    }
+
+    // 2. Load and Preprocess Image
+    cv::Mat img = cv::imread(image_path);
+    if (img.empty()) {
+        std::cerr << "Failed to load image from " << image_path << std::endl;
+        uninit_network(context);
+        return -1;
+    }
+
+    cv::Mat img_resized;
+    cv::resize(img, img_resized, cv::Size(MODEL_INPUT_WIDTH, MODEL_INPUT_HEIGHT));
+
+    cv::Mat img_rgb;
+    cv::cvtColor(img_resized, img_rgb, cv::COLOR_BGR2RGB);
+
+    nn_input inData;
+    memset(&inData, 0, sizeof(nn_input));
+    inData.input_type = BINARY_RAW_DATA;
+    inData.input = img_rgb.data;
+    inData.input_index = 0;
+    inData.size = MODEL_INPUT_WIDTH * MODEL_INPUT_HEIGHT * 3 * sizeof(unsigned char);
+
+    if (aml_module_input_set(context, &inData) != 0) {
+         std::cerr << "Failed to set input." << std::endl;
+         uninit_network(context);
+         return -1;
+    }
+
+    // 3. Run Inference
+    auto start_time = std::chrono::high_resolution_clock::now();
+
+    aml_output_config_t outconfig;
+    memset(&outconfig, 0, sizeof(aml_output_config_t));
+    outconfig.typeSize = sizeof(aml_output_config_t);
+    outconfig.format = AML_OUTDATA_FLOAT32;
+    
+    nn_output* outdata = (nn_output*)aml_module_output_get(context, outconfig);
+    if (!outdata) {
+         std::cerr << "Failed to run network (get output)." << std::endl;
+         uninit_network(context);
+         return -1;
+    }
+
+    auto end_time = std::chrono::high_resolution_clock::now();
+    std::chrono::duration<double, std::milli> inference_time = end_time - start_time;
+    std::cout << "Inference time: " << inference_time.count() << " ms" << std::endl;
+
+    // 4. Postprocess
+    if (outdata->num <= 0) {
+        std::cerr << "No output from network." << std::endl;
+        uninit_network(context);
+        return -1;
+    }
+
+    float* output_buffer = (float*)outdata->out[0].buf;
+    int num_classes = outdata->out[0].size / sizeof(float);
+
+    std::vector<ClassScore> scores;
+    for (int i = 0; i < num_classes; ++i) {
+        scores.push_back({i, output_buffer[i]});
+    }
+
+    std::sort(scores.begin(), scores.end(), compare_scores);
+
+    std::vector<std::string> labels = load_labels(labels_path);
+
+    std::cout << "\nTop-" << TOP_K << " Classification Results:" << std::endl;
+    for (int i = 0; i < std::min(TOP_K, num_classes); ++i) {
+        int idx = scores[i].index;
+        float score = scores[i].score;
+        std::string label = (idx < labels.size()) ? labels[idx] : "Class " + std::to_string(idx);
+        printf("  %d. %-20s (score: %.6f)\n", i + 1, label.c_str(), score);
+    }
+
+    // 5. Cleanup
+    uninit_network(context);
+
+    return 0;
+}