更改目录下所有文件后缀+批量导入三维页岩图像

Zhaohui_Zhang

已于 2022-04-02 20:52:35 修改

阅读量2.5k

点赞数

分类专栏：三维数字岩芯模型的几何相似性判别及建模还原度评价文章标签： python

于 2022-03-19 13:21:58 首次发布

本文链接：https://blog.csdn.net/Zhaohui_Zhang/article/details/123589139

版权

三维数字岩芯模型的几何相似性判别及建模还原度评价专栏收录该内容

11 篇文章 4 订阅

订阅专栏

1、更换目录下所有文件后缀

import os
data = os.walk(r'H:/U-net/dataset/shalestone512_ihui/')
for i in data:
    for i2 in i[2]: 
        a=i2.replace('.tif','.jpg')                        //(“原格式”，“目标格式”)
        os.rename(str(i[0])+'\\'+str(i2),i[0]+'\\'+a)

2、批量导入页岩图像并生成三维矩阵

# # from tqdm import tqdm_notebook as tqdm
CT_Layer = 512
CT_Height = 512
CT_Width = 512

grayscale_images = np.zeros((CT_Layer,CT_Height,CT_Width))

for index_layer in tqdm(range(CT_Layer)):
    
    if index_layer < 10:
        Image_BMP = Image.open(f"E:/shalestone512/shalestone512/shale1000{index_layer}.bmp",'r')
    elif index_layer < 100:
        Image_BMP = Image.open(f"E:/shalestone512/shalestone512/shale100{index_layer}.bmp",'r')
    else:
        Image_BMP = Image.open(f"E:/shalestone512/shalestone512/shale10{index_layer}.bmp",'r')
    Image_Int = Image_BMP.load()

    Image_array = np.array(Image_BMP)
    
    grayscale_images[index_layer] = np.copy(Image_array)
    
print(grayscale_images.shape)

#把512张图像存储为三维矩阵
import os
image_directory = ('E:/shalestone512/shalestone512_ihui')
grayscale_images = np.zeros((512,512,512))
index_z = 0

for image_file in os.listdir(image_directory):
    gray = cv2.imread(os.path.join(image_directory, image_file), cv2.IMREAD_GRAYSCALE)#灰度化
    grayscale_images[index_z] = np.copy(gray)
    index_z += 1
print(grayscale_images.shape)