影像中缺失部分，其标签设为背景值0

原创于 2021-11-03 13:40:12 发布 · 287 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#python #opencv #计算机视觉

gdal（python，C++）同时被 3 个专栏收录

38 篇文章

订阅专栏

pytorch

27 篇文章

订阅专栏

遥感图像处理

19 篇文章

订阅专栏

本文介绍了一种处理影像数据的方法，通过Python脚本实现对影像中缺失部分的自动填充，并将其对应的标签设为背景类。此外，还介绍了如何为处理后的影像添加地理坐标信息。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

import numpy as np
import imageio
import cv2
from osgeo import osr, gdal
 
 
def assign_spatial_reference_byfile(src_path, dst_path):
    src_ds = gdal.Open(src_path, gdal.GA_ReadOnly)
    sr = osr.SpatialReference()
    sr.ImportFromWkt(src_ds.GetProjectionRef())
    geoTransform = src_ds.GetGeoTransform()
    dst_ds = gdal.Open(dst_path, gdal.GA_Update)
    dst_ds.SetProjection(sr.ExportToWkt())
    dst_ds.SetGeoTransform(geoTransform)
    dst_ds = None
    src_ds = None
    

def add_background(img,gt):
    # 缺失影像，标签为背景值0
    h,w,ch = img.shape
    temp = np.zeros((h,w,ch)).astype(np.uint8)
    for c,j in zip(range(ch),[0,0,0]):
        
        temp[:,:,c][img[:,:,c]==j]=1
    gt[np.sum(temp,axis=2)==3]=0
    return gt

if __name__ == '__main__':
    
    # 将影像中缺失部分，将其标签设为背景类0
    img = cv2.imread('5_288.tif')
    gt = cv2.imread('5_288_ESA_label.tif',0)
    #print(img.shape,gt.shape)
    result_gt = add_background(img,gt)
    
    imageio.imsave(r'5_288gt.tif',result_gt)
    # 添加地理坐标
    src_path = '5_288_ESA_label.tif'
    dst_path = r'5_288gt.tif'
    assign_spatial_reference_byfile(src_path, dst_path)