cassdlcm/mediapipe
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix setting BaseOptions delegate type error

Browse Source 
fix setting BaseOptions delegate type error
This commit is contained in:
zhuwen 2023-09-08 15:36:36 +08:00
parent cbbbc9e49e
commit c5aa96c020
3 changed files with 12 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
mediapipe/tasks/ios/core/sources/MPPBaseOptions.h
View File

@ -17,8 +17,7 @@
typedef enum : NSUInteger { typedef enum : NSUInteger {
MPPBaseOptionsDelegateType_Unknown, MPPBaseOptionsDelegateType_Unknown,
MPPBaseOptionsDelegateType_Cpu, MPPBaseOptionsDelegateType_Cpu,
MPPBaseOptionsDelegateType_Gpu, MPPBaseOptionsDelegateType_Gpu
MPPBaseOptionsDelegateType_Tpu
} MPPBaseOptionsDelegateType; } MPPBaseOptionsDelegateType;
NS_ASSUME_NONNULL_BEGIN NS_ASSUME_NONNULL_BEGIN

1
mediapipe/tasks/ios/core/sources/MPPBaseOptions.m
View File

@ -28,6 +28,7 @@
MPPBaseOptions *baseOptions = [[MPPBaseOptions alloc] init]; MPPBaseOptions *baseOptions = [[MPPBaseOptions alloc] init];
baseOptions.modelAssetPath = self.modelAssetPath; baseOptions.modelAssetPath = self.modelAssetPath;
baseOptions.delegateType = self.delegateType;
return baseOptions; return baseOptions;
} }

21
mediapipe/tasks/ios/core/utils/sources/MPPBaseOptions+Helpers.mm
View File

@ -35,17 +35,16 @@ using BaseOptionsProto = ::mediapipe::tasks::core::proto::BaseOptions;
} }
if (self.delegateType != MPPBaseOptionsDelegateType_Unknown) { if (self.delegateType != MPPBaseOptionsDelegateType_Unknown) {
switch (self.delegateType) { switch (self.delegateType) {
case MPPBaseOptionsDelegateType_Cpu: case MPPBaseOptionsDelegateType_Cpu: {
baseOptionsProto->mutable_acceleration()->mutable_tflite(); baseOptionsProto->mutable_acceleration()->mutable_tflite();
break; }
case MPPBaseOptionsDelegateType_Gpu: break;
baseOptionsProto->mutable_acceleration()->mutable_gpu()->set_use_advanced_gpu_api(true); case MPPBaseOptionsDelegateType_Gpu: {
break; baseOptionsProto->mutable_acceleration()->mutable_gpu()->set_use_advanced_gpu_api(true);
case MPPBaseOptionsDelegateType_Tpu: }
baseOptionsProto->mutable_acceleration()->mutable_nnapi()->set_accelerator_name("google-edgetpu"); break;
break; default:
default: break;
break;
} }
} }
} }