栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

利用windows系统把tif格式转传统三通道格式的图片

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

利用windows系统把tif格式转传统三通道格式的图片

1.首先安装GDAL地理信息空间软件,在window上很好装,在Linux系统上安装非常麻烦,所以所有图片在本地进行处理,再用numpy.save 保存成二维数据,然后上传到云服务器,用numpy.load读取,再送入神经网络进行训练

2.代码

import numpy as np
import random
import pandas as pd
from osgeo import gdal
import matplotlib.pyplot as plt
from tensorflow.keras.utils import to_categorical
#path :../data/train  ,../data/test 
def load_data(choose_path,class_num):
    #列表
    data = []#数据x
    label = []#标签y
    dataframe = pd.read_csv(choose_path)
    image_paths=dataframe.id.tolist()
    #label也是一个列表
    label=dataframe.label.tolist()
    counter=1
    for each_path in image_paths:
        #读入imge的完整路径
        print(each_path)
        #print(type(each_path))读出each_path是string类型
        print("循环数",counter)
        counter = counter+1
        #读入图像  error 96*3*3  expected:96*96*3
        image  = gdal.Open(each_path)
        #显示通道数量:3通道
        num_bands = image.RasterCount
        print("通道数",num_bands)
        #3,96,96 ,ReadAsArray()方法用于把栅格数据转换成数组
        tmp_img = image.ReadAsArray()
        print("栅格数据转换成的数组:",tmp_img.shape)
        #转置矩阵
        img = tmp_img.transpose(2, 1, 0)
        print("转置矩阵",img.shape)
        img_rgb = img[:,:,0:3]
        img_rgb = np.array(img_rgb, dtype = np.uint8)
        # 行:96 列:3 通道:3
        print("传统图像通道img_rgb:",img_rgb.shape)
        plt.imshow(img_rgb)       
        #把numpy矩阵中的整数转换成浮点数
        #image = img_to_array(img_rgb)
        data.append(img_rgb)
    #将列表转化为数组
    datas = np.array(data,dtype="float32")/255.0#归一化
    label = np.array(label)
    labels = to_categorical(label,num_classes=class_num)#one-hot
    #在本地电脑端生成二维数组文件,再放到云服务器端进行处理
    np.save(r"F:user-datanewprojectcancer_CNNprocesstestdatas.npy",datas)
    np.save(r"F:user-datanewprojectcancer_CNNprocesstestlabels.npy",labels)
    return datas,labels

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/876236.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号