yoloseg/net_message_trans.h

#pragma once

#include "proto/def.grpc.pb.h"
#include <opencv2/opencv.hpp>
#include "tool/log.hpp"

class NetMessageTrans {
public:
    static void lableMat2Proto(int lable, cv::Mat &mat, JetStream::LabelImage &image) {
        image.set_label(lable);
        image.mutable_ir()->set_width(mat.cols);
        image.mutable_ir()->set_height(mat.rows);
        image.mutable_ir()->set_channel(mat.type());
        image.mutable_ir()->set_data(mat.data, mat.dataend - mat.datastart);
        // LOG(INFO) << "mat.dataend - mat.datastart = " << mat.dataend - mat.datastart;
    }

    static void Proto2lableMat(JetStream::LabelImage &image, int &lable, cv::Mat &mat) {
        lable = image.label();
        mat = cv::Mat(image.ir().height(), image.ir().width(), image.ir().channel(), (void*)image.ir().data().data());
    }

    static void getSegBoxData() {

    }

    static void data2SegBox() {

    }
};