surface_svl_simple.c

Go to the documentation of this file.

 
#include "../common/common.h"
 
sb_t_err create_surface_project_file(SB_HANDLE* surface);
 
sb_t_err set_labeling(SB_HANDLE surface);
 
sb_t_err execute_training(void);
 
#define MODEL_NAME "w"
 
int main(void)
{
    sb_t_err err = SB_ERR_NONE;
    sb_t_info* sb_info = NULL;
    SB_HANDLE project = 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(create_surface_project_file(&project))
 
    
    CHECK_FN_GOTO(set_labeling(project))
 
    
    CHECK_FN_GOTO(execute_training())
 
FnExit:
    sb_destroy_info(&sb_info);
    sb_project_destroy(&project);
 
    printf("Release SqueezeBrains library\n");
    sb_release();
 
    printf("Press ENTER to terminate\n");
    getchar();
    return (int)err;
}
 
sb_t_err create_surface_project_file(SB_HANDLE* surface)
{
    sb_t_err err = SB_ERR_NONE;
    sb_t_par par;
 
    // Erase the project file to begin from zero
    remove("dataset/surface." SB_SOLUTION_EXT);
 
    // Create the project file
    CHECK_FN_GOTO(sb_project_create(surface, "surface", SB_PROJECT_TYPE_SURFACE))
    CHECK_FN_GOTO(sb_project_get_par(*surface, &par))
    CHECK_FN_GOTO(sb_par_add_model(&par, MODEL_NAME))
    CHECK_FN_GOTO(sb_par_add_level(&par, MODEL_NAME, 1.0f))
    CHECK_FN_GOTO(sb_project_set_par(*surface, &par))
    CHECK_FN_GOTO(sb_project_save(*surface, "dataset/surface." SB_SOLUTION_EXT, SB_PROJECT_MODE_DETECTION_AND_SVL))
FnExit:
    return err;
}
 
sb_t_err set_labeling(SB_HANDLE surface)
{
    sb_t_err err = SB_ERR_NONE;
    SB_HANDLE image_info = NULL;
    sb_t_roi *defects = NULL;
    sb_t_roi *roi = NULL;
    sb_t_image *img = NULL;
 
    // Erase the rtn to begin from zero
    remove("dataset/surface_001.rtn");
 
    // load image and create image info
    CHECK_FN_GOTO(sb_image_load(&img, "dataset/surface_001.png"))
    CHECK_FN_GOTO(sb_image_info_load(&image_info, "dataset/surface_001.png", surface))
    // Reset the image info
    CHECK_FN_GOTO(sb_image_info_reset(image_info))
    CHECK_FN_GOTO(sb_image_info_set_type(image_info, SB_IMAGE_INFO_TYPE_SVL))
    // Get the roi
    CHECK_FN_GOTO(sb_image_info_get_roi(image_info, &roi, img->width, img->height, 0))
    // Set a ellipse ROI
    CHECK_FN_GOTO(sb_roi_set_ellipse(roi, 255, roi->img->width / 2, roi->img->height / 2, 250, 150, 1))
    CHECK_FN_GOTO(sb_image_info_set_roi(image_info, roi))
    // Get the defects roi
    CHECK_FN_GOTO(sb_image_info_get_roi_defects(image_info, &defects, img->width, img->height, 0))
    // Set a circle defect area
    CHECK_FN_GOTO(sb_roi_set_circle(defects, 255, 478, 156, 12, 1))
    CHECK_FN_GOTO(sb_image_info_set_roi_defects(image_info, defects))
    CHECK_FN_GOTO(sb_image_info_save(image_info))
FnExit:
    sb_image_info_destroy(&image_info);
    sb_image_destroy(&img);
    sb_roi_destroy(&roi);
    sb_roi_destroy(&defects);
    return err;
}
 
sb_t_err execute_training(void)
{
    sb_t_err err = SB_ERR_NONE;
    SB_HANDLE svl = NULL;
    sb_t_svl_res* svl_res = NULL;
    sb_t_par par;
    sb_t_solution_info* solution = NULL;
 
    CHECK_FN_GOTO(sb_solution_get_info("dataset/surface." SB_SOLUTION_EXT, &solution));
    CHECK_FN_GOTO(sb_project_load(&svl, "dataset/surface." SB_SOLUTION_EXT, solution->projects.info[solution->current_project].uuid, SB_PROJECT_MODE_DETECTION_AND_SVL))
    CHECK_FN_GOTO(sb_svl_reset(svl))
    CHECK_FN_GOTO(sb_project_get_par(svl, &par))
    strcpy(par.svl.image_ext, SB_IMAGE_FILE_EXTENSIONS);
    strcpy(par.svl.project_path, "dataset/");
    par.svl.sl.goodness_target = 0.5f;
    CHECK_FN_GOTO(sb_project_set_par(svl, &par))
    printf("Training is running, please wait . . .\n");
    CHECK_FN_GOTO(sb_svl_run(svl))
    CHECK_FN_GOTO(sb_svl_get_res(svl, &svl_res))
    printf("goodness=%f\n", svl_res->global.goodness);
    printf("time=%1.3fs\n", svl_res->time_ms / 1000.0f);
    CHECK_FN_GOTO(sb_project_save(svl, NULL, SB_PROJECT_MODE_DETECTION_AND_SVL))
FnExit:
    sb_svl_destroy_res(&svl_res);
    sb_project_destroy(&svl);
    sb_solution_destroy_info(&solution);
    return err;
}