import numpy as np from tensorflow import keras # 加载手写数字图像和标签 def load_data(): train_data = np.loadtxt('train_images.csv', delimiter=',') train_labels = np.loadtxt('train_labels.csv', delimiter=',') test_data = np.loadtxt('test_image.csv', delimiter=',') return train_data, train_labels, test_data # 数据预处理 def preprocess_data(train_data, test_data): # 归一化到 [0, 1] 范围 train_data = train_data / 255.0 test_data = test_data / 255.0 # 将数据 reshape 成适合 CNN 的输入形状 (样本数, 高度, 宽度, 通道数) train_data = train_data.reshape(-1, 28, 28, 1) test_data = test_data.reshape(-1, 28, 28, 1) return train_data, test_data # 构建 CNN 模型 def build_model(): model = keras.Sequential([ keras.layers.Conv2D(filters=32, kernel_size=(3, 3), activation='relu', input_shape=(28, 28, 1)), keras.layers.MaxPooling2D(pool_size=(2, 2)), keras.layers.Flatten(), keras.layers.Dense(units=128, activation='relu'), keras.layers.Dense(units=10, activation='softmax') ]) model.compile(optimizer='adam', loss='sparse_categorical_crossentropy', metrics=['accuracy']) return model # 进行数字识别 def recognize_digit(image, model): probabilities = model.predict(image) digit = np.argmax(probabilities) return digit # 主函数 def main(): # 加载数据 train_data, train_labels, test_data = load_data() # 数据预处理 train_data, test_data = preprocess_data(train_data, test_data) # 构建并训练模型 model = build_model() model.fit(train_data, train_labels, epochs=10, batch_size=32) # 进行数字识别 recognized_digit = recognize_digit(test_data, model) print("识别结果：", recognized_digit) if __name__ == '__main__': main()

这段代码是用于构建一个卷积神经网络（CNN）模型来进行手写数字识别。代码的流程如下：

1. 导入必要的库：numpy和tensorflow的keras模块。
2. 定义一个函数load_data()，用于加载手写数字图像和标签数据。
3. 定义一个函数preprocess_data()，用于对数据进行预处理，包括归一化和reshape操作。
4. 定义一个函数build_model()，用于构建CNN模型。
5. 定义一个函数recognize_digit()，用于进行数字识别。
6. 定义主函数main()，在其中完成数据加载、预处理、模型构建、训练和数字识别的整个流程。
7. 最后，在主函数中调用各个函数，完成手写数字识别的任务。

请注意，这段代码中使用了一些特定的数据文件（train_images.csv、train_labels.csv和test_image.csv），你需要根据实际情况提供相应的数据文件。另外，该代码假设输入的手写数字图像是28x28像素的灰度图像。