retina_svl_simple.c

Go to the documentation of this file.

 
#include "../common/common.h"
 
sb_t_err create_retina_project_file(SB_HANDLE* retina);
 
sb_t_err set_samples(SB_HANDLE retina);
 
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 retina = 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_retina_project_file(&retina))
 
    
    CHECK_FN_GOTO(set_samples(retina))
 
    
    CHECK_FN_GOTO(execute_training())
 
FnExit:
    sb_destroy_info(&sb_info);
    sb_project_destroy(&retina);
 
    printf("Release SqueezeBrains library\n");
    sb_release();
 
    printf("Press ENTER to terminate\n");
    getchar();
    return (int)err;
}
 
sb_t_err create_retina_project_file(SB_HANDLE* retina)
{
    sb_t_err err = SB_ERR_NONE;
    sb_t_par* par = NULL;
 
    // Erase the project file to begin from zero
    remove("../images/wheels." SB_SOLUTION_EXT);
 
    // Create the project file
    CHECK_FN_GOTO(sb_project_create(retina, "wheels", SB_PROJECT_TYPE_RETINA))
    CHECK_FN_GOTO(sb_project_get_par(*retina, &par))
    CHECK_FN_GOTO(sb_par_add_model(par, MODEL_NAME))
    par->models.model[0].obj_size = sb_size(136, 128);
    par->models.model[0].obj_min_distance = sb_size(100, 128);
    par->models.model[0].obj_stride_coarse = sb_size(8, 8);
    par->models.model[0].obj_stride_fine = sb_size(4, 4);
    par->models.model[0].perturbations.size = 1;
    par->models.model[0].perturbations.perturbation[0].flip_vertical = 1;
    par->models.model[0].perturbations.perturbation[0].num_synthetic_samples = 1;
    CHECK_FN_GOTO(sb_project_set_par(*retina, par))
    CHECK_FN_GOTO(sb_project_save(*retina, "../images/wheels." SB_SOLUTION_EXT, SB_PROJECT_MODE_DETECTION_AND_SVL))
FnExit:
    sb_par_destroy(&par);
    return err;
}
 
sb_t_err set_samples(SB_HANDLE retina)
{
    sb_t_err err = SB_ERR_NONE;
    SB_HANDLE image_info = NULL;
    sb_t_sample sample;
 
    memset(&sample, 0, sizeof(sample));
    sample.type = SB_OBJ_TEST;  
    sample.classify_mode = SB_SAMPLE_REQUIRED;  
    sample.scale = 1.0f;    
    strcpy(sample.model_name, MODEL_NAME);
    // first image
    CHECK_FN_GOTO(sb_image_info_load(&image_info, "../images/001.ppm", retina))
    // 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))
    sample.centre = sb_point(78, 141);  
    CHECK_FN_GOTO(sb_get_uuid(sample.uuid, sizeof(sample.uuid)))
    CHECK_FN_GOTO(sb_image_info_add_sample(image_info, &sample))
    sample.centre = sb_point(234, 138);
    CHECK_FN_GOTO(sb_get_uuid(sample.uuid, sizeof(sample.uuid)))
    CHECK_FN_GOTO(sb_image_info_add_sample(image_info, &sample))
    sample.centre = sb_point(537, 146);
    CHECK_FN_GOTO(sb_get_uuid(sample.uuid, sizeof(sample.uuid)))
    CHECK_FN_GOTO(sb_image_info_add_sample(image_info, &sample))
    CHECK_FN_GOTO(sb_image_info_save(image_info))
    CHECK_FN_GOTO(sb_image_info_destroy(&image_info))
    // second image
    CHECK_FN_GOTO(sb_image_info_load(&image_info, "../images/002.ppm", retina))
    // 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))
    sample.centre = sb_point(333, 143);
    CHECK_FN_GOTO(sb_get_uuid(sample.uuid, sizeof(sample.uuid)))
    CHECK_FN_GOTO(sb_image_info_add_sample(image_info, &sample))
    CHECK_FN_GOTO(sb_image_info_save(image_info))
    CHECK_FN_GOTO(sb_image_info_destroy(&image_info))
    // third image
    CHECK_FN_GOTO(sb_image_info_load(&image_info, "../images/003.ppm", retina))
    // 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));
    sample.centre = sb_point(511, 136);
    CHECK_FN_GOTO(sb_get_uuid(sample.uuid, sizeof(sample.uuid)))
    CHECK_FN_GOTO(sb_image_info_add_sample(image_info, &sample))
    CHECK_FN_GOTO(sb_image_info_save(image_info))
    CHECK_FN_GOTO(sb_image_info_destroy(&image_info))
FnExit:
    sb_image_info_destroy(&image_info);
    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 = NULL;
    sb_t_solution_info* solution = NULL;
 
    CHECK_FN_GOTO(sb_solution_get_info("../images/wheels." SB_SOLUTION_EXT, &solution));
    CHECK_FN_GOTO(sb_project_load(&svl, "../images/wheels." 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, "../images/");
    par->svl.sl.goodness_target = 0.8f;
    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_par_destroy(&par);
    sb_project_destroy(&svl);
    sb_solution_destroy_info(&solution);
    return err;
}