File config.h

File List > config.h

Go to the documentation of this file

#pragma once
#ifdef DLL_EXPORT
#define DLL_API __declspec(dllexport)
#else
#define DLL_API __declspec(dllimport)
#endif

#include "type.h"
#include <string>

namespace inspection
{
    struct DomainConfig
    {
        string crop = "normal";
        int windowSize = 720;
        int stride = 360;
        float densityThreshold = 0.0;
        float angle1 = 50, angle2 = 0, angle3 = 0;
        float keypointMergeDistance = 10.0;
        int minimumPixels = 100;
        int imageWidth = 2100;
        int imageHeight = 800;
        int numOfClass = 3;
        string direct = "top";
        string clusterType = "linear";
    };

    struct ModelConfig
    {
        string device = "0";
        TaskType task = TaskType::SEMENTIC_SEGMENTATION;
        string weightPath = "";
        DomainConfig domainConfig;
    };
}

//DomainConfig GetDomainConfig(CellType cellType, string configPath = string(""));
//ModelConfig CreateModelConfig(string device, TaskType task, CellType cellType, string weightPath)