继上篇。
CNN特征+其他分类器
提取特征。
import gdal
from keras import Model
from keras.models import load_model
import numpy as np
import cv2
import os
import pandas as pd
model=load_model("output/classify3_vgg16_pro.h5")
# model.trainable=False
dense1_layer_model = Model(inputs=model.input,outputs=model.layers[-2].output)
dense1_layer_model.trainable=False
dense1_layer_model.summary()
dataset = gdal.Open(r"D:/test.tif")
size=128
def get_cell( pos_x, pos_y):
try:
output = []
for i in [3,2,1]:
band = dataset.GetRasterBand(i)
t = band.ReadAsArray(int(pos_x - size / 2), int(pos_y - size / 2), size, size)
output.append(t)
img2 = np.moveaxis