LenzTensorFlowLib.h 601 B

123456789101112131415161718192021222324252627
  1. //
  2. // LenzTensorFlow.h
  3. // LenzNormalCapture_Example
  4. //
  5. // Created by Zero on 2020/6/12.
  6. // Copyright © 2020 zero. All rights reserved.
  7. //
  8. #import <Foundation/Foundation.h>
  9. // Set TFLITE_USE_GPU_DELEGATE to 1 to use TFLite GPU Delegate.
  10. #ifndef TFLITE_USE_GPU_DELEGATE
  11. #define TFLITE_USE_GPU_DELEGATE 0
  12. #endif
  13. @interface LenzTensorFlowLib : NSObject
  14. //default 0.1
  15. @property (nonatomic,assign) float threshold;
  16. //modelPath不能为空
  17. - (id)initWithModelPath:(NSString *)modelPath labels:(NSArray <NSString *> *)labels;
  18. - (NSDictionary *)runModelOnFrame:(CVPixelBufferRef)pixelBuffer;
  19. @end