随机森林分类（python代码）

# -*- coding: utf-8 -*-
import os sys time
import gdal
from osgeo import ogr
from osgeo import gdal
from osgeo import gdal_array as ga
from gdalconst import *
from skimage import morphologyfilters
import numpy as np
from numba import jit vectorize int64
import warnings 
from sklearn.tree import DecisionTreeClassifier
from sklearn.ensemble import RandomForestClassifier GradientBoostingClassifier AdaBoostClassifier
from sklearn.ensemble import ExtraTreesClassifier

def read_img（filename）:
    dataset=gdal.Open（filename）

    im_width = dataset.RasterXSize
    im_height = dataset.RasterYSize

    im_geotrans = dataset.GetGeoTransform（）
    im_proj = dataset.GetProjection（）
    im_data = dataset.ReadAsArray（00im_widthim_height）

    del dataset 
    return im_projim_geotransim_width im_heightim_data

def write_img（filename im_proj im_geotrans im_data）:
    if ‘int8‘ in im_data.dtype.name:
        datatype = gdal.GDT_Byte
    elif ‘int16‘ in i