DeepCortexDetect.cs

using System;
using sb_cs;
using Tutorials_Common;
 
namespace Tutorial12_DeepCortexDetect
{
    class DeepCortexDetect
    {
        static SbError execute()
        {
            SbProject prj = null;
            SbImage img = 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");
            if (err != SbError.SB_ERR_NONE)
            {
                Console.WriteLine("Sb.InitDl failed with error " + err);
                return err;
            }
 
            String solution_file = "../../solution/deep_cortex_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;
            }
 
            SbPar par = prj.GetPar();
            par.devices.type = SbDeviceType.SB_DEVICE_GPU;  // 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;
                    }
 
                    err = prj.Detection(img, null);
                    if (err != SbError.SB_ERR_NONE)
                    {
                        Console.WriteLine("SbProject.Detection failed with error " + err);
                        break;
                    }
 
                    SbRes 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("\tmodel=" + res.samples[0].model_name + ",\tweight=" + res.samples[0].weight);
 
                    if (img != null)
                    {
                        img.Dispose();
                        img = null;
                    }
                }
            }
 
        FnExit:
            if (prj != null)
                prj.Dispose();
            if (img != null)
                img.Dispose();
 
            Console.WriteLine("Release SqueezeBrains library");
            Sb.Release();
 
            Console.WriteLine("Press ENTER to terminate");
            Console.ReadKey();
 
            return err;
        }
 
        static void Main(string[] args)
        {
            execute();
        }
    }
}