サンプルコード ( マルチラベル画像分類 )

マルチラベル画像分類のサンプルコードです

• 最も単純なサンプルコード: L.5- smp_mlcls_simple()
• ベンチマークを計測するサンプルコード: L.99: smp_mlcls_benchmark()
• FIE の関数を前処理として行うサンプルコード: L.218: smp_mlcls_with_fie()

#pragma once
#include "samples.h"
#include "oal_aloc.h"
 
int smp_mlcls_simple(const char* model_name, const char* img_name)
{
    INT ret = F_ERR_NONE;
    FHANDLE hsrc = NULL;
    H_MODEL hmodel = NULL;
    MODEL_CATEGORY model_category;
    MODEL_BACKEND model_backend;
    INT ch, w, h;
    FLOAT* scores = NULL;
    size_t n_scores;
    FLOAT score_thresh = 0.5;
    UINT* result_cls_ids = NULL;
    FLOAT* result_scores = NULL;
    size_t num_pred_labels = 0;
 
    // FIEライブラリの使用前に必ずコールする必要があります。
    fnFIE_setup();
 
    // 画像読込
    ret = fnFIE_load_img_file(img_name, &hsrc, F_COLOR_IMG_TYPE_UC8);
    printf_s("img: %d\r\n", ret);
    if (F_ERR_NONE != ret) { goto finally; }
 
    // ライセンスチェック
    ret = fnPDL_check_license();
    printf_s("lic : %d\r\n", ret);
    if (F_ERR_NONE != ret) { goto finally; }
 
    // ハンドルの生成
    hmodel = fnPDL_create_handle();
    if (NULL == hmodel) {
        ret = F_ERR_NOMEMORY;
        printf_s("failed to create handle\r\n");
        goto finally;
    }
    else { printf_s("handle created\r\n"); }
    if (F_ERR_NONE != ret) { goto finally; }
 
    // モデル読込
    ret = fnPDL_load_model(model_name, hmodel);
    printf_s("load: %d\r\n", ret);
    if (F_ERR_NONE != ret) { goto finally; }
 
    // モデルの種別の確認
    ret = fnPDL_get_model_category(hmodel, &model_category);
    printf_s("par : %d  category=%d\r\n", ret, model_category);
    if (F_ERR_NONE != ret) { goto finally; }
    if (MULTI_LABEL_CLASSIFICATION != model_category) {
        ret = F_ERR_INVALID_OBJECT;
        printf_s("unmatch model\r\n");
        goto finally;
    }
 
    // モデルの推論バックエンドの確認
    ret = fnPDL_get_model_backend(hmodel, &model_backend);
    printf_s("ret : %d  backend=%d\r\n", ret, model_backend);
    if (F_ERR_NONE != ret) { goto finally; }
 
    // モデルの期待する画像パラメータの確認
    ret = fnPDL_get_input_image_size(hmodel, &ch, &w, &h);
    printf_s("par : %d  ch=%d  w=%d  h=%d\r\n", ret, ch, w, h);
    if (F_ERR_NONE != ret) { goto finally; }
 
    // 推論実行
    ret = fnPDL_predict(hmodel, hsrc, &scores, &n_scores);
    printf_s("pred: %d\r\n", ret);
    if (F_ERR_NONE != ret) { goto finally; }
    ret = fnPDL_postproc_multi_label_cls(
        hmodel, scores, n_scores, score_thresh, &result_cls_ids, &result_scores, &num_pred_labels
    );
    printf_s("postproc: %d\r\n", ret);
    if (F_ERR_NONE != ret) { goto finally; }
    if (NULL != result_cls_ids && NULL != result_scores) {
        size_t i_result;
        for (i_result = 0; i_result < num_pred_labels; i_result++) {
            printf_s("    label%d: %.6e\r\n", result_cls_ids[i_result], result_scores[i_result]);
        }
    }
    printf_s("\r\n");
 
finally:
    // ハンドルの破棄
    fnPDL_dispose(hmodel);
 
    if (NULL != hsrc) { fnFIE_free_object(hsrc); }
    if (NULL != scores) { fnOAL_free(scores); }
    if (NULL != result_cls_ids) { fnOAL_free(result_cls_ids); }
    if (NULL != result_scores) { fnOAL_free(result_scores); }
 
    fnFIE_teardown();
 
    return ret;
}
 
int smp_mlcls_benchmark(const char* model_name, const char* img_name, const int bench_iter)
{
    // benchmark 用
    std::chrono::system_clock::time_point start, end;
 
    INT ret = F_ERR_NONE;
    FHANDLE hsrc = NULL;
    H_MODEL hmodel = NULL;
    MODEL_CATEGORY model_category;
    FLOAT* scores = NULL;
    size_t n_scores;
    FLOAT score_thresh = 0.5;
    UINT* result_cls_ids = NULL;
    FLOAT* result_scores = NULL;
    size_t num_pred_labels = 0;
 
    // FIEライブラリの使用前に必ずコールする必要があります。
    fnFIE_setup();
 
    // 画像読込
    ret = fnFIE_load_img_file(img_name, &hsrc, F_COLOR_IMG_TYPE_UC8);
    printf_s("img: %d\r\n", ret);
    if (F_ERR_NONE != ret) { goto finally; }
 
    // ライセンスチェック
    ret = fnPDL_check_license();
    printf_s("lic : %d\r\n", ret);
    if (F_ERR_NONE != ret) { goto finally; }
 
    // ハンドルの生成
    hmodel = fnPDL_create_handle();
    if (NULL == hmodel) {
        ret = F_ERR_NOMEMORY;
        printf_s("failed to create handle\r\n");
        goto finally;
    }
    else { printf_s("handle created\r\n"); }
 
    // モデル読込
    ret = fnPDL_load_model(model_name, hmodel);
    printf_s("load: %d\r\n", ret);
    if (F_ERR_NONE != ret) { goto finally; }
 
    // モデルの種別の確認
    ret = fnPDL_get_model_category(hmodel, &model_category);
    printf_s("par : %d  category=%d\r\n", ret, model_category);
    if (F_ERR_NONE != ret) { goto finally; }
    if (MULTI_LABEL_CLASSIFICATION != model_category) {
        ret = F_ERR_INVALID_OBJECT;
        printf_s("unmatch model\r\n");
        goto finally;
    }
 
    printf_s("--- start ---\r\n");
    printf_s("number, elapsed[msec]\r\n");
    printf_s("    result(class, score)\r\n");
    // 本体ループ
    for (int i = 0; i < bench_iter; i++) {
        double elapsed;
 
        // 開始時刻
        start = std::chrono::system_clock::now();
 
        // 推論実行
        ret = fnPDL_predict(hmodel, hsrc, &scores, &n_scores);
        if (F_ERR_NONE != ret) {
            printf_s("pred. err: %d\r\n", ret);
            goto finally;
        }
        ret = fnPDL_postproc_multi_label_cls(
            hmodel, scores, n_scores, score_thresh, &result_cls_ids, &result_scores, &num_pred_labels
        );
        if (F_ERR_NONE != ret) {
            printf_s("postproc. err: %d\r\n", ret);
            goto finally;
        }
 
        // 完了時刻
        end = std::chrono::system_clock::now();
 
        // 推論時間 [msec]
        elapsed = std::chrono::duration_cast<std::chrono::microseconds>(end - start).count() / (double)1000;
 
        // コンソールに出力
        printf_s("%04d, %.3e\r\n", i, elapsed);
 
        if (NULL != result_cls_ids && NULL != result_scores) {
            size_t i_result;
            for (i_result = 0; i_result < num_pred_labels; i_result++) {
                printf_s("    label%d: %.6e\r\n", result_cls_ids[i_result], result_scores[i_result]);
            }
        }
 
        if (NULL != result_cls_ids) { 
            fnOAL_free(result_cls_ids);
            result_cls_ids = NULL;
        }
        if (NULL != result_scores) {
            fnOAL_free(result_scores);
            result_scores = NULL;
        }
    }
    printf_s("--- finish ---\r\n");
    printf_s("\r\n");
 
finally:
    // ハンドルの破棄
    fnPDL_dispose(hmodel);
    
    if (NULL != hsrc) { fnFIE_free_object(hsrc); }
    if (NULL != scores) { fnOAL_free(scores); }
    if (NULL != result_cls_ids) { fnOAL_free(result_cls_ids); }
    if (NULL != result_scores) { fnOAL_free(result_scores); }
 
    fnFIE_teardown();
 
    return ret;
}
 
int smp_mlcls_with_fie(const char* model_name, const char* img_name)
{
    INT ret = F_ERR_NONE;
    FHANDLE hsrc = NULL;
    H_MODEL hmodel = NULL;
    MODEL_CATEGORY model_category;
    FLOAT* scores = NULL;
    size_t n_scores;
    FLOAT score_thresh = 0.5;
    UINT* result_cls_ids = NULL;
    FLOAT* result_scores = NULL;
    size_t num_pred_labels = 0;
 
    FHANDLE hfiltered = NULL;
    INT ch, w, h, type;
    INT_PTR step;
 
    // FIEライブラリの使用前に必ずコールする必要があります。
    fnFIE_setup();
 
    // 画像読込
    ret = fnFIE_load_img_file(img_name, &hsrc, F_COLOR_IMG_TYPE_UC8);
    printf_s("img: %d\r\n", ret);
    if (F_ERR_NONE != ret) { goto finally; }
 
    // ライセンスチェック
    ret = fnPDL_check_license();
    printf_s("lic : %d\r\n", ret);
    if (F_ERR_NONE != ret) { goto finally; }
 
    // ハンドルの生成
    hmodel = fnPDL_create_handle();
    if (NULL == hmodel) {
        ret = F_ERR_NOMEMORY;
        printf_s("failed to create handle\r\n");
        goto finally;
    }
    else { printf_s("handle created\r\n"); }
 
    // モデル読込
    ret = fnPDL_load_model(model_name, hmodel);
    printf_s("load: %d\r\n", ret);
    if (F_ERR_NONE != ret) { goto finally; }
 
    // モデルの種別の確認
    ret = fnPDL_get_model_category(hmodel, &model_category);
    printf_s("par : %d  category=%d\r\n", ret, model_category);
    if (F_ERR_NONE != ret) { goto finally; }
    if (MULTI_LABEL_CLASSIFICATION != model_category) {
        ret = F_ERR_INVALID_OBJECT;
        printf_s("unmatch model\r\n");
        goto finally;
    }
 
    // 読み込んだ画像をそのまま推論する
 
    // 推論実行
    ret = fnPDL_predict(hmodel, hsrc, &scores, &n_scores);
    printf_s("pred(original): %d\r\n", ret);
    if (F_ERR_NONE != ret) { goto finally; }
    ret = fnPDL_postproc_multi_label_cls(
        hmodel, scores, n_scores, score_thresh, &result_cls_ids, &result_scores, &num_pred_labels
    );
    printf_s("postproc(original): %d\r\n", ret);
    if (F_ERR_NONE != ret) { goto finally; }
    if (NULL != result_cls_ids && NULL != result_scores) {
        size_t i_result;
        for (i_result = 0; i_result < num_pred_labels; i_result++) {
            printf_s("    label%d: %.6e\r\n", result_cls_ids[i_result], result_scores[i_result]);
        }
    }
 
    // 次で使うので開放してNULLを代入しなおす
    if (NULL != result_cls_ids) { 
        fnOAL_free(result_cls_ids);
        result_cls_ids = NULL;
    }
    if (NULL != result_scores) {
        fnOAL_free(result_scores);
        result_scores = NULL;
    }
 
    // フィルタをかけた画像を推論する
 
    // ここでは単純な平均化フィルタ
    ret = fnFIE_img_get_params(hsrc, &ch, &type, &step, &w, &h);
    if (F_ERR_NONE != ret) { goto finally; }
    hfiltered = fnFIE_img_root_alloc(type, ch, w, h);
    if (NULL == hfiltered) { ret = F_ERR_NOMEMORY; goto finally; }
    ret = fnFIE_average(hsrc, hfiltered, 0, 0);
    if (F_ERR_NONE != ret) { goto finally; }
 
    // 推論実行
    ret = fnPDL_predict(hmodel, hfiltered, &scores, &n_scores);
    printf_s("pred(average): %d\r\n", ret);
    if (F_ERR_NONE != ret) { goto finally; }
    ret = fnPDL_postproc_multi_label_cls(
        hmodel, scores, n_scores, score_thresh, &result_cls_ids, &result_scores, &num_pred_labels
    );
    printf_s("postproc(average): %d\r\n", ret);
    if (F_ERR_NONE != ret) { goto finally; }
    if (NULL != result_cls_ids && NULL != result_scores) {
        size_t i_result;
        for (i_result = 0; i_result < num_pred_labels; i_result++) {
            printf_s("    label%d: %.6e\r\n", result_cls_ids[i_result], result_scores[i_result]);
        }
    }
    printf_s("\r\n");
 
finally:
    // ハンドルの破棄
    fnPDL_dispose(hmodel);
 
    if (NULL != hsrc) { fnFIE_free_object(hsrc); }
    if (NULL != hfiltered) { fnFIE_free_object(hfiltered); }
    if (NULL != scores) { fnOAL_free(scores); }
    if (NULL != result_cls_ids) { fnOAL_free(result_cls_ids); }
    if (NULL != result_scores) { fnOAL_free(result_scores); }
 
    // 終了処理
    fnFIE_teardown();
 
    return ret;
}