1// easydl-stitch-ios/EasyDLStitch/EasyDLStitchParams.h
2
3#define kThreshold 75 // 判断重合的阈值,0~100之间
4#define kStrategy "phash" // 重合算法,类型：["ahash","phash","dhash"]
5#define kThetaZ 60 // 手机倾斜Z轴角度阈值
6#define kThetaXY 20 // 手机倾斜XY轴角度阈值
7#define kOverLapControl true // 是否与遮罩重合才可以拍照
8#define kOrientationControl true // 是否手机持握方向符合要求才可以拍照
9#define kSkipFrames 3 // 跳过视频帧比对的数量，比如3为每3帧比对一次
10#define kBrightnessLow -3 // 光线强度阈值，-99~99之间
11#define kBrightnessHigh 5 // 光线强度阈值，-99~99之间
12#define kBrightnessControl false // 是否光线符合要求才可以拍照

1// easydl-stitch-ios/ViewController/StitchViewController.m
2
3static int frameInterval = 1;//截帧间隔（秒）

1UIAlertController *actionSheet = [UIAlertController alertControllerWithTitle:@"选择对象(视频长度不能超过60s)" message:nil preferredStyle:UIAlertControllerStyleActionSheet];

1EasyDL-Image-Stitching-iOS
2|- LIB
3    |- include
4        |- montage_algo/ // 手机端实时拼接功能库头文件
5        |- EasyDL/       // 手机端实时拼接模糊图像检测引擎头文件
6        |- paddlelite/   // 手机端实时拼接模糊图像检测引擎头文件
7    |- libs
8        |- opencv2.framework 		        // OpenCV库
9        |- libmontage_algo.a 		        // 手机端实时拼接功能库
10        |- libEasyDL.a                  // 手机端实时拼接模糊图像检测引擎
11        |- libpaddle_api_full_bundled.a // 手机端实时拼接模糊图像检测引擎
12|- EasyDLStitch
13    |- images/              // 资源文件
14    |- easydl-stitch-ios/   // Demo工程文件
15    |- RES/
16        |- conf.json       // API配置文件
17        |- fuzzy_model/    // 模糊图像检测模型

1{
2  "ak": "Mz0zhObvEZ6lnG1K3renXXXX", // API Key的值
3
4  "sk": "188fRHYvLPmlPrNCDpBnkhL3ydXXXXX", // Secret Key的值
5
6  "apiUrl": "https://aip.baidubce.com/rpc/2.0/ai_custom_retail/v1/detection/XXXX"  // 定制商品检测服务API
7}

1- (void)createSpliceTaskWithConfig:(NSDictionary *)config successHandler:(SuccessBlock)successHandler failHandler:(FailureBlock)failHandler;

1- (void)uploadImageWithConfig:(NSDictionary *)config successHandler:(SuccessBlock)successHandler failHandler:(FailureBlock)failHandler;

1-(void)startSpliceTaskWithConfig:(NSDictionary *)config successHandler:(SuccessBlock)successHandler failHandler:(FailureBlock)failHandler;

1-(void)queryTaskResultWithConfig:(NSDictionary *)config successHandler:(SuccessBlock)successHandler failHandler:(FailureBlock)failHandler;

1-(void)listTaskWithConfig:(NSDictionary *)config successHandler:(SuccessBlock)successHandler failHandler:(FailureBlock)failHandler;

1-(void)terminateTaskWithConfig:(NSDictionary *)config successHandler:(SuccessBlock)successHandler failHandler:(FailureBlock)failHandler;

1- (void)prepare;
2
3// 调用示例
4[[EasyDLMBStitchApiService sharedService] prepare];

1- (void)compareImage:(UIImage *)image firstFrame:(BOOL)isFirstFrame completionHandler:(CompletionHandler)completionHandler;
2
3// 调用示例
4[[EasyDLMBStitchApiService sharedService] compareImage:image firstFrame:_firstFrameCompare completionHandler:^(id responseObject, NSError *error) {
5    if (error) {
6        NSLog(@"%@", error.localizedDescription);
7    } else {
8      	// 回调结果
9        CompareResult *compareResult = responseObject;
10      	// do something
11    }
12}];

1// montage_algo/EasyDLStitchAlgo.h
2
3// 当前图片相对上一张参与拼接的图片的方位
4@property(nonatomic) ImageDirection direction;
5/**
6 * 是否需要判断方向，如当拍摄完图片过近时，direction可能由于两图过于相似而不可靠，这种情况不需要判断方向，即该值=false
7 * 一般direction不可靠时，该值=false
8 * 对比的两张图是第一次对比时，该值=false
9 * 当该值=true时，请在调用实时拼接API前确认方法是否合法，否则可能导致拼接失败
10 */
11@property(nonatomic, getter = needCheckDirection) BOOL checkDirection;
12// 对比结果中的方位是否合法，非法的方位将无法完成拼接
13@property(nonatomic) BOOL directionValid;
14// 两张图片重叠部分的点位
15@property(nonatomic, retain) NSArray<NSValue *> *points;
16// 两张图片重叠状态
17@property(nonatomic) OverlapStatus overlapStatus;
18// 最后一次参与拼接的图片序号，从1开始
19@property(nonatomic) int lastImgIndex;

1switch (compareResult.overlapStatus) {
2    case OverlapStatus_Correct:
3        if (compareResult.needCheckDirection && !compareResult.directionValid) {
4            // 非法，当前参与对比的图片方位不正确，无法拼接
5        } else {
6            // 合法
7        }
8        break;
9    case OverlapStatus_TooFar:
10        // 非法，两张图重叠度过低
11        break;
12    case OverlapStatus_TooClose:
13        // 非法，两张图重叠度过高
14        break;
15}

1- (void)stitchImageWithCompareResult:(CompareResult *)compareResult completionHandler:(CompletionHandler)completionHandler;
2
3// 调用示例
4[[EasyDLMBStitchApiService sharedService] stitchImageWithCompareResult:_latestCompareResult completionHandler:^(id responseObject, NSError *error) {
5    if (error) {
6        NSLog(@"%@", error.localizedDescription);
7    } else {
8      	// 回调结果
9        StitchResult *stitchResult = responseObject;
10      	// do something
11    }
12}];
13
14// 保存now.jpg示例
15NSURL *workDirUrl = [EasyDLMBStitchApiService sharedService].currentTask.workDir;
16/* 小图为拼接，大图为获得更好的商品检测效果 */
17[EasyDLFileManager saveImage:image toUrl:[workDirUrl URLByAppendingPathComponent:FILENAME_IMAGE_NOW] andResizeTo:CGSizeMake(648, 864)];
18// SDK默认使用保存的一系列`{index}.jpg`调用商品检测API并取得结果。
19// 由于建议减小该系列图片尺寸以获得更优的拼接效率，但同时更小尺寸的图片对商品检测精度有一定影响，因此为提高精度，建议同时保存最佳尺寸的图片用于上传云端。
20[EasyDLFileManager saveImageForBestInfer:image toUrl:[workDirUrl URLByAppendingPathComponent:[NSString stringWithFormat:@"%@/%@", DIR_NAME_FULL_IMAGE, FILENAME_IMAGE_NOW]]];

1// montage_algo/EasyDLStitchAlgo.h
2
3// 拼接错误码，0表示成功，其他为错误
4@property(nonatomic) int errCode;
5// 最近一张参与拼接的图片的序号
6@property(nonatomic) int latestImgIndex;
7// 缩略拼接图路径
8@property(nonatomic, retain) NSURL *thumbnailUrl;
9// 完整拼接图路径
10@property(nonatomic, retain) NSURL *fullImageUrl;

1// easydl-montage-ios/EasyDLStitch/EasyDLMBStitchApiService.h
2
3@protocol EasyDLMBAPIStitchDelegate <NSObject>
4- (void)onStitchThumbnailGenerated:(NSURL *)thumbnailURL;
5- (void)onStitchFullImageGenerated:(NSURL *)fullImageURL;
6@end
7  
8// 配置示例
9[EasyDLMBStitchApiService sharedService].stitchDelegate = self;

1- (void)mergeDetectedResultsWithCompletionHandler:(CompletionHandler)completionHandler;
2
3// 调用示例
4[[EasyDLMBStitchApiService sharedService] mergeDetectedResultsWithCompletionHandler:^(id responseObject, NSError *error) {
5    if (error) {
6        NSLog(@"%@", error.localizedDescription);
7    } else {
8      	// 回调结果
9        MergeResult *mergeResult = responseObject;
10      	// do something
11    }
12}];

1// montage_algo/EasyDLStitchAlgo.h
2
3// 错误码，0表示成功，其他为错误
4@property(nonatomic) int errCode;
5// 商品检测并去重合并后的结果
6@property(nonatomic, retain) NSDictionary *correctSKUDict;

1// easydl-montage-ios/EasyDLStitch/EasyDLMBStitchApiService.h
2
3@protocol EasyDLMBAPIMergeDelegate <NSObject>
4- (void)onMergeResultsStarted:(int)totalImageCount;
5- (void)onMergeProgressUpdated:(int)totalImageCount completedCount:(int)completedCount;
6@end
7  
8// 配置示例
9[EasyDLMBStitchApiService sharedService].mergeDelegate = self;

1- (void)undoLastTakenImageWithCompletionHandler:(CompletionHandler)completionHandler;
2
3// 调用示例
4[[EasyDLMBStitchApiService sharedService] undoLastTakenImageWithCompletionHandler:^(id responseObject, NSError *error) {
5    if (error) {
6        NSLog(@"%@", error.localizedDescription);
7    } else {
8      	// 回调结果，包含撤销后，当前最后一次参与拼接的图片信息
9        ImageInfo *lastImageInfo = responseObject;
10      	// do something
11    }
12}];

1// easydl-montage-ios/EasyDLStitch/EasyDLMBStitchApiService.h
2
3// 图片序号
4@property(nonatomic) int index;
5// 图片列坐标
6@property(nonatomic) int x;
7// 图片行坐标
8@property(nonatomic) int y;

1// 模型初始化
2- (NSError *)modelInit;
3// 推理图像判断是否模糊
4- (void)inferImage:(UIImage *)image completionHandler:(void (^)(BOOL fuzzy, NSError *error))completionHandler;
5
6// 调用示例
7FuzzyModelProxy *fuzzyModelProxy = [[FuzzyModelProxy alloc] init];
8[fuzzyModelProxy modelInit];
9if (fuzzyModelProxy && fuzzyModelProxy.engineActive) {
10    [fuzzyModelProxy inferImage:image completionHandler:^(BOOL fuzzy, NSError *error) {
11        if (!error && !fuzzy) {
12            // 图像非模糊
13        } else {
14            // 图像模糊或推理失败
15        }
16    }];
17}

1// easydl-montage-ios/EasyDLStitch/EasyDLMBStitchApiService.h
2
3/**
4 * 最小拼接引导置信度
5 */
6@property(nonatomic) CGFloat minIOUThreshold;
7/**
8 * 最大拼接引导置信度
9 */
10@property(nonatomic) CGFloat maxIOUThreshold;
11/**
12 * 去重置信度
13 */
14@property(nonatomic) CGFloat nmsIOUThreshold;
15/**
16 * 商品检测API重试次数
17 */
18@property(nonatomic) int detectRetryTimes;
19/**
20 * 商品检测API最大并发数
21 */
22@property(nonatomic) int maxQPS;

1[[EasyDLDoorAPIService sharedService] startLocation];

1// easydl-montage-ios/EasyDLDoor/EasyDLDoorAPIService.h
2
3// 开始门脸文字识别
4[[EasyDLDoorAPIService sharedService] detectDoorImage:image];
5// 获取门脸识别结果
6[EasyDLDoorAPIService sharedService].blockNSDictionary = ^(NSDictionary * _Nonnull blockNSDictionary, NSError * _Nonnull error) {
7    if(blockNSDictionary != nil && error == nil) {
8      // 门脸图片识别成功
9      ..............
10      // 校验门店结果
11      [EasyDLStitchAlgo checkDoorData:documentsDirectory ocrInfo:ocrJson];
12    }
13}

1// easydl-montage-ios/EasyDLDoor/CheckImageConfig.h
2
3- (BOOL)getPirateImageCheck;
4
5// 调用示例
6/**
7 * 窜拍开关默认开启
8 */
9[CheckImageConfig sharedService].pirateImageCheck = true;
10// 获取开关状态
11[[CheckImageConfig sharedService] getPirateImageCheck];