ResizedImageInfo

功能

用于记录图像类任务中模型前处理中的缩放方式,提供给模型后处理的坐标还原使用。

结构定义

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
class ResizedImageInfo {
public:
    
    ResizedImageInfo() {}
    ResizedImageInfo(uint32_t wResize, uint32_t hResize, uint32_t wOriginal, uint32_t hOriginal, ResizeType rType, float kARScaling) :
                     widthResize(wResize), heightResize(hResize), widthOriginal(wOriginal), heightOriginal(hOriginal), resizeType(rType),
                     keepAspectRatioScaling(kARScaling) {}
    uint32_t widthResize = 0;
    uint32_t heightResize = 0;
    uint32_t widthOriginal = 0;
    uint32_t heightOriginal = 0;
    ResizeType resizeType = RESIZER_STRETCHING;
    float keepAspectRatioScaling = 0;
};

参数说明

参数名

说明

widthResize

缩放后图像宽,即模型的输入宽。

heightResize

缩放后图像高,即模型的输入高。

widthOriginal

缩放前图像宽。

heightOriginal

缩放前图像高。

resizeType

枚举类型ResizeType,代表图像的缩放方式。

  • RESIZER_STRETCHING:拉伸缩放,默认缩放方式。
  • RESIZER_TF_KEEP_ASPECT_RATIO:等比缩放,和TensorFlow框架FastRCNN模型缩放方式对应。
  • RESIZER_MS_KEEP_ASPECT_RATIO:等比缩放,使图片等比缩放至在指定宽高的区域内面积最大化。

keepAspectRatioScaling

等比例缩放的缩放比例,等比缩放的缩放方式下生效。