surface_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_roi* roi = NULL;
    sb_t_image* image = NULL;
    sb_t_res* res = NULL;
    sb_t_par par;
    sb_t_solution_info* solution = NULL;
 
    printf("Initialize the SqueezeBrains library\n");
    CHECK_FN_GOTO(sb_init("../sb.lic"))
 
    // 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/surface_tutorial." SB_SOLUTION_EXT, &solution));
    CHECK_FN_GOTO(sb_project_load(&project, "solution/surface_tutorial." SB_SOLUTION_EXT, solution->projects.info[solution->current_project].uuid, SB_PROJECT_MODE_DETECTION_ONLY))
    CHECK_FN_GOTO(sb_image_load(&image, "dataset/surface_001.png"))
    CHECK_FN_GOTO(sb_roi_create(&roi, image->width, image->height))
    // Set a rectangular roi equal to image dimensions.
    CHECK_FN_GOTO(sb_roi_set_rect(roi, 255, sb_rect(0, 0, image->width, image->height), 0))
    // Set the blob analysis par
    CHECK_FN_GOTO(sb_project_get_par(project, &par))
    par.surface_blob_analysis = 1;
    par.blob_par.area.min = 10;
    par.blob_par.merge_area.min = 20;
    par.blob_par.blob_rle = 1;
    par.blob_par.blob_contour = 1;
    par.blob_par.merge_distance = 10;
    CHECK_FN_GOTO(sb_project_set_par(project, &par))
    CHECK_FN_GOTO(sb_project_detection(project, image, roi, NULL, NULL))
    CHECK_FN_GOTO(sb_project_get_res(project, &res, 0))
    // Print the results
    printf("%d defect blobs found:\n", res->global.surface->blobs->size);
    for (int j = 0; j < res->global.surface->blobs->size; ++j)
        printf("    ul=(%d,%d), size=(%d,%d), area=%d\n", res->global.surface->blobs->blob[j].rect.x, res->global.surface->blobs->blob[j].rect.y, res->global.surface->blobs->blob[j].rect.width, res->global.surface->blobs->blob[j].rect.height, res->global.surface->blobs->blob[j].area);
 
FnExit:
    sb_destroy_info(&sb_info);
    sb_project_destroy(&project);
    sb_image_destroy(&image);
    sb_roi_destroy(&roi);
    sb_res_destroy(&res);
    sb_solution_destroy_info(&solution);
 
    printf("Release SqueezeBrains library\n");
    sb_release();
 
    printf("Press ENTER to terminate\n");
    getchar();
    return (int)err;
}