前言

将任意格式的图像转换成tiff格式
如指定了tiff_sample则采用与tiff_sample相同的投影和变换方法。

代码

save_tiff.py

import sys
import imageio
import numpy as np
from osgeo import gdaldef save_tiff(out_path, in_data, tiff_sample=None, dtype=None):"""保存tiff文件:param out_path: 保存tif文件路径:param in_data: 原始数据路径 或 数据array(h,w), array(c,h,w) 或 array(h,w,c) 要求c<h,w:param tiff_sample: tif样本路径, 用于提供投影和变换:param dtype: gdal数据类型, 默认自动根据输入识别:return:"""# 数据if isinstance(in_data, str):in_data = imageio.imread(in_data)# 调整维度if len(in_data.shape) == 2:in_data = in_data.reshape(1, *in_data.shape)dim = 1elif len(in_data.shape) == 3:if in_data.shape[0] > in_data.shape[-1]:in_data = in_data.transpose(2, 0, 1)dim = in_data.shape[0]else:raise NotImplementedError# 检测类型if dtype is None:if in_data.dtype == np.dtype('uint8'):dtype = gdal.GDT_Byteelse:dtype = gdal.GDT_Float32# 参照文件if isinstance(tiff_sample, str):tiff_sample = gdal.Open(tiff_sample)# 创建TIFF文件tiff_driver = gdal.GetDriverByName('GTiff').Create(out_path, *[i for i in in_data.shape[::-1]], dtype)if tiff_sample:tiff_driver.SetProjection(tiff_sample.GetProjection())  # 投影tiff_driver.SetGeoTransform(tiff_sample.GetGeoTransform())  # 变换# 写入数据for i in range(dim):tiff_driver.GetRasterBand(i + 1).WriteArray(in_data[i])tiff_driver.FlushCache()returnif __name__ == '__main__':save_tiff(sys.argv[1], sys.argv[2], sys.argv[3])

使用范例：

python save_tiff.py "C:\output.tiff" "C:\input.png" "C:\sample.tiff"

python 保存为tiff文件, 图像转tiff, tif文件添加投影相关推荐

python保存大列表（list）数据到文件并后续重新加载为列表（list）对象实战
python保存大列表(list)数据到文件并后续重新加载为列表(list)对象实战笔者遇到的问题是这样的, 在做机器学习模型的时候,有的时候会使用相关性分析的方法来进行特征的筛选,去除冗余特征,降 ...
python保存数据使用csv和excel哪种文件最好_Python数据持久化-csv、excel篇
2018年7月4日笔记学习目标: 1.会使用Python第三方模块操作CSV文件 2.会使用Python第三方模块操作EXCEL文件本章内容: Python操作CSV:什么是CSV.Python如 ...
python保存代码_python入门（5）使用文件编辑器编写代码并保存执行
原博文 2017-04-21 17:21 − python入门(5)使用文件编辑器编写代码并保存执行两款文本编辑器: 一个是Sublime Text,免费使用,但是不付费会弹出提示框: 一个是Not ...
易康ecognition中如何导出分好类的TIF 文件
随着地学研究中技术的改进和研究者对结果准确度要求不断提升,越来越多的研究者会用到Ecognition这个软件,用它来做分类是最常用的功能,但是初学者一般会有一个疑问,就是我分完类别之后怎么导出呢? 第 ...
使用基于phython的OpenCV库imwrite()保存tiff格式图像
目录问题描述解决方案问题描述保存.tiff图像格式时如何保证与原图像属性相同? 解决方案与图像格式无关,在调用OpenCV库imwrite()函数时,可以在第3位可缺省标志位设置输出图像的参 ...
python matplotlib模块画出的图像转换为.tiff格式
利用python matplotlib模块(python3.7.1)画出的图表暂不支持直接导出.tiff格式的图片,但是可以导出eps, pdf, pgf, png, ps, raw, rgba, s ...
保存多序列tiff文件_解码TIFF文件
0 引言最近想在Unity中加载一张TIFF图片,因为该图片存储的是海洋流场数据,所以每个像素存的是四通道的32位float,并且还采用了LZW压缩.在网上找了很多读取TIFF文件的代码,也试了下载 ...
Python地学分析 — GDAL将多个遥感图像叠加保存为tif文件
欢迎关注博主的微信公众号:"智能遥感". 该公众号将为您奉上Python地学分析.爬虫.数据分析.Web开发.机器学习.深度学习等热门源代码. 本人的GitHub代码资料主页(持续 ...
python图片保存jpg、show变成bmp_Python 实现判断图片格式并转换,将转换的图像存到生成的文件夹中...
Python 实现判断图片格式并转换,将转换的图像存到生成的文件夹中我就废话不多说了,直接上代码吧! import Image from datetime import datetime impor ...
R语言使用tiff函数将可视化图像结果保存tiff格式、使用tiff函数打开图像设备、使用dev.off函数关闭图像设备、使用compression参数自定义压缩类型（形式）
R语言使用tiff函数将可视化图像结果保存tiff格式.使用tiff函数打开图像设备.使用dev.off函数关闭图像设备.使用compression参数自定义压缩类型(形式) 目录

python 保存为tiff文件, 图像转tiff, tif文件添加投影

前言

代码

python 保存为tiff文件, 图像转tiff, tif文件添加投影相关推荐

最新文章

热门文章