DeepSurfaceDetect.cs

using System;
using sb_cs;
using Tutorials_Common;
 
namespace Tutorial14_DeepSurfaceDetect
{
    class DeepSurfaceDetect
    {
        static SbError execute()
        {
            SbProject prj = null;
            SbImage img = null;
            SbRoi roi = null;
            SbRes res = null;
            SbPar par = null;
            SbError err = SbError.SB_ERR_NONE;
 
            Console.WriteLine("Sb Init");
            err = Sb.Init("../../../sb.lic");
            if (err != SbError.SB_ERR_NONE)
            {
                Console.WriteLine("Sb.Init failed with error " + err);
                return err;
            }
 
            Console.WriteLine("Wait for license...");
            err = Common.WaitLicense();
            if (err != SbError.SB_ERR_NONE)
            {
                Console.WriteLine("SbLicense.WaitLicense failed with error " + err);
                goto FnExit;
            }
 
            err = Sb.InitDl("../../../../../win_x64/dl_framework", SbDlLibraryType.SB_DL_LIBRARY_PYTORCH);
            if (err != SbError.SB_ERR_NONE)
            {
                Console.WriteLine("Sb.InitDl failed with error " + err);
                return err;
            }
 
            String solution_file = "../../solution/deep_surface_tutorial." + SbDefines.SbSolutionExt;
            SbSolutionInfo solutionInfo = SbSolution.GetInfo(solution_file);
            if (solutionInfo == null || solutionInfo.Error() != SbError.SB_ERR_NONE)
            {
                err = solutionInfo == null ? SbError.SB_ERR_INSUFFICIENT_FREE_MEMORY : solutionInfo.Error();
                Console.WriteLine("SbSolution.GetInfo failed");
                goto FnExit;
            }
            else if (solutionInfo.projects.Length == 0)
            {
                err = SbError.SB_ERR_PROJECT_NOT_FOUND;
                Console.WriteLine("SbSolution.GetInfo no projects found");
                goto FnExit;
            }
            Console.WriteLine("GetInfo, found " + solutionInfo.projects.Length + " project/s");
 
            prj = SbProject.Load(solution_file, solutionInfo.projects[0].uuid, SbProjectMode.SB_PROJECT_MODE_DETECTION_ONLY);
            if (prj == null || prj.Error() != SbError.SB_ERR_NONE)
            {
                err = prj == null ? SbError.SB_ERR_INSUFFICIENT_FREE_MEMORY : prj.Error();
                Console.WriteLine("SbProject.Load failed with error " + err);
                goto FnExit;
            }
 
            par = prj.GetPar();
            // blob parameters
            par.surface_blob_analysis = 1;
            par.blob_par.area.min = 10;
            par.blob_par.merge_area.min = 50;
            par.blob_par.blob_rle = 1;
            par.blob_par.blob_contour = 1;
            par.blob_par.merge_distance = 20;
            //par.surface_blob_analysis = 1;
            // devices parameters
            par.devices.type = SbDeviceType.SB_DEVICE_GPU_NVIDIA;  // set GPU device
            par.devices.id[0] = 0;                          // set the first GPU founded on the machine (with index 0)
            err = prj.SetPar(par);
            if (err != SbError.SB_ERR_NONE || prj.Error() != SbError.SB_ERR_NONE)
            {
                Console.WriteLine("SbProject.SetPar failed with error " + err);
                goto FnExit;
            }
 
            SbFolder folder = SbFolder.Load("../../dataset", "png", true, 0);
            if (folder != null)
            {
                for (int i = 0; i < folder.files.Length; i++)
                {
                    img = SbImage.Load(folder.files[i]);
                    if (img == null || img.Error() != SbError.SB_ERR_NONE)
                    {
                        Console.WriteLine("SbImage.Load " + folder.files[i] + " failed");
                        err = img == null ? SbError.SB_ERR_INTERNAL : img.Error();
                        break;
                    }
 
                    roi = SbRoi.Create(img.Width(), img.Height());
                    if (roi == null || roi.Error() != SbError.SB_ERR_NONE)
                    {
                        Console.WriteLine("SbRoi.Create failed");
                        err = roi == null ? SbError.SB_ERR_INTERNAL : roi.Error();
                        goto FnExit;
                    }
 
                    err = roi.SetRect(255, new SbRect(0, 0, img.Width(), img.Height()), false);
                    if (err != SbError.SB_ERR_NONE)
                    {
                        Console.WriteLine("SbRoi.SetRect failed with error " + err);
                        goto FnExit;
                    }
 
                    err = prj.Detection(img, roi);
                    if (err != SbError.SB_ERR_NONE)
                    {
                        Console.WriteLine("SbProject.Detection failed with error " + err);
                        break;
                    }
 
                    res = prj.GetRes(false);
                    if (res == null || prj.Error() != SbError.SB_ERR_NONE)
                    {
                        err = res == null ? SbError.SB_ERR_INSUFFICIENT_FREE_MEMORY : prj.Error();
                        Console.WriteLine("SbProject.GetRes failed with error " + err);
                        break;
                    }
 
                    Console.WriteLine("image: " + folder.files[i]);
                    Console.WriteLine("\t" + res.global.surface.blobs.Length + " defect blobs found:");
                    for(int j = 0; j < res.global.surface.blobs.Length; j++)
                    {
                        Console.WriteLine("\t\tul=(" + res.global.surface.blobs[j].rect.x + "," + res.global.surface.blobs[j].rect.y +
                            "), size=(" + res.global.surface.blobs[j].rect.width + "," + res.global.surface.blobs[j].rect.height +
                            "), area=" + res.global.surface.blobs[j].area);
                    }
 
                    if (img != null)
                    {
                        img.Dispose();
                        img = null;
                    }
                    if (roi != null)
                    {
                        roi.Dispose();
                        roi = null;
                    }
                    if (res != null)
                    {
                        res.Dispose();
                        res = null;
                    }
                }
            }
 
        FnExit:
            if (prj != null)
                prj.Dispose();
            if (img != null)
                img.Dispose();
            if (roi != null)
                roi.Dispose();
            if (par != null)
                par.Dispose();
            if (res != null)
                res.Dispose();
 
            Console.WriteLine("Release SqueezeBrains library");
            Sb.Release();
 
            Console.WriteLine("Press ENTER to terminate");
            Console.ReadKey();
 
            return err;
        }
 
        static void Main(string[] args)
        {
            execute();
        }
    }
}