"examples/community/pipeline_flux_rf_inversion.py" did not exist on "4e74206b0c443f9d272401f397d781d9d0630073"
Unet.h 557 Bytes
Newer Older
1
2
3
4
5
#ifndef __UNET_H__
#define __UNET_H__

#include <migraphx/program.hpp>

liucong's avatar
liucong committed
6
#include <CommonDefinition.h>
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35

namespace migraphxSamples
{

class  Unet
{
public:
    Unet();
    
    ~Unet();

    ErrorCode Initialize(InitializationParameterOfSegmentation initParamOfSegmentationUnet);

    ErrorCode Segmentation(const cv::Mat &srcImage, cv::Mat &maskImage);

private:
    cv::FileStorage configurationFile;

    migraphx::program net;
    cv::Size inputSize;
    std::string inputName;
    migraphx::shape inputShape;
    float scale;
         
};

}
    
#endif