LibTorch函数读取img返回Tensor时值改变

LibTorch 读取img,归一化并返回tensor

  1. 函数需要返回深拷贝,from_bob传递的是引用,img超出作用域后会销毁,导致tensor传递出错
torch::Tensor preprocessImage(const std::string& img_path, int out_size) {
	// Load the image using OpenCV
	cv::Mat test_image_data = cv::imread(img_path);
	cv::Mat rgb_image;
	cv::cvtColor(test_image_data, rgb_image, cv::COLOR_BGR2RGB);
	cv::Mat resized_image;
	cv::resize(rgb_image, resized_image, cv::Size(out_size, out_size));  // resize 方法待改为与train一致
	// Normalize the image data (assuming RGB values are in [0, 255])
	cv::Mat normalized_image;
	resized_image.convertTo(normalized_image, CV_32F, 1.0 / 255.0);
	cv::subtract(normalized_image, cv::Scalar(0.48145466, 0.4578275, 0.40821073), normalized_image);
	cv::divide(normalized_image, cv::Scalar(0.26862954, 0.26130258, 0.27577711), normalized_image);

	// Create a tensor to hold the preprocessed image
	torch::Tensor img_tensor = torch::from_blob(normalized_image.data, { 1, out_size, out_size, 3 }, torch::kFloat);
	img_tensor = img_tensor.permute({ 0, 3, 1, 2 });  // Change to PyTorch convention: [batch_size, channels, height, width]

	return img_tensor.clone();  //需要返回深拷贝,from_bob传递的是引用,img超出作用域后会销毁导致tensor传递出错
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值