4.6. BMImage

BMImage封装了一张图片的全部信息，可利用Bmcv接口将BMImage转换为Tensor进行模型推理。

BMImage也是通过Bmcv接口进行其他图像处理操作的基本数据类型。

4.6.1. 构造函数

初始化BMImage。

接口形式:

BMImage();

BMImage(
    Handle&                  handle,
    int                      h,
    int                      w,
    bm_image_format_ext      format,
    bm_image_data_format_ext dtype);

BMImage(
    Handle                   &handle,
    void*                    buffer,
    int                      h,
    int                      w,
    bm_image_format_ext      format,
    bm_image_data_format_ext dtype = DATA_TYPE_EXT_1N_BYTE,
    std::vector<int>         strides = {},
    size_t                   offset = 0);

参数说明:

handle: Handle

设定BMImage所在的设备句柄。

h: int

图像的高。

w: int

图像的宽。

format : bm_image_format_ext

图像的像素格式。支持 sail.Format 中的像素格式。

dtype: bm_image_data_format_ext

图像的数据类型。支持 ImgDtype 中的数据类型。

buffer: void*

用buffer创建图像时buffer的地址。

strides: std::vector<int>

用buffer创建图像时图像的步长。单位为byte，默认为空，表示和一行的数据宽度相同。如果需要指定，注意stride元素个数要与图像plane数一致

offset: int

用buffer创建图像时有效数据相对buffer起始地址的偏移量。单位为byte，默认为0

4.6.2. width

获取图像的宽。

接口形式:

int width();

返回值说明:

width : int

返回图像的宽。

4.6.3. height

获取图像的高。

接口形式:

int height();

返回值说明:

height : int

返回图像的高。

4.6.4. format

获取图像的格式。

接口形式:

bm_image_format_ext format();

返回值说明:

format : bm_image_format_ext

返回图像的格式。

4.6.5. dtype

获取图像的数据类型。

接口形式:

bm_image_data_format_ext dtype() const;

返回值说明:

dtype: bm_image_data_format_ext

返回图像的数据类型。

4.6.6. data

获取BMImage内部的bm_image。

接口形式:

bm_image& data();

返回值说明:

img : bm_image

返回图像内部的bm_image。

4.6.7. get_device_id

获取BMImage中的设备id号。

接口形式:

int get_device_id() const;

返回值说明:

device_id : int

返回BMImage中的设备id号

4.6.8. get_handle

获取BMImage中的Handle。

接口形式:

Handle& get_handle();

返回值说明:

Handle : Handle

返回BMImage中的Handle

4.6.9. get_plane_num

获取BMImage中图像plane的数量。

接口形式:

int get_plane_num() const;

返回值说明:

planes_num : int

返回BMImage中图像plane的数量。

4.6.10. align

将BMImage 64对齐

接口形式:

int align();

返回值说明:

ret : int

返回BMImage是否对齐成功, -1代表BMImage未创建, 0代表成功, 其他代表失败

4.6.11. check_align

获取BMImage中图像是否对齐

接口形式:

bool check_align()const;

返回值说明:

ret : bool

1代表已对齐, 0代表未对齐, -1代表BMImage未创建

4.6.12. unalign

将BMImage不对齐

接口形式:

int unalign();

返回值说明:

ret : int

返回BMImage是否不对齐成功, -1代表BMImage未创建, 0代表成功, 其他代表失败

4.6.13. check_contiguous_memory

获取BMImage中图像内存是否连续

接口形式:

bool check_contiguous_memory()const;

返回值说明:

ret : bool

1代表连续,0代表不连续

示例代码:

#include <sail/cvwrapper.h>

int main() {
    int dev_id = 0;
    sail::Handle handle(dev_id);
    std::string image_name = "your_image.jpg";
    sail::Decoder decoder(image_name, true, dev_id);
    sail::BMImage BMimg = decoder.read(handle);

    // Get the image information
    int width = BMimg.width();
    int height = BMimg.height();
    bm_image_format_ext format = BMimg.format();
    bm_image_data_format_ext dtype = BMimg.dtype();

    // Convert BMImage to bm_image data structure
    bm_image bmimg = BMimg.data();

    // Get the device id and handle
    int device_id = BMimg.get_device_id();
    sail::Handle handle_ = BMimg.get_handle();
    int plane_num = BMimg.get_plane_num();
    std::cout << "Width: " << width << ", Height: " << height << ", Format: " << format << ", Data Type: " << dtype << ", Device ID: " << device_id << ", Plane Num: " << plane_num << std::endl;

    int ret;
    // Align the image
    ret = BMimg.align();
    if (ret != 0) {
        std::cout << "Failed to align the image!" << std::endl;
    }
    std::cout << "is align: " << BMimg.check_align() << std::endl;

    // unalign the image
    ret = BMimg.unalign();
    if (ret != 0) {
        std::cout << "Failed to unalign the image!" << std::endl;
    }
    std::cout << "is align: " << BMimg.check_align() << std::endl;

    // check contiguous memory
    std::cout << "is continues: " <<BMimg.check_contiguous_memory()<< std::endl;

    // create BMImage with data from buffer
    std::vector<uint8_t> buf(200 * 100 * 3);
    for (int i = 0; i < 200 * 100 * 3; ++i) {
        buf[i] = i % 256;
    }
    sail::BMImage img_fromRawdata(handle, buf.data(), 200, 100, sail::Format::FORMAT_BGR_PACKED);

    return 0;
}

4.6.14. get_pts_dts

获取pts和dts

接口形式:

vector<double> get_pts_dts()

返回值说明:

result: vector<double>

输出结果。输出具体的pts和dts值。

示例代码:

#include <sail/cvwrapper.h>

using namespace std;
using namespace sail;

int main() {
    string file_path = "your_video_file_path.mp4";
    int tpu_id = 0;

    Handle handle(tpu_id);
    Decoder decoder(file_path, true, tpu_id);
    BMImage image;

    int ret = decoder.read(handle, image);
    if (ret != 0) {
        cout << "Failed to read a frame!" << endl;
        return ret;
    }

    std::vector<int> pts_dts;
    pts_dts = image.get_pts_dts();
    cout << "pts: " << pts_dts[0] << endl;
    cout << "dts: " << pts_dts[1] << endl;
    return 0;
}