Caffe2 - C++ API
A deep learning, cross platform ML framework
ios_caffe.h
1 
18 #ifdef __cplusplus
19 
20 #include <string>
21 #include <vector>
22 #include "caffe2/core/predictor.h"
23 #include "caffe2/mobile/contrib/ios/ios_caffe_defines.h"
24 #include "caffe2/mobile/contrib/ios/ios_caffe_predictor.h"
25 
26 extern "C" {
27 
28 IOS_CAFFE_EXPORT Caffe2IOSPredictor* MakeCaffe2Predictor(const std::string& init_net_str,
29  const std::string& predict_net_str,
30  bool disableMultithreadProcessing,
31  bool allowMetalOperators,
32  std::string& errorMessage);
33 IOS_CAFFE_EXPORT void GenerateStylizedImage(std::vector<float>& originalImage,
34  const std::string& init_net_str,
35  const std::string& predict_net_str,
36  int height,
37  int width,
38  std::vector<float>& dataOut);
39 }
40 
41 #endif