遥感影像读写

from osgeo import gdal_array as ga

from osgeo import gdal

from glob import glob

from PIL import Image

import numpy as np

import matplotlib.pyplot as plt

import cv2

 

# file_dir = 'D:\\ly\\Data\\190805\\msi\\*.tif'

# files = glob(file_dir)

mask_dir = 'D:\\ly\\coding\\venv\\SVM_class\\dataset\\*.jpg'

masks = glob(mask_dir)

out_path = 'D:\\ly\\coding\\venv\\SVM_class\\dataset\\riverway\\'

for ith_file in range(len(files)+1):

    files =  glob('D:\\ly\\Data\\190805\\msi\\MSI0'+str(2394+ith_file)+'.tif')

    file = files[0]

    name = file.split('\\')[-1]

    mask = masks[ith_file]

    mask = cv2.imread(mask)[:,:,0]

    arr_data = ga.LoadFile(file)

    arr0 = np.multiply(arr_data[0], mask)

    arr1 = np.multi