Added iOS language detector options
This commit is contained in:
		
							parent
							
								
									5366aa9d0a
								
							
						
					
					
						commit
						3c13e4b6d6
					
				
							
								
								
									
										27
									
								
								mediapipe/tasks/ios/text/language_detector/BUILD
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										27
									
								
								mediapipe/tasks/ios/text/language_detector/BUILD
									
									
									
									
									
										Normal file
									
								
							| 
						 | 
					@ -0,0 +1,27 @@
 | 
				
			||||||
 | 
					# Copyright 2023 The MediaPipe Authors.
 | 
				
			||||||
 | 
					#
 | 
				
			||||||
 | 
					# Licensed under the Apache License, Version 2.0 (the "License");
 | 
				
			||||||
 | 
					# you may not use this file except in compliance with the License.
 | 
				
			||||||
 | 
					# You may obtain a copy of the License at
 | 
				
			||||||
 | 
					#
 | 
				
			||||||
 | 
					#      http://www.apache.org/licenses/LICENSE-2.0
 | 
				
			||||||
 | 
					#
 | 
				
			||||||
 | 
					# Unless required by applicable law or agreed to in writing, software
 | 
				
			||||||
 | 
					# distributed under the License is distributed on an "AS IS" BASIS,
 | 
				
			||||||
 | 
					# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 | 
				
			||||||
 | 
					# See the License for the specific language governing permissions and
 | 
				
			||||||
 | 
					# limitations under the License.
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					package(default_visibility = [
 | 
				
			||||||
 | 
					    "//mediapipe/tasks:internal",
 | 
				
			||||||
 | 
					])
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					licenses(["notice"])
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					objc_library(
 | 
				
			||||||
 | 
					    name = "MPPLanguageDetectorOptions",
 | 
				
			||||||
 | 
					    srcs = ["sources/MPPLanguageDetectorOptions.m"],
 | 
				
			||||||
 | 
					    hdrs = ["sources/MPPLanguageDetectorOptions.h"],
 | 
				
			||||||
 | 
					    deps = ["//mediapipe/tasks/ios/core:MPPTaskOptions"],
 | 
				
			||||||
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
| 
						 | 
					@ -0,0 +1,61 @@
 | 
				
			||||||
 | 
					// Copyright 2023 The MediaPipe Authors.
 | 
				
			||||||
 | 
					//
 | 
				
			||||||
 | 
					// Licensed under the Apache License, Version 2.0 (the "License");
 | 
				
			||||||
 | 
					// you may not use this file except in compliance with the License.
 | 
				
			||||||
 | 
					// You may obtain a copy of the License at
 | 
				
			||||||
 | 
					//
 | 
				
			||||||
 | 
					//      http://www.apache.org/licenses/LICENSE-2.0
 | 
				
			||||||
 | 
					//
 | 
				
			||||||
 | 
					// Unless required by applicable law or agreed to in writing, software
 | 
				
			||||||
 | 
					// distributed under the License is distributed on an "AS IS" BASIS,
 | 
				
			||||||
 | 
					// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 | 
				
			||||||
 | 
					// See the License for the specific language governing permissions and
 | 
				
			||||||
 | 
					// limitations under the License.
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					#import <Foundation/Foundation.h>
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					#import "mediapipe/tasks/ios/core/sources/MPPTaskOptions.h"
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					NS_ASSUME_NONNULL_BEGIN
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					/**
 | 
				
			||||||
 | 
					 * Options for setting up a `LanguageDetector`.
 | 
				
			||||||
 | 
					 */
 | 
				
			||||||
 | 
					NS_SWIFT_NAME(LanguageDetectorOptions)
 | 
				
			||||||
 | 
					@interface MPPLanguageDetectorOptions : MPPTaskOptions <NSCopying>
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					/**
 | 
				
			||||||
 | 
					 * The locale to use for display names specified through the TFLite Model Metadata, if any. Defaults
 | 
				
			||||||
 | 
					 * to English.
 | 
				
			||||||
 | 
					 */
 | 
				
			||||||
 | 
					@property(nonatomic, copy) NSString *displayNamesLocale;
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					/**
 | 
				
			||||||
 | 
					 * The maximum number of top-scored classification results to return. If < 0, all available results
 | 
				
			||||||
 | 
					 * will be returned. If 0, an invalid argument error is returned.
 | 
				
			||||||
 | 
					 */
 | 
				
			||||||
 | 
					@property(nonatomic) NSInteger maxResults;
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					/**
 | 
				
			||||||
 | 
					 * Score threshold to override the one provided in the model metadata (if any). Results below this
 | 
				
			||||||
 | 
					 * value are rejected.
 | 
				
			||||||
 | 
					 */
 | 
				
			||||||
 | 
					@property(nonatomic) float scoreThreshold;
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					/**
 | 
				
			||||||
 | 
					 * The allowlist of category names. If non-empty, detection results whose category name is not in
 | 
				
			||||||
 | 
					 * this set will be filtered out. Duplicate or unknown category names are ignored. Mutually
 | 
				
			||||||
 | 
					 * exclusive with categoryDenylist.
 | 
				
			||||||
 | 
					 */
 | 
				
			||||||
 | 
					@property(nonatomic, copy) NSArray<NSString *> *categoryAllowlist;
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					/**
 | 
				
			||||||
 | 
					 * The denylist of category names. If non-empty, detection results whose category name is in this
 | 
				
			||||||
 | 
					 * set will be filtered out. Duplicate or unknown category names are ignored. Mutually exclusive
 | 
				
			||||||
 | 
					 * with categoryAllowlist.
 | 
				
			||||||
 | 
					 */
 | 
				
			||||||
 | 
					@property(nonatomic, copy) NSArray<NSString *> *categoryDenylist;
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@end
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					NS_ASSUME_NONNULL_END
 | 
				
			||||||
| 
						 | 
					@ -0,0 +1,40 @@
 | 
				
			||||||
 | 
					// Copyright 2023 The MediaPipe Authors.
 | 
				
			||||||
 | 
					//
 | 
				
			||||||
 | 
					// Licensed under the Apache License, Version 2.0 (the "License");
 | 
				
			||||||
 | 
					// you may not use this file except in compliance with the License.
 | 
				
			||||||
 | 
					// You may obtain a copy of the License at
 | 
				
			||||||
 | 
					//
 | 
				
			||||||
 | 
					//      http://www.apache.org/licenses/LICENSE-2.0
 | 
				
			||||||
 | 
					//
 | 
				
			||||||
 | 
					// Unless required by applicable law or agreed to in writing, software
 | 
				
			||||||
 | 
					// distributed under the License is distributed on an "AS IS" BASIS,
 | 
				
			||||||
 | 
					// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 | 
				
			||||||
 | 
					// See the License for the specific language governing permissions and
 | 
				
			||||||
 | 
					// limitations under the License.
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					#import "mediapipe/tasks/ios/text/language_detector/sources/MPPLanguageDetectorOptions.h"
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@implementation MPPLanguageDetectorOptions
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					- (instancetype)init {
 | 
				
			||||||
 | 
					  self = [super init];
 | 
				
			||||||
 | 
					  if (self) {
 | 
				
			||||||
 | 
					    _maxResults = -1;
 | 
				
			||||||
 | 
					    _scoreThreshold = 0;
 | 
				
			||||||
 | 
					  }
 | 
				
			||||||
 | 
					  return self;
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					- (id)copyWithZone:(NSZone *)zone {
 | 
				
			||||||
 | 
					  MPPLanguageDetectorOptions *languageDetectorOptions = [super copyWithZone:zone];
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					  languageDetectorOptions.scoreThreshold = self.scoreThreshold;
 | 
				
			||||||
 | 
					  languageDetectorOptions.maxResults = self.maxResults;
 | 
				
			||||||
 | 
					  languageDetectorOptions.categoryDenylist = self.categoryDenylist;
 | 
				
			||||||
 | 
					  languageDetectorOptions.categoryAllowlist = self.categoryAllowlist;
 | 
				
			||||||
 | 
					  languageDetectorOptions.displayNamesLocale = self.displayNamesLocale;
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					  return languageDetectorOptions;
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@end
 | 
				
			||||||
		Loading…
	
		Reference in New Issue
	
	Block a user