retina_multi_project.c

Go to the documentation of this file.

 
#include "../common/common.h"
 
int main(int argc, char* argv[])
{
    sb_t_err err = SB_ERR_NONE;
    sb_t_info* sb_info = NULL;
    char path[256] = "";
    SB_HANDLE slice = NULL;
    SB_HANDLE hole = NULL;
    sb_t_folder* folder = NULL;
    sb_t_roi* roi = NULL;
    sb_t_image* image = NULL;
    sb_t_res* res_slice = NULL;
    sb_t_res* res_hole = NULL;
    sb_t_par* par_slice = NULL;
    sb_t_par* par_hole = NULL;
    sb_t_solution_info* solution = NULL;
    const char solution_file[] = "solution/tutorial_7_retina_multi_project." SB_SOLUTION_EXT;
    char str[8192];
    int i;
 
    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_file, &solution));
    CHECK_FN_GOTO(sb_project_load(&slice, solution_file, solution->projects.info[0].uuid, SB_PROJECT_MODE_DETECTION_ONLY))
    CHECK_FN_GOTO(sb_project_get_par(slice, &par_slice))
    CHECK_FN_GOTO(sb_par_format(par_slice, str, sizeof(str)))
    printf(str);
    CHECK_FN_GOTO(sb_project_load(&hole, solution_file, solution->projects.info[1].uuid, SB_PROJECT_MODE_DETECTION_ONLY))
    CHECK_FN_GOTO(sb_project_get_par(hole, &par_hole))
    CHECK_FN_GOTO(sb_par_format(par_hole, str, sizeof(str)))
    printf(str);
 
    CHECK_FN_GOTO(sb_folder_load(&folder, "dataset", "png", 1, 0))
    for (i = 0; i < folder->n; ++i)
    {
        int j;
        // Find the slices of gruyere.
        printf("\nimage=\"%s\"\n", folder->file[i].fname);
        CHECK_FN_GOTO(sb_image_load(&image, folder->file[i].name))
        // Create the roi
        if (roi && roi->img && (roi->img->width != image->width || roi->img->height != image->height))
            sb_roi_destroy(&roi);
        if (!roi)
        {
            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), 1))
        // Elaborate the image.
        CHECK_FN_GOTO(sb_project_detection(slice, image, roi, NULL, NULL))
        // Get the results.
        CHECK_FN_GOTO(sb_project_get_res(slice, &res_slice, 0))
        printf("    slice detection time=%1.3fms, %d slices found\n", res_slice->time_us / 1000.0f, res_slice->samples.size);
        for (j = 0; j < res_slice->samples.size; ++j)
        {
            // Find the holes in the slice.
            int k;
            // Reset the previous ROI and set a rectangular ROI 230x130 center on the slice.
            CHECK_FN_GOTO(sb_roi_set_rect(roi, 255, sb_rect(res_slice->samples.sample[j].centre.x - 115, res_slice->samples.sample[j].centre.y - 65, 230, 130), 1))
            // Elaborate the image
            CHECK_FN_GOTO(sb_project_detection(hole, image, roi, NULL, NULL))
            // Get the results.
            CHECK_FN_GOTO(sb_project_get_res(hole, &res_hole, 0))
            printf("        slice %d hole detection time=%1.3fms, %d holes found\n", j, res_hole->time_us/1000.0f, res_hole->samples.size);
            // Print the results
            for (k = 0; k < res_hole->samples.size; ++k)
                printf("            centre=(%d,%d), weight=%1.3f\n", res_hole->samples.sample[k].centre.x, res_hole->samples.sample[k].centre.y, res_hole->samples.sample[k].weight);
            CHECK_FN_GOTO(sb_res_destroy(&res_hole))
        }
        // Destroy the handles.
        CHECK_FN_GOTO(sb_image_destroy(&image))
        CHECK_FN_GOTO(sb_res_destroy(&res_slice))
    }
 
FnExit:
    sb_destroy_info(&sb_info);
    sb_project_destroy(&slice);
    sb_project_destroy(&hole);
    sb_image_destroy(&image);
    sb_roi_destroy(&roi);
    sb_par_destroy(&par_slice);
    sb_par_destroy(&par_hole);
    sb_res_destroy(&res_slice);
    sb_res_destroy(&res_hole);
    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;
}