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 {
MPPBaseOptionsDelegateType_Unknown,
MPPBaseOptionsDelegateType_Cpu,
MPPBaseOptionsDelegateType_Gpu,
MPPBaseOptionsDelegateType_Tpu
MPPBaseOptionsDelegateType_Gpu
} MPPBaseOptionsDelegateType;
NS_ASSUME_NONNULL_BEGIN

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

@ -28,6 +28,7 @@
MPPBaseOptions *baseOptions = [[MPPBaseOptions alloc] init];
baseOptions.modelAssetPath = self.modelAssetPath;
baseOptions.delegateType = self.delegateType;
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) {
switch (self.delegateType) {
case MPPBaseOptionsDelegateType_Cpu:
baseOptionsProto->mutable_acceleration()->mutable_tflite();
break;
case MPPBaseOptionsDelegateType_Gpu:
baseOptionsProto->mutable_acceleration()->mutable_gpu()->set_use_advanced_gpu_api(true);
break;
case MPPBaseOptionsDelegateType_Tpu:
baseOptionsProto->mutable_acceleration()->mutable_nnapi()->set_accelerator_name("google-edgetpu");
break;
default:
break;
case MPPBaseOptionsDelegateType_Cpu: {
baseOptionsProto->mutable_acceleration()->mutable_tflite();
}
break;
case MPPBaseOptionsDelegateType_Gpu: {
baseOptionsProto->mutable_acceleration()->mutable_gpu()->set_use_advanced_gpu_api(true);
}
break;
default:
break;
}
}
}