Initial commit

inc/colorize_process.h

@@ -0,0 +1,63 @@
+/**
+* Copyright 2020 Huawei Technologies Co., Ltd
+*
+* Licensed under the Apache License, Version 2.0 (the "License");
+* you may not use this file except in compliance with the License.
+* You may obtain a copy of the License at
+
+* http://www.apache.org/licenses/LICENSE-2.0
+
+* Unless required by applicable law or agreed to in writing, software
+* distributed under the License is distributed on an "AS IS" BASIS,
+* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+* See the License for the specific language governing permissions and
+* limitations under the License.
+
+* File sample_process.h
+* Description: handle acl resource
+*/
+#pragma once
+#include "utils.h"
+#include "acl/acl.h"
+#include "model_process.h"
+#include <memory>
+
+using namespace std;
+
+/**
+* ColorizeProcess
+*/
+class ColorizeProcess {
+public:
+    ColorizeProcess(const char* modelPath, uint32_t modelWidth, uint32_t modelHeight);
+    ~ColorizeProcess();
+
+    Result Init();
+    Result Preprocess(const string& imageFile);
+    Result Inference(aclmdlDataset*& inferenceOutput);
+    Result Postprocess(const string& origImageFile,
+                       aclmdlDataset* modelOutput);
+private:
+    Result InitResource();
+    Result InitModel(const char* omModelPath);
+
+    void* GetInferenceOutputItem(uint32_t& itemDataSize,
+                                 aclmdlDataset* inferenceOutput);
+    void SaveImage(const string& origImageFile, cv::Mat& image);
+    void DestroyResource();
+
+private:
+    int32_t deviceId_;
+
+    ModelProcess model_;
+
+    const char* modelPath_;
+    uint32_t modelWidth_;
+    uint32_t modelHeight_;
+    uint32_t inputDataSize_;
+    void*    inputBuf_;
+    aclrtRunMode runMode_;
+
+    bool isInited_;
+};
+