tensorflow遥感图像分类代码

以下是使用 TensorFlow 进行遥感图像分类的示例代码：

import tensorflow as tf
import numpy as np
import os

# 定义超参数
learning_rate = 0.001
training_epochs = 50
batch_size = 128
display_step = 1

# 定义网络参数
n_input = 784  # MNIST 数据输入（图像尺寸: 28*28）
n_classes = 6  # 总共有 6 类遥感图像

# 加载数据集
def load_data():
    x_train = np.load('x_train.npy')  # 训练集数据
    y_train = np.load('y_train.npy')  # 训练集标签
    x_test = np.load('x_test.npy')    # 测试集数据
    y_test = np.load('y_test.npy')    # 测试集标签
    
    return x_train, y_train, x_test, y_test

# 创建模型
def neural_net(x):
    # 定义权重和偏置
    weights = {
        'h1': tf.Variable(tf.random_normal([n_input, 256])),
        'h2': tf.Variable(tf.random_normal([256, 256])),
        'out': tf.Variable(tf.random_normal([256, n_classes]))
    }
    biases = {
        'b1': tf.Variable(tf.random_normal([256])),
        'b2': tf.Variable(tf.random_normal([256])),
        'out': tf.Variable(tf.random_normal([n_classes]))
    }
    
    # 创建网络模型
    layer_1 = tf.add(tf.matmul(x, weights['h1']), biases['b1'])
    layer_1 = tf.nn.relu(layer_1)
    
    layer_2 = tf.add(tf.matmul(layer_1, weights['h2']), biases['b2'])
    layer_2 = tf.nn.relu(layer_2)
    
    out_layer = tf.matmul(layer_2, weights['out']) + biases['out']
    return out_layer

# 创建损失函数和优化器
def train_neural_net(x_train, y_train, x_test, y_test):
    x = tf.placeholder(tf.float32, [None, n_input])
    y = tf.placeholder(tf.float32, [None, n_classes])
    
    # 创建模型
    logits = neural_net(x)
    
    # 定义损失函数和优化器
    loss_op = tf.reduce_mean(tf.nn.softmax_cross_entropy_with_logits(logits=logits, labels=y))
    optimizer = tf.train.AdamOptimizer(learning_rate=learning_rate)
    train_op = optimizer.minimize(loss_op)
    
    # 定义评估模型的准确率的操作
    correct_pred = tf.equal(tf.argmax(logits, 1), tf.argmax(y, 1))
    accuracy = tf.reduce_mean(tf.cast(correct_pred, tf.float32))
    
    # 初始化变量
    init = tf.global_variables_initializer()
    
    with tf.Session() as sess:
        sess.run(init)
        
        for epoch in range(training_epochs):
            total_batch = int(len(x_train) / batch_size)
            for i in range(total_batch):
                batch_x = x_train[i*batch_size:(i+1)*batch_size]
                batch_y = y_train[i*batch_size:(i+1)*batch_size]
                
                # 训练模型
                sess.run(train_op, feed_dict={x: batch_x, y: batch_y})
                
            # 每一轮结束后计算训练集和测试集上的准确率
            train_acc = sess.run(accuracy, feed_dict={x: x_train, y: y_train})
            test_acc = sess.run(accuracy, feed_dict={x: x_test, y: y_test})
            
            # 输出训练过程
            if epoch % display_step == 0:
                print("Epoch:", epoch+1, "Train acc:", train_acc, "Test acc:", test_acc)
        
        print("Optimization Finished!")
        
        # 计算最终模型在测试集上的准确率
        test_acc = sess.run(accuracy, feed_dict={x: x_test, y: y_test})
        print("Test Accuracy:", test_acc)
        
if __name__ == '__main__':
    x_train, y_train, x_test, y_test = load_data()
    train_neural_net(x_train, y_train, x_test, y_test)

在这个示例代码中，我们使用了一个基于多层感知器的神经网络来进行遥感图像分类。数据集是一个由 Numpy 数组保存的遥感图像数据集，其中训练集和测试集的数据和标签分别保存在 `x_train.npy`、`y_train.npy`、`x_test.npy` 和 `y_test.npy` 文件中。

代码首先定义了模型的超参数和网络参数，然后定义了一个函数 `load_data()` 来加载数据集。接着，代码定义了神经网络模型的结构，包括输入层、两个隐藏层和输出层。然后，我们定义了损失函数和优化器，并使用训练集来训练模型，并在每一轮结束后计算训练集和测试集上的准确率。最后，我们计算最终模型在测试集上的准确率。