Blame view

src/common/dvppx/dvpp_processx.h 2.61 KB
20396d5c   Hu Chunming   添加车头车尾算法
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
  /**
  * Copyright 2020 Huawei Technologies Co., Ltd
  *
  * 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.
  
  * File dvpp_process.h
  * Description: handle dvpp process
  */
  #pragma once
  #include <cstdint>
  
  #include "acl/acl.h"
  #include "acl/ops/acl_dvpp.h"
  #include "utils.h"
  
  
  typedef struct PicDesc {
      std::string picName;
  
      uint32_t width;
      uint32_t height;
      uint32_t widthStride;
      uint32_t heightStride;
  
      acldvppPixelFormat format;
  
      uint32_t dataSize;
      void* dataBuffer;
  
      uint32_t predictedJpegDecodeSize;
  } PicDesc;
  
  
  typedef struct PicRoi {
      uint32_t xmin;
      uint32_t ymin;
      uint32_t xmax;
      uint32_t ymax;
  } PicRoi;
  
  
  /**
   * DvppProcessx
   */
  class DvppProcessx {
  public:
      DvppProcessx();
  
      ~DvppProcessx();
  
      int CropAndPadding(ImageData& src, ImageData& dest,
                    uint32_t width, uint32_t height);
      int CropAndPaste(ImageData& src, ImageData& dest,
                    uint32_t width, uint32_t height);
b3012672   Hu Chunming   天啊及hp,road_seg子sd...
65
      ImageData* Crop_naked(ImageData& src, uint32_t xmin, uint32_t ymin, uint32_t xmax, uint32_t ymax, uint32_t width, uint32_t height);
20396d5c   Hu Chunming   添加车头车尾算法
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
      int PatchCropAndPaste(ImageData& dest, ImageData& src, uint32_t xmin, uint32_t ymin,
                    uint32_t xmax, uint32_t ymax, uint32_t width, uint32_t height);
      int Crop2Paste(ImageData& dest, ImageData& leftImage, ImageData& rightImage, 
                     ImageData& src,uint32_t width, uint32_t height);
      int CvtJpegToYuv420sp(ImageData& src, ImageData& dest);
      int ConvertColor(PicDesc& inPicDesc, PicDesc& outPicDesc);
      int CropAndPaddingBatch(ImageData* src, ImageData* dest, int batchsize, uint32_t width, uint32_t height);
      int CropAndPasteBatch(ImageData* src, ImageData* dest, int batchsize, uint32_t width, uint32_t height);
      int CropAndPasteBatchV2(ImageData* src, ImageData* dest, PicRoi* PicRois, int batchsize, uint32_t width, uint32_t height);
      // vector<vpc_img_info> crop_batch(DeviceMemory *devMem, vector<video_object_info> objs);
      int InitResource(aclrtStream& stream);
      void DestroyResource();
  
  protected:
      int isInitOk_;
      aclrtStream stream_;
      acldvppChannelDesc *dvppChannelDesc_;
      bool isGlobalContext_;
  
  };