deep_cortex_detect.c

Go to the documentation of this file.

 
#include "../common/common.h"
 
int main(void)
{
    sb_t_err err = SB_ERR_NONE;
    sb_t_info* sb_info = NULL;
    SB_HANDLE project = NULL;
    sb_t_image* image = NULL;
    sb_t_res* res = NULL;
    sb_t_par* par = NULL;
    sb_t_folder* folder = NULL;
    sb_t_solution_info* solution = NULL;
    int i;
 
    printf("Initialize the SqueezeBrains library\n");
    CHECK_FN_GOTO(sb_init("../sb.lic"));
 
#if defined(_WIN32)
    // "../../../win_x64/dl_framework" is the pre-defined relative path between working directory and SB Deep Learning Framework into the installation folder.
    CHECK_FN_GOTO(sb_init_dl("../../../win_x64/dl_framework", SB_DL_LIBRARY_PYTORCH));
    CHECK_FN_GOTO(sb_init_dl("../../../win_x64/dl_framework", SB_DL_LIBRARY_OPENVINO));
#else 
    CHECK_FN_GOTO(sb_init_dl(NULL, SB_DL_LIBRARY_PYTORCH));
    CHECK_FN_GOTO(sb_init_dl(NULL, SB_DL_LIBRARY_OPENVINO));
#endif
 
    // Print library version.
    CHECK_FN_GOTO(sb_get_info(&sb_info, 0));
    printf("sb ver. %s, compiled on %s %s\n", sb_info->version_str, sb_info->compile_date, sb_info->compile_time);
 
    printf("wait for license...\n");
    CHECK_FN_GOTO(wait_license());
 
    CHECK_FN_GOTO(sb_solution_get_info("solution/deep_cortex_tutorial." SB_SOLUTION_EXT, &solution));
    CHECK_FN_GOTO(sb_project_load(&project, "solution/deep_cortex_tutorial." SB_SOLUTION_EXT, solution->projects.info[solution->current_project].uuid, SB_PROJECT_MODE_DETECTION_ONLY));
    CHECK_FN_GOTO(sb_project_get_par(project, &par));
    par->devices.type = SB_DEVICE_GPU_NVIDIA; // set GPU device
    par->devices.id[0] = 0;                   // set the first GPU detected on the machine (with index 0)
    CHECK_FN_GOTO(sb_project_set_par(project, par));
    CHECK_FN_GOTO(sb_folder_load(&folder, "dataset", "png", 1, 0));
    for (i = 0; i < folder->n; i++)
    {
        CHECK_FN_GOTO(sb_image_load(&image, folder->file[i].name));
        CHECK_FN_GOTO(sb_project_detection(project, image, NULL, NULL, NULL));
        CHECK_FN_GOTO(sb_project_get_res(project, &res, 0));
        // Print the results
        printf("image: %s\n", folder->file[i].fname);
        printf("\t model=\"%s\"\t weight=%f\n", res->samples.sample[0].model_name, res->samples.sample[0].weight);
        // Destroy the handles.
        CHECK_FN_GOTO(sb_image_destroy(&image));
        CHECK_FN_GOTO(sb_res_destroy(&res));
    }
 
FnExit:
    sb_destroy_info(&sb_info);
    sb_par_destroy(&par);
    sb_project_destroy(&project);
    sb_image_destroy(&image);
    sb_res_destroy(&res);
    sb_folder_destroy(&folder);
    sb_solution_destroy_info(&solution);
 
    printf("Release SqueezeBrains library\n");
    sb_release();
 
    printf("Press ENTER to terminate\n");
    getchar();
    return (int)err;
}