1 /* 2 * Copyright (c) 2022 HiSilicon (Shanghai) Technologies CO., LIMITED. 3 * Licensed under the Apache License, Version 2.0 (the "License"); 4 * you may not use this file except in compliance with the License. 5 * You may obtain a copy of the License at 6 * 7 * http://www.apache.org/licenses/LICENSE-2.0 8 * 9 * Unless required by applicable law or agreed to in writing, software 10 * distributed under the License is distributed on an "AS IS" BASIS, 11 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 12 * See the License for the specific language governing permissions and 13 * limitations under the License. 14 */ 15 16 #ifndef TENNIS_DETECT_H 17 #define TENNIS_DETECT_H 18 19 #include <iostream> 20 #include "sample_comm_nnie.h" 21 22 #if __cplusplus 23 extern "C" { 24 #endif 25 26 typedef struct tagIPC_IMAGE { 27 HI_U64 u64PhyAddr; 28 HI_U64 u64VirAddr; 29 HI_U32 u32Width; 30 HI_U32 u32Height; 31 } IPC_IMAGE; 32 33 class tennis_detect { 34 public: 35 HI_S32 TennisDetectLoad(uintptr_t* model); 36 HI_S32 TennisDetectUnload(uintptr_t model); 37 /* 38 * 网球检测推理 39 * Tennis detect calculation 40 */ 41 HI_S32 TennisDetectCal(uintptr_t model, VIDEO_FRAME_INFO_S *srcFrm, VIDEO_FRAME_INFO_S *dstFrm); 42 }; 43 44 #ifdef __cplusplus 45 } 46 #endif 47 #endif