python 读取 Raw文件 生成 Numpy 数组

一般情况下,有python的三方库的软件,可以读RGB二进制的Raw数据,生成Numpy数据。比如说OpenCV之类的。但是,我们也可以使用python的裸代码,得到Numpy。


知识点:

  • numpy.frombuffer 将bytes格式转换为numpy ndarray
  • numpy.reshape 可以将一维的numpy数组转换成三维的数组
  • numpy.dstack 可以将三维的RBA数据扩充成RGBA 数据
  • numpy.concatenate 可以扩充到更高维度的numpy 数据

代码例子如下:

#!/usr/bin/env python
# encoding: utf-8
import os
import numpy as np


def read_raw_to_rgb(raw_path, width, height, channel):
    """ 读取raw数据转换成 RGB 的numpy 格式的数据 """
    assert os.path.exists(raw_path), r'输入的文件必须存在'
    raw_data = open(raw_path, 'rb').read()
    raw_data_shape = (width, height, channel)
    np_raw_data = np.frombuffer(raw_data, dtype=np.uint8)
    reshaped_data = np_raw_data.reshape(raw_data_shape)
    return reshaped_data


def convert_rgb_to_rgba(rgb_data, alpha):
    """ 将 RGB 数据扩充成 RGBA 的数据"""
    r = rgb_data[:, :, 0]
    g = rgb_data[:, :, 1]
    b = rgb_data[:, :, 2]
    a = np.zeros(r.shape, np.uint8)
    a[:, :] = alpha
    rgba_data = np.dstack((r, g, b, a))
    return rgba_data


def create_alpha_data_according_rgb_data(rgb_data, alpha):
    """ 根据rgb_data的维度创建alpha_data"""
    r = rgb_data[:, :, 0]
    a = np.zeros(r.shape, np.double)
    a[:, :] = alpha
    return a

    vr_path = r'D:\temp\result\vr_512_512.raw'
    sr_path = r'D:\temp\result\sr_512_512.raw'

    vr_rgb_data = read_raw_to_rgb(vr_path, 512, 512, 3)
    vr_alpha_data = create_alpha_data_according_rgb_data(vr_rgb_data, 0.1)

    sr_rgb_data = read_raw_to_rgb(sr_path, 512, 512, 3)
    sr_alpha_data = create_alpha_data_according_rgb_data(sr_rgb_data, 0.9)
    print(512*512*4)

    vr_rgb_data_image_idx = vr_rgb_data[np.newaxis, :, :, :]
    sr_rgb_data_image_idx = sr_rgb_data[np.newaxis, :, :, :]
    rgb_data_array = np.concatenate((vr_rgb_data_image_idx, sr_rgb_data_image_idx), axis=0)
    print(rgb_data_array.shape)

    vr_alpha_data_image_idx = vr_alpha_data[np.newaxis, :, :]
    sr_alpha_data_image_idx = sr_alpha_data[np.newaxis, :, :]
    alpha_data_array = np.concatenate((vr_alpha_data_image_idx, sr_alpha_data_image_idx), axis=0)
    print(alpha_data_array.shape)

测试环境中的图像,可以使用 OpenCV读取一张图, 将numpy保存到文件中。

  • 1
    点赞
  • 16
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
要使用Python读取raw文件,可以使用numpy库中的fromfile函数读取文件,并使用reshape函数将数据重新排列。以下是一个示例代码: ```python import numpy as np def readRaw(path): imgData = np.fromfile(path, dtype='uint8') rows = 512 # 图像的行数 cols = 640 # 图像的列数 channels = 1 # 图像的通道数,灰度图为1 imgData = imgData.reshape(rows, cols, channels) return imgData path = 'path_to_your_raw_file.raw' img = readRaw(path) ``` 在上述代码中,readRaw函数接受一个文件路径作为参数,并使用numpy的fromfile函数读取raw文件。然后,通过reshape函数将数据重新排列成图片的形状,最后返回读取到的图像数据。 请注意,上述代码是一个示例,你需要将'path_to_your_raw_file.raw'替换为你要读取的实际raw文件的路径。<span class="em">1</span><span class="em">2</span><span class="em">3</span> #### 引用[.reference_title] - *1* [Python——读取raw格式文件](https://blog.csdn.net/weixin_42686816/article/details/126874854)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v92^chatsearchT3_1"}}] [.reference_item style="max-width: 33.333333333333336%"] - *2* [lena.raw图片文件下载及打开方式](https://blog.csdn.net/lic1697067085/article/details/116353553)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v92^chatsearchT3_1"}}] [.reference_item style="max-width: 33.333333333333336%"] - *3* [RAW图像详解及使用Python读取raw格式图像并显示](https://blog.csdn.net/m0_46378271/article/details/126090531)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v92^chatsearchT3_1"}}] [.reference_item style="max-width: 33.333333333333336%"] [ .reference_list ]
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值