From ad2895079465fb571b743f09d51a3471e64289a6 Mon Sep 17 00:00:00 2001
From: liuruilong <liuruilong@baidu.com>
Date: Fri, 6 Jul 2018 17:21:40 +0800
Subject: [PATCH] update ios compile

---
 doc/development_doc.md    | 12 ++++++------
 src/ios_io/PaddleMobile.h | 19 +++++++++++++++++++
 tools/build.sh            |  1 +
 3 files changed, 26 insertions(+), 6 deletions(-)

diff --git a/doc/development_doc.md b/doc/development_doc.md
index ee7d2f69db..18a1974576 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 863c0d0044..091a892ab9 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<NSNumber *> *)dim means:(NSArray<NSNumber *> *)means scale:(float)scale;
+
+/*
+	进行预测
+*/
 - (NSArray *)predict:(CGImageRef)image dim:(NSArray<NSNumber *> *)dim;
+
+/*
+	清理内存
+*/
 - (void)clear;
 
 @end
diff --git a/tools/build.sh b/tools/build.sh
index f1efd24ea5..e1a47e64e6 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
-- 
GitLab