pytorch transform数据处理转c++问题

 更新时间:2023年02月02日 11:12:19   作者:young_s%  
这篇文章主要介绍了pytorch transform数据处理转c++问题,具有很好的参考价值,希望对大家有所帮助。如有错误或未考虑完全的地方,望不吝赐教

pytorch transform数据处理转c++

python推理代码转c++ sdk过程遇到pytorch数据处理的转换

1.python代码

import torch
from PIL import Image
from torchvision import transforms

data_transform = transforms.Compose(
     [transforms.Resize(256),
      transforms.CenterCrop(224),
      transforms.ToTensor(),
      transforms.Normalize([0.485, 0.456, 0.406], [0.229, 0.224, 0.225])])

 img = Image.open(img_path)
 img = data_transform(img)

2.transforms.Resize(256)

Parameters
size (sequence or int) –
Desired output size. If size is a sequence like (h, w), output size will be matched to this. If size is an int, smaller edge of the image will be matched to this number. i.e, if height > width, then image will be rescaled to (size * height / width, size).

3.transforms.ToTensor()

Convert a PIL Image or numpy.ndarray to tensor. This transform does not support torchscript.
Converts a PIL Image or numpy.ndarray (H x W x C) in the range [0, 255] to a torch.FloatTensor of shape (C x H x W) in the range [0.0, 1.0] if the PIL Image belongs to one of the modes (L, LA, P, I, F, RGB, YCbCr, RGBA, CMYK, 1) or if the numpy.ndarray has dtype = np.uint8

cv::Mat ClsSixPrivate::processImage(cv::Mat &img) {
    int inW = img.cols;
    int inH = img.rows;
    cv::Mat croped_image;
    if (inW > inH)
    {
        int newWidth = 256 * inW / inH;
        cv::resize(img, img, cv::Size(newWidth, 256), 0, 0, cv::INTER_LINEAR);
        croped_image = img(cv::Rect((newWidth - 224) / 2, 16, 224, 224)).clone();
    }
    else {
        int newHeight= 256 * inH / inW;
        cv::resize(img, img, cv::Size(256, newHeight), 0, 0, cv::INTER_LINEAR);
        croped_image = img(cv::Rect(16, (newHeight - 224) / 2, 224, 224)).clone();
    }
    
    std::vector<float> mean_value{ 0.485, 0.456,0.406 };
    std::vector<float> std_value{ 0.229, 0.224, 0.225 }; 
    cv::Mat dst;
    std::vector<cv::Mat> rgbChannels(3);
    cv::split(croped_image, rgbChannels);

    for (auto i = 0; i < rgbChannels.size(); i++)
    {
        rgbChannels[i].convertTo(rgbChannels[i], CV_32FC1, 1.0 / (std_value[i] * 255.0), (0.0 - mean_value[i]) / std_value[i]);
    }

    cv::merge(rgbChannels, dst);
    return dst;
}

总结

以上为个人经验,希望能给大家一个参考,也希望大家多多支持脚本之家。

相关文章

  • python3+PyQt5实现拖放功能

    python3+PyQt5实现拖放功能

    这篇文章主要为大家详细介绍了python3+PyQt5实现拖放功能,文中示例代码介绍的非常详细,具有一定的参考价值,感兴趣的小伙伴们可以参考一下
    2018-04-04
  • Numpy中如何创建矩阵并等间隔抽取数据

    Numpy中如何创建矩阵并等间隔抽取数据

    这篇文章主要介绍了Numpy中如何创建矩阵并等间隔抽取数据问题,具有很好的参考价值,希望对大家有所帮助。如有错误或未考虑完全的地方,望不吝赐教
    2023-05-05
  • Python count函数使用方法实例解析

    Python count函数使用方法实例解析

    这篇文章主要介绍了Python count函数使用方法实例解析,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
    2020-03-03
  • 基于Python __dict__与dir()的区别详解

    基于Python __dict__与dir()的区别详解

    下面小编就为大家带来一篇基于Python __dict__与dir()的区别详解。小编觉得挺不错的,现在就分享给大家,也给大家做个参考。一起跟随小编过来看看吧
    2017-10-10
  • Django+Celery实现定时任务的示例

    Django+Celery实现定时任务的示例

    Celery是一个基于python开发的分布式任务队列,而做python WEB开发最为流行的框架莫属Django,本示例使用主要依赖包Django+Celery实现定时任务,感兴趣的朋友一起看看吧
    2021-06-06
  • python实现健康码查验系统

    python实现健康码查验系统

    这篇文章主要介绍了 python实现健康码查验系统,主要用到的是python用了opencv库和pyzbar库,文中给大家提供一段代码判断是否绿码,需要的朋友可以参考下
    2022-04-04
  • Python实现学生管理系统的完整代码(面向对象)

    Python实现学生管理系统的完整代码(面向对象)

    这篇文章主要介绍了Python实现学生管理系统的完整代码(面向对象),本文通过实例代码给大家介绍的非常详细,对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下
    2021-04-04
  • 利用Python第三方库xlrd读取Excel中数据实例代码

    利用Python第三方库xlrd读取Excel中数据实例代码

    python操作excel主要用到xlrd和xlwt这两个库,即xlrd是读excel,xlwt是写excel的库,下面这篇文章主要给大家介绍了关于利用Python第三方库xlrd读取Excel中数据的相关资料,文中通过实例代码介绍的非常详细,需要的朋友可以参考下
    2022-07-07
  • 详解Python requests模块

    详解Python requests模块

    今天给大家带来的是关于Python的相关知识,文章围绕着Python requests模块展开,文中有非常详细的介绍及代码示例,需要的朋友可以参考下
    2021-06-06
  • python,Django实现的淘宝客登录功能示例

    python,Django实现的淘宝客登录功能示例

    这篇文章主要介绍了python,Django实现的淘宝客登录功能,结合实例形式分析了Django框架基于淘宝接口的登录功能相关操作技巧,需要的朋友可以参考下
    2019-06-06

最新评论