資源簡介
利用隨機森林算法對圖像進行像素分類
代碼片段和文件信息
#?-*-?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
- 上一篇:python五子棋
- 下一篇:python實現(xiàn)貪吃蛇游戲
評論
共有 條評論