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 npdef 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_datadef 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[:, :] = alphargba_data = np.dstack((r, g, b, a))return rgba_datadef 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[:, :] = alphareturn avr_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保存到文件中。

python 读取 Raw文件生成 Numpy 数组相关推荐

三维目标检测：（一）制作自己的ModelNet数据集（1.读取pcd文件为numpy数组）
Python读取pcd文件为numpy数组 pcd文件的格式读取pcd文件存储到numpy数组中补充 pcd文件的格式 1.第二行:每行数据的内容可能是(X Y Z R G B)或者(X Y Z ...
python 读取csv文件生成散点图
废话不说,直接上图形,如下: 代码如下: # -*- coding: utf-8 -*- ''' 使用python 绘制曲线图Author: yufuchang Date: 2021-05-12 '' ...
python读取log文件生成图形表_python提取log文件内容并画出图表
之前在excel里面分析log数据,简直日了*了. 现在用python在处理日志数据. 主要涉及 matplotlib,open和循环的使用. 日志内容大致如下 2016-10-21 21:07:59 ...
python生成表格文件_python 读取excel文件生成sql文件实例详解
python 读取excel文件生成sql文件实例详解学了python这么久,总算是在工作中用到一次.这次是为了从excel文件中读取数据然后写入到数据库中.这个逻辑用java来写的话就太重了,所以 ...
python读取csv某一列存入数组_python 读取.csv文件数据到数组(矩阵)的实例讲解
利用numpy库 (缺点:有缺失值就无法读取) 读: import numpy my_matrix = numpy.loadtxt(open("1.csv","rb&qu ...
python读取txt文件的行数_python读取txt文件符合条件的行数-女性时尚流行美容健康娱乐mv-ida网...
女性时尚流行美容健康娱乐mv-ida网 mvida时尚娱乐网首页美容护肤化妆技巧发型服饰健康情感美体美食娱乐明星八卦首页 > 高级搜索 python 关系网图 pyth ...
python读取raw数据文件_在python下读取并展示raw格式的图片实例
raw文件可能有些人没有,因此,先用一张图片创建一个raw格式的文件(其实可以是其他类型的格式文件) import numpy as np import cv2 img = cv2.imread('c ...
python读取raw图片文件_在python下读取并展示raw格式的图片实例
raw文件可能有些人没有,因此,先用一张图片创建一个raw格式的文件(其实可以是其他类型的格式文件) import numpy as np import cv2 img = cv2.imread('c ...
python读取raw图片文件_python读取raw binary图片并提取统计信息的实例
python读取raw binary图片并提取统计信息的实例发布时间:2020-09-08 23:22:52 来源:脚本之家阅读:66 用python语言读取二进制图片文件,并提取非零数据统计信息 ...

python 读取 Raw文件生成 Numpy 数组

python 读取 Raw文件生成 Numpy 数组相关推荐

最新文章

热门文章

python 读取 Raw文件 生成 Numpy 数组

python 读取 Raw文件 生成 Numpy 数组相关推荐

最新文章

热门文章

python 读取 Raw文件生成 Numpy 数组

python 读取 Raw文件生成 Numpy 数组相关推荐