diff --git a/doc/development_doc.md b/doc/development_doc.md index ee7d2f69dbad87935afec0e2b1eead71f313f611..18a1974576d8e488fe7f4e70cad18ae088b794a9 100644 --- a/doc/development_doc.md +++ b/doc/development_doc.md @@ -4,8 +4,6 @@ ## 编译 -### 一. 使用 build.sh 编译 - ```sh # 在 paddle-mobile 目录下: @@ -21,9 +19,7 @@ cd ../build/release/ios/build ``` -### 二. 集成 - -#### 如使用 oc 接口 +## 集成 ``` 将上一步生成的: @@ -32,7 +28,11 @@ libpaddle-mobile.a /src/ios_io/ 下的 PaddleMobile.h ``` -拖入工程, 接口如下: +拖入工程 + +#### oc 接口 + +接口如下: ``` /* diff --git a/src/ios_io/PaddleMobile.h b/src/ios_io/PaddleMobile.h index 863c0d004440dc6098eb7dc1ed490fde20f237c9..091a892ab9be086c6842eb52ef6266f7fba9ce96 100644 --- a/src/ios_io/PaddleMobile.h +++ b/src/ios_io/PaddleMobile.h @@ -19,10 +19,29 @@ @interface PaddleMobile : NSObject +/* + 创建对象 +*/ - (instancetype)init; + +/* + load 模型, 开辟内存 +*/ - (BOOL)load:(NSString *)modelPath andWeightsPath:(NSString *)weighsPath; + +/* + 进行预测, means 和 scale 为训练模型时的预处理参数, 如训练时没有做这些预处理则直接使用 predict +*/ - (NSArray *)predict:(CGImageRef)image dim:(NSArray *)dim means:(NSArray *)means scale:(float)scale; + +/* + 进行预测 +*/ - (NSArray *)predict:(CGImageRef)image dim:(NSArray *)dim; + +/* + 清理内存 +*/ - (void)clear; @end diff --git a/tools/build.sh b/tools/build.sh index f1efd24ea548c0ca0cba14e82e018100d3f8b24d..e1a47e64e6ea87dd9d70969214412fc0c2fa2d67 100755 --- a/tools/build.sh +++ b/tools/build.sh @@ -119,6 +119,7 @@ build_for_ios() { fi cd "${BUILD_DIR}" make -j 8 + cp ../../../src/ios_io/PaddleMobile.h ./build/PaddleMobile.h cd ./build # 生成符号表 ranlib *.a