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

Updated category allowlist and denylist names

Browse Source
This commit is contained in:
Prianka Liz Kariat 2022-12-22 01:59:38 +05:30
parent c185dc9ad7
commit 20c3388ab6
3 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mediapipe/tasks/ios/components/processors/sources/MPPClassifierOptions.h
View File

@ -36,10 +36,10 @@ NS_SWIFT_NAME(ClassifierOptions)
@property(nonatomic) float scoreThreshold; @property(nonatomic) float scoreThreshold;
/** If set, all classes not in this list will be filtered out from the results. */ /** If set, all classes not in this list will be filtered out from the results. */
@property(nonatomic, copy) NSArray<NSString *> *categoryAllowList; @property(nonatomic, copy) NSArray<NSString *> *categoryAllowlist;
/** If set, all classes in this list will be filtered out from the results. */ /** If set, all classes in this list will be filtered out from the results. */
@property(nonatomic, copy) NSArray<NSString *> *categoryDenyList; @property(nonatomic, copy) NSArray<NSString *> *categoryDenylist;
@end @end

4
mediapipe/tasks/ios/components/processors/sources/MPPClassifierOptions.m
View File

@ -30,8 +30,8 @@
classifierOptions.scoreThreshold = self.scoreThreshold; classifierOptions.scoreThreshold = self.scoreThreshold;
classifierOptions.maxResults = self.maxResults; classifierOptions.maxResults = self.maxResults;
classifierOptions.categoryDenyList = self.categoryDenyList; classifierOptions.categoryDenylist = self.categoryDenylist;
classifierOptions.categoryAllowList = self.categoryAllowList; classifierOptions.categoryAllowlist = self.categoryAllowlist;
classifierOptions.displayNamesLocale = self.displayNamesLocale; classifierOptions.displayNamesLocale = self.displayNamesLocale;
return classifierOptions; return classifierOptions;

4
mediapipe/tasks/ios/components/processors/utils/sources/MPPClassifierOptions+Helpers.mm
View File

@ -30,11 +30,11 @@ using ClassifierOptionsProto = ::mediapipe::tasks::components::processors::proto
classifierOptionsProto->set_score_threshold(self.scoreThreshold); classifierOptionsProto->set_score_threshold(self.scoreThreshold);
for (NSString *category in self.categoryAllowList) { for (NSString *category in self.categoryAllowlist) {
classifierOptionsProto->add_category_allowlist(category.cppString); classifierOptionsProto->add_category_allowlist(category.cppString);
} }
for (NSString *category in self.categoryDenyList) { for (NSString *category in self.categoryDenylist) {
classifierOptionsProto->add_category_denylist(category.cppString); classifierOptionsProto->add_category_denylist(category.cppString);
} }
} }