【工具】raw与jpg互转python-cpp

于 2024-01-30 11:06:24 发布
阅读量321 收藏 4
点赞数 3
分类专栏： 工具代码 文章标签： python c++ raw2jpg jpg2raw
by: JW
本文链接：https://blog.csdn.net/zhoujinwang/article/details/135928333
版权

在工作中常常需要将图像转化为raw数据或者yuv数据，这里将提供 cpp 版本和 python 版本的互转代码

代码链接见文档尾部。

cpp 版本

jpg2raw.cpp

#include <fstream>
#include <iostream>
#include <opencv2/core.hpp>
#include <opencv2/highgui.hpp>

// Windows 和 Linux 下通用的创建目录函数
#ifdef _WIN32
#include <direct.h> // For _mkdir on Windows
#define MKDIR(dir) _mkdir(dir)
#else
#include <sys/stat.h> // For mkdir on POSIX systems
#define MKDIR(dir) mkdir(dir, 0755)
#endif

bool create_directory(const std::string &dir_path)
{
#ifdef _WIN32
    if (_mkdir(dir_path.c_str()) != 0)
    {
        return false;
    }
#else
    if (MKDIR(dir_path.c_str()) != 0 && errno != EEXIST)
    {
        return false;
    }
#endif
    return true;
}

int main(int argc, char **argv)
{
    if (argc != 2)
    {
        std::cerr << "Usage: ./image_to_raw input_image.jpg" << std::endl;
        return -1;
    }

    std::string input_image_path = argv[1];
    std::string base_name = input_image_path.substr(input_image_path.find_last_of('/') + 1); // Unix风格路径分隔符
#ifdef _WIN32
    std::replace(base_name.begin(), base_name.end(), '\\', '/'); // 对于Windows下的反斜杠，统一转换为正斜杠
#endif

    std::cout <<" ----------- input_image_path: " << input_image_path << " ------------------------"<<std::endl;
    std::cout <<" ----------- basename: " << base_name << " ------------------------"<<std::endl;

    std::string raw_dir = "raws/";
    // std::string output_raw_path = raw_dir + base_name + ".raw";

    std::string file_name = input_image_path.substr(input_image_path.find_last_of("/\\") + 1);
    std::string file_name_without_suffix = file_name.substr(0, file_name.find_last_of("."));
    std::string file_name_with_suffix = file_name_without_suffix + ".raw";
    std::string output_raw_path = raw_dir + file_name_with_suffix;


    // 创建 'raw' 子目录（如果不存在）
    if (!create_directory(raw_dir))
    {
        std::cerr << "Failed to create directory: " << raw_dir << std::endl;
        //return -1;
    }

    cv::Mat img = cv::imread(input_image_path, 1);
    if (img.empty())
    {
        std::cerr << "Failed to load the image: " << input_image_path << std::endl;
        return -1;
    }

    int im_width = img.cols;
    int im_height = img.rows;
    int im_depth = img.channels();

    std::cout << "im_width: " << im_width << std::endl;
    std::cout << "im_height: " << im_height << std::endl;
    std::cout << "im_depth: " << im_depth << std::endl;

    // 写入 RAW 文件
    std::ofstream raw_file(output_raw_path, std::ios::binary);
    if (!raw_file.is_open())
    {
        std::cerr << "Failed to open file for writing: " << output_raw_path << std::endl;
        return -1;
    }
    raw_file.write(reinterpret_cast<const char *>(img.data), im_width * im_height * im_depth);
    raw_file.close();

    std::cout << "Converted " << input_image_path << " to " << output_raw_path << std::endl;

    return 0;
}

raw2jpg

#include <fstream>
#include <iostream>
#include <opencv2/core.hpp>
#include <opencv2/highgui.hpp>
#include <opencv2/imgproc.hpp>
#include <unordered_map>

#ifdef _WIN32
#include <direct.h> // For _mkdir on Windows
#define MKDIR(dir) _mkdir(dir)
#else
#include <sys/stat.h> // For mkdir on POSIX systems
#define MKDIR(dir) mkdir(dir, 0755)
#endif

struct ImageSize
{
    int width;
    int height;
};

const std::unordered_map<int, ImageSize> image_sizes = {
    {921600, {640, 480}},
    {2764800, {1280, 720}},
    {5858640, {1896, 1030}},
    {6220800, {1920, 1080}},
    {11059200, {2560, 1440}},
    {12257280, {2688, 1520}},
    {15116544, {2592, 1944}},
    {3686400, {1280, 960}}};

// 跨平台创建目录函数（仅支持Linux和Windows）
bool create_directory(const std::string &path)
{
#ifdef _WIN32
    return (_mkdir(path.c_str()) == 0);
#else
    return (::mkdir(path.c_str(), S_IRWXU | S_IRWXG | S_IROTH | S_IXOTH) == 0);
#endif
}

int main(int argc, char **argv)
{
    if (argc != 2)
    {
        std::cerr << "Usage: ./raw_image_viewer raw_image.raw" << std::endl;
        return -1;
    }

    std::string raw_image_path = argv[1];
    std::cout << "raw_image_path: " << raw_image_path << std::endl;

    // 获取文件大小
    std::ifstream file(raw_image_path, std::ios::binary | std::ios::ate);
    size_t file_size = file.tellg();
    file.close();
    std::cout << "size: " << file_size << " bytes" << std::endl;

    // 查找对应的图像尺寸
    auto it = image_sizes.find(file_size);
    if (it == image_sizes.end())
    {
        std::cout << file_size << " is not supported!" << std::endl;
        return -1;
    }
    const ImageSize &size = it->second;
    int im_width = size.width;
    int im_height = size.height;
    int im_depth = 3; // 假设所有图像都是RGB三通道

    std::cout << "im_width: " << im_width << std::endl;
    std::cout << "im_height: " << im_height << std::endl;
    std::cout << "im_depth: " << im_depth << std::endl;

    cv::Mat raw_image(im_height, im_width, CV_8UC3);

    // 读取 RAW 文件的内容
    std::ifstream raw_file(raw_image_path, std::ios::binary);
    raw_file.read(reinterpret_cast<char *>(raw_image.data), im_width * im_height * im_depth);
    raw_file.close();

    // 提取并创建 images 文件夹（跨平台）
    std::string images_path = "images/";
    bool dir_created = create_directory(images_path);

    std::string file_name = raw_image_path.substr(raw_image_path.find_last_of("/\\") + 1);
    std::string file_name_without_suffix = file_name.substr(0, file_name.find_last_of("."));
    std::string file_name_with_suffix = file_name_without_suffix + ".jpg";
    std::string output_path = images_path + file_name_with_suffix;

    cv::imwrite(output_path, raw_image);

    std::cout << " images_path: " << images_path << std::endl;
    std::cout << " file_name_with_suffix: " << file_name_with_suffix << std::endl;
    std::cout << " file save name: " << output_path << std::endl;

    if (!dir_created)
    {
        std::cerr << "Failed to create 'images' directory." << std::endl;
    }

    // （可选）显示图像
    // cv::imshow("raw_image", raw_image);
    // cv::waitKey(5);
    // cv::destroyAllWindows();

    // （可选）删除原始 RAW 文件
    // std::remove(raw_image_path.c_str());

    return 0;
}

CMakeLists.txt

cmake_minimum_required(VERSION 3.10)

project(convertRAW2jpg CXX)

# 设置安装目录
set(CMAKE_INSTALL_PREFIX "${PROJECT_BINARY_DIR}/install")
set(CMAKE_SKIP_INSTALL_RPATH FALSE)
set(CMAKE_BUILD_WITH_INSTALL_RPATH TRUE)
set(CMAKE_INSTALL_RPATH "${CMAKE_INSTALL_PREFIX}/lib")

# 要求编译器支持C++11
set(CMAKE_CXX_STANDARD 11)
set(CMAKE_CXX_STANDARD_REQUIRED True)

# Windows 特定选项和标志
if(MSVC)
    add_compile_options(/utf-8)
    add_definitions(-D_CRT_SECURE_NO_WARNINGS)
endif()


# OpenCV 手动配置部分（保留 find_package 并注释）
if(WIN32)
    # find_package(OpenCV REQUIRED)  # 注释掉以使用自定义路径
    set(OpenCV_DIR "D:/opencv480vc16/build" CACHE PATH "Path to a custom OpenCV build on Windows")
    set(OPENCV_INCLUDE_DIRS "${OpenCV_DIR}/include")
    set(OPENCV_LIB_DIR     "${OpenCV_DIR}/x64/vc16/lib") # 根据实际编译器版本调整

    include_directories(${OPENCV_INCLUDE_DIRS})
    link_directories(${OPENCV_LIB_DIR})

    list(APPEND EXTRA_LIBS opencv_world480.lib) # 根据实际OpenCV库名称调整
elseif(NOT WIN32)
    # Linux 下的手动 OpenCV 配置
    # find_package(OpenCV REQUIRED)  # 注释掉以使用自定义路径
    set(OpenCV_DIR "~/install/opencv4.5.4" CACHE PATH "Path to a custom OpenCV build on Unix-like systems")
    set(OPENCV_INCLUDE_DIRS "${OpenCV_DIR}/include/opencv4")
    set(OPENCV_LIB_DIR     "${OpenCV_DIR}/lib")

    include_directories(${OPENCV_INCLUDE_DIRS})
    link_directories(${OPENCV_LIB_DIR})

    list(APPEND EXTRA_LIBS opencv_world -fopenmp -lpthread -lm) # 根据实际OpenCV库名称和依赖调整
endif()

include_directories(${OpenCV_INCLUDE_DIRS})

# 添加源文件
add_executable(raw2jpg src/raw2jpg.cpp)
add_executable(jpg2raw src/jpg2raw.cpp)


target_link_libraries(raw2jpg PRIVATE ${EXTRA_LIBS})
target_link_libraries(jpg2raw PRIVATE ${EXTRA_LIBS})

# 安装目标及辅助脚本
set(TMP_TEST_DIR ${PROJECT_SOURCE_DIR}/temp_example)
install(TARGETS raw2jpg
        RUNTIME DESTINATION examples
        RUNTIME DESTINATION ${TMP_TEST_DIR})

install(TARGETS jpg2raw
        RUNTIME DESTINATION examples
        RUNTIME DESTINATION ${TMP_TEST_DIR})

# 找到 assets 目录下所有的 raw 文件，并安装到临时目录
FILE(GLOB RAWS assets/raws/*.raw)
FILE(GLOB JPGS assets/jpgs/*.jpg)

install(FILES ${RAWS} DESTINATION ${TMP_TEST_DIR}/test_raws)
install(FILES ${JPGS} DESTINATION ${TMP_TEST_DIR}/test_jpgs)

if (WIN32)
    install(FILES ${PROJECT_SOURCE_DIR}/onekey_windows_batch2jpg.bat
            DESTINATION examples
            DESTINATION ${TMP_TEST_DIR})
    install(FILES ${PROJECT_SOURCE_DIR}/onekey_windows_batch2raw.bat
            DESTINATION examples
            DESTINATION ${TMP_TEST_DIR})
else()
    install(FILES ${PROJECT_SOURCE_DIR}/onekey_ubuntu_batch2jpg.sh
            DESTINATION examples
            DESTINATION ${TMP_TEST_DIR})
    install(FILES ${PROJECT_SOURCE_DIR}/onekey_ubuntu_batch2raw.sh
            DESTINATION examples
            DESTINATION ${TMP_TEST_DIR})
endif(WIN32)

批量将 raw 转换为 jpg 的脚本

ubuntu

onekey_ubuntu_batch2jpg.sh

#!/bin/bash
set -e

if [ $# -eq 0 ]; then
	im_path=`pwd`
else
	im_path=$1
fi

echo "batch convert the raw image to jpg images in path: $im_path"
i=0
for im in $(ls $im_path)
do 
	if [ "${im##*.}" = "raw" ]; then
		if [ -f $im_path/$im ];then
			echo "convert $file to jpg image..."
            ./raw2jpg $im_path/$im
			let i+=1
		fi
	fi
done 
echo "DONE: $i pictures been gennerated!"**s**
windows

onekey_windows_batch2jpg.bat

@echo off

@REM 如果没有输入路径, 就使用当前路径
@if "%~1"=="" (
    set "im_path=%cd%"
) else (
    set "im_path=%~1"
)

@REM 输出批处理将要转换文件的路径信息
echo Batch converting RAW images to JPG in the current directory or specified path: %im_path%

@REM 遍历指定目录及其子目录下的所有.raw文件
for /R "%im_path%" %%i in (*.raw) do (
    @REM 获取图片名称（不包括完整路径）
    set "img_file=%%~ni"
    @REM 输出正在处理的原始图像文件名
    echo Processing image: %%i
    @REM 调用 raw2jpg.exe 进行转换，由于程序内部已经设置了输出目录和文件名规则，所以这里仅传入原始文件路径即可
    .\raw2jpg.exe "%%i"
)

@REM 暂停脚本执行，等待用户按键继续
pause

批量将 jpg 转换为 raw 的脚本

ubuntu

onekey_ubuntu_batch2raw.sh

#!/bin/bash
set -e

if [ $# -eq 0 ]; then
	im_path=`pwd`
else
	im_path=$1
fi

echo "batch convert the jpg images to raw image in path: $im_path"
i=0
for im in $(ls $im_path)
do 
	if [ "${im##*.}" = "jpg" ]; then
		if [ -f $im_path/$im ];then
			echo "convert $file to jpg image..."
            ./jpg2raw $im_path/$im
			let i+=1
		fi
	fi
done 
echo "DONE: $i pictures been gennerated!"s
windows

onekey_windows_batch2raw.bat

@echo off

@REM 如果没有输入路径, 就使用当前路径
if "%1"=="" (
    set "im_path=%cd%"
) else (
    set "im_path=%~1"
)

@REM 输出当前处理的路径
echo Batch converting image files to raw images in path: %im_path%

@REM 遍历指定目录及其子目录下的所有.jpg, .jpeg, .bmp, .png文件
for /R "%im_path%" %%i in (*.jpg, *.jpeg, *.bmp, *.png) do (
    @REM 获取图片名称（这里保留完整路径）
    echo Processing image: %%i
    .\jpg2raw.exe "%%i"
)

pause

所有代码见

raw2jpg-github

黄金旺铺
关注
  • 3
    点赞
  • 踩
  • 4
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
专栏目录
python应用——将raw文件转化为jpg文件，并显示图像
北落师门XY的博客
08-26 1万+
例一： 一、要求 读入0.raw的文件，转化为0.jpg文件。 二、代码 import numpy as np import imageio rawfile = np.fromfile('0.raw', dtype=np.float32) # 以float32读图片 print(rawfile.shape) rawfile.shape = (480, 640) print(rawfile...
java互转jpg和raw格式图片
12-13
java依赖jai-imageio，非常方便地实现jpg图片转为raw格式图片，raw格式图片转为jpg图片
在python下读取并展示raw格式的图片实例
09-19
今天小编就为大家分享一篇在python下读取并展示raw格式的图片实例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
python通过pil模块将raw图片转换成png图片的方法
09-22
主要介绍了python通过pil模块将raw图片转换成png图片的方法,实例分析了Python中pil模块的使用技巧,并Image.fromstring函数进行了较为详尽的分析说明,需要的朋友可以参考下
raw文件与jpg文件的读取/转换/显示
北落师门XY的博客
09-05 6310
⚠️cv2中为bgr，而matplotlib/imageio中为rgb ⚠️基本上基于np和cv2就可以实现了 注：.jpg到.raw 使用.tofile实现，图上忘写了 #jpg---raw import cv2 import numpy as np rgbimg=cv2.imread('rgb.jpg') print(rgbimg.dtype)#np.uint8 imgshape...
python图片raw转换成jpg
世上再无张显宗
04-13 1133
个人笔记
python3 raw 数据转换为jpg
aaronmorgan的博客
10-05 1562
我们大家都知道，sensor 直接出来的裸数据为raw 数据，没有经过编解码，压缩。 我们需要将raw数据转换为其他格式比如jpg,png,bmp 人眼才能看到。 import numpy as np import imageio rawfile = np.fromfile('C:/Users/awssome/Desktop/3.raw', dtype='uint16') # 以int16读图片 print(rawfile.shape) rawfile.shape = (1520, 2688) prin
Python实现raw文件批量转tif/png/jpg
王小帅的学习笔记
11-26 7954
用PS一个一个转格式太麻烦了，所以写了一个python批处理小工具，可以将raw文件批量转格式为tif/png/jpg… 以下代码用于转为tif格式，要转为png/jpg等格式修改注释提示的部分即可，具体的参数设置可以参考OpenCV官方文档和这篇文章 """ 用于批量处理某一文件夹下的图像文件，由raw格式转为tif/png/jpg """ import os import cv2 import numpy as np # 在这里修改存放raw文件的文件夹路径，不支持中文路径！ path = 'E:/
Python 基于Rawpy 索尼A73 RAW文件转为JPG功能实现
Tohyama的博客
05-12 736
Python课程作业，因为不限方向，而且自己平时不会保存jpg文件，所以基于rawpy写了这个自用的程序，可以实现RAW文件向JPG文件的批量转换。
python 膝关节数据raw和mhd转化为二维png，jpg等常规格式的切片
weixin_40893448的博客
03-14 307
【代码】python raw和mhd转化为二维png，jpg等常规格式的切片。
图片格式转换 .raw/.dng 转 .png/.jpg等
04-13
简单地Python代码，实现图片的格式转换，从 .raw/.dng 转 .png/.jpg等。
raw格式图片转换为bmp格式
10-13
此程序实现10位的CCD纯数据格式raw文件转换为8位的bmp格式文件，使用时需改变其中的H、W为自己图片的高和宽。
python工具 png文件转成raw
最新发布
02-29
python工具 png文件转成raw 供大家使用
Python-VoiceactivitydetectionVAD语音端点检测工具包
08-11
语音端点检测工具包，包括DNN，bDNN，LSTM和基于ACAM的VAD。 我们还提供我们直接记录的数据集。
python显示.arw相机原始数据，并保存为jpg
爱哭鬼
04-27 1470
1.安装rawpy解析相机(各种Sony,Apple等)的原始bayer数据 sudo pip install rawpy 2.解析相机数据 import rawpy import imageio import matplotlib.pylab as plt raw = rawpy.imread('image.raw') #直接调用postprocess可能出现偏色问题 rgb = raw....
python 医学图像膝关节数据raw和mhd转换为png，jpg,bmp和tif格式, plt.savefig去白边, 坐标值和轴
weixin_40893448的博客
03-11 1647
现有的软件只能将其转化为二值化或者灰度图像，这些软件仅适用于将原始数据转换为其他格式，但不适用于处理原始数据对应的彩色标签。raw和mhd转换为png，jpg,bmp和tif格式。解决在plt保存时去白边, 坐标值和轴进行保存。注意：mhd和raw这两个文件必须同时存在。当mhd和raw文件是彩色图像时。
使用ImageJ将Raw格式图片批量转换为JPG
qq_42940869的博客
12-25 951
然后选择你的图片，设置好下图的参数（这里的参数你自己应该清楚），点击OK，然后你的Raw图片就被打开了。下图2中，红框标记的部分（从“image”开始，到引号截止）即为“自动方法”中第5步的imageInfo，4，完成前2步后，如果你是windows系统，那么需要将路径中所有的反斜杠“\”改成正斜杠“/”，并且路径最后的斜杠不能省略。5，imageInfo也需要根据你的Raw图片参数进行修改，具体修改方式请参考下方的手动方法。如有疑问欢迎加博主微信w2689115745交流！
python3使用opencv读取raw格式图片并保存为bmp格式图片
热门推荐
kevin198528的专栏
10-10 1万+
import numpy as np import cv2 rawImg = np.fromfile('./test.raw', dtype=np.uint8) rawImg = rawImg.reshape(1200, 1792, 1) cv2.imwrite('./img_test.bmp', rawImg)  
raw转换成jpg代码python
07-11
将RAW图像转换为JPEG格式的代码示例如下： ```python from PIL import Image import numpy as np def raw_to_jpg(raw_path, jpg_path, width, height): # 读取RAW图像数据 with open(raw_path, 'rb') as f: raw_data = f.read() # 将RAW数据转换为numpy数组 raw_array = np.frombuffer(raw_data, dtype=np.uint8) raw_array = raw_array.reshape((height, width)) # 创建PIL图像对象 raw_image = Image.fromarray(raw_array, 'L') # 保存为JPEG格式 raw_image.save(jpg_path) # 示例用法 raw_to_jpg('input.raw', 'output.jpg', 640, 480) ``` 请确保在运行代码之前，安装了Pillow库（`pip install pillow`）来处理图像。此代码假设RAW图像是灰度图像，因此使用了`'L'`模式创建PIL图像对象。如果RAW图像包含彩色信息，你可以更改模式为`'RGB'`或`'RGBA'`来创建相应的图像对象。

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

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

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

黄金旺铺

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付：¥1
获取中
扫码支付

您的余额不足，请更换扫码支付或充值

打赏作者

实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

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

余额充值