mirror of https://github.com/alibaba/MNN.git
33 lines
645 B
C++
33 lines
645 B
C++
|
//
|
||
|
|
// revertMNNModel.hpp
|
||
|
|
// MNN
|
||
|
|
//
|
||
|
|
// Created by MNN on 2019/01/31.
|
||
|
|
// Copyright © 2018, Alibaba Group Holding Limited
|
||
|
|
//
|
||
|
|
|
||
|
|
#ifndef REVERTMNNMODEL_HPP
|
||
|
|
#define REVERTMNNMODEL_HPP
|
||
|
|
|
||
|
|
#include "converter/source/IR/MNN_generated.h"
|
||
|
|
|
||
|
|
class Revert {
|
||
|
|
public:
|
||
|
|
Revert(const char* originalModelFileName);
|
||
|
|
~Revert();
|
||
|
|
void* getBuffer() const;
|
||
|
|
const size_t getBufferSize() const;
|
||
|
|
void initialize();
|
||
|
|
static float getRandValue();
|
||
|
|
|
||
|
|
private:
|
||
|
|
Revert();
|
||
|
|
std::unique_ptr<MNN::NetT> mMNNNet;
|
||
|
|
size_t mBufferSize;
|
||
|
|
std::shared_ptr<uint8_t> mBuffer;
|
||
|
|
void randStart();
|
||
|
|
void packMNNNet();
|
||
|
|
};
|
||
|
|
|
||
|
|
#endif // REVERTMNNMODEL_HPP
|