1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
|
#ifdef __cplusplus
#include <string>
#include <vector>
#include "caffe2/mobile/contrib/ios/ios_caffe_defines.h"
#include "caffe2/mobile/contrib/ios/ios_caffe_predictor.h"
#include "caffe2/predictor/predictor.h"
extern "C" {
IOS_CAFFE_EXPORT Caffe2IOSPredictor* MakeCaffe2Predictor(const std::string& init_net_str,
const std::string& predict_net_str,
bool disableMultithreadProcessing,
bool allowMetalOperators,
std::string& errorMessage);
IOS_CAFFE_EXPORT void GenerateStylizedImage(std::vector<float>& originalImage,
const std::string& init_net_str,
const std::string& predict_net_str,
int height,
int width,
std::vector<float>& dataOut);
}
#endif
|
