cutimg.py

# -*- coding: utf-8 -*-
"""
Created on Tue Oct 30 19:03:22 2018

@author: Neoooli
"""

import numpy as np
import os
import glob
from gdaldiy import *

"""
The source file should be organized as follows by running fuseimg.py first:
sourcedir
        /10m/file1.tif...filen.tif   (4 bands 2/3/4/8)
        /20m/file1.tif...filen.tif   (6 bands 5/6/7/8A/11/12)
        /60m/file1.tif...filen.tif   (3 bands 1/9/10)
"""
 
sourcedir = '/data/whus-cd+/'#source dir  
types = ["dataset"]  
names = ['10m','20m','60m']
window_sizes, strides = [512,256,85],[512,256,85]

def cut_data(filetype, name, window_size, stride):
    filedir = sourcedir + filetype + "/" + name
    print(filedir)
    filedirs = glob.glob(os.path.join(filedir, '*'))
    for i in range(len(filedirs)):
        filepath = filedirs[i]
        print(filepath)
        savedirname = filepath.split('\\')[-1].split('.tif')[0][33:44]
        savedirpath = filedir.replace(filetype, filetype + 'DNclips') + "/" + savedirname
        if not os.path.exists(savedirpath):
            os.makedirs(savedirpath)
        img = imgread(filepath)
        h, w = img.shape[0], img.shape[1]
        h_steps = (w - window_size) // stride+1
        w_steps = (h - window_size) // stride+1
        n = 0
        for i in range(h_steps):
            high = i * stride
            for j in range(w_steps):
                width = j * stride
                n += 1
                if np.all(img[high:high+window_size, width:width+window_size] > 0):
                    imgwrite(savedirpath + "/" + str(n) + '.tif', img[high:high+window_size, width:width+window_size])

def multi_dir(filetype, name, window_size, stride):
    cut_data(filetype, name, window_size, stride)

for i in range(len(types)):
    filetype = types[i]
    for j in range(len(names)):
        multi_dir(filetype, names[j], window_sizes[j], strides[j])