LenzTensorFlowLib.h 633 B

12345678910111213141516171819202122232425262728
  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. #import <CoreVideo/CoreVideo.h>
  10. // Set TFLITE_USE_GPU_DELEGATE to 1 to use TFLite GPU Delegate.
  11. #ifndef TFLITE_USE_GPU_DELEGATE
  12. #define TFLITE_USE_GPU_DELEGATE 0
  13. #endif
  14. @interface LenzTensorFlowLib : NSObject
  15. //default 0.1
  16. @property (nonatomic,assign) float threshold;
  17. //modelPath不能为空
  18. - (id)initWithModelPath:(NSString *)modelPath labels:(NSArray <NSString *> *)labels;
  19. - (NSDictionary *)runModelOnFrame:(CVPixelBufferRef)pixelBuffer;
  20. @end