### Python机器视觉人体检测模型训练示例 对于使用Python进行机器视觉中的人体检测模型训练，可以采用多种框架和技术栈。下面提供了一个基于`lightgbm`的轻量级梯度提升决策树(LightGBM)用于分类任务的例子，不过针对人体检测通常会更倾向于深度学习的方法如YOLO、SSD等目标检测算法。这里为了贴合需求给出一个简化版本的目标检测流程，假设已经有一个标注好的人体图片数据集。 #### 准备工作 安装必要的库： ```bash pip install opencv-python-headless tensorflow keras numpy matplotlib scikit-image albumentations imgaug ``` 加载所需模块并定义一些辅助函数： ```python import os from sklearn.model_selection import train_test_split import cv2 import numpy as np import matplotlib.pyplot as plt from skimage.io import imread from skimage.transform import resize from tensorflow.keras.preprocessing.image import ImageDataGenerator from tensorflow.keras.models import Sequential from tensorflow.keras.layers import Conv2D, MaxPooling2D, Flatten, Dense, Dropout from tensorflow.keras.optimizers import Adam from tensorflow.keras.callbacks import EarlyStopping ``` 读取和预处理图像数据： ```python def load_data(data_dir): images = [] labels = [] for label_name in sorted(os.listdir(data_dir)): path = os.path.join(data_dir, label_name) if not os.path.isdir(path): continue for image_file in os.listdir(path)[:100]: # 只取前100张作为样例 try: img_path = os.path.join(path, image_file) img = imread(img_path) resized_img = resize(img, (64, 64)) images.append(resized_img) labels.append(int(label_name == 'person')) # 假设文件夹名为'person' except Exception as e: print(f"Error loading {img_path}: ", str(e)) X = np.array(images).astype('float32') Y = np.array(labels).astype('uint8') return X, Y data_directory = './path_to_your_dataset/' # 替换为实际路径 X, y = load_data(data_directory) x_train, x_val, y_train, y_val = train_test_split(X, y, test_size=0.2, random_state=42) ``` 创建卷积神经网络(CNN): ```python model = Sequential([ Conv2D(32, kernel_size=(3, 3), activation='relu', input_shape=(64, 64, 3)), MaxPooling2D(pool_size=(2, 2)), Conv2D(64, kernel_size=(3, 3), activation='relu'), MaxPooling2D(pool_size=(2, 2)), Flatten(), Dense(128, activation='relu'), Dropout(rate=0.5), Dense(units=1, activation='sigmoid') # 输出层只有一个节点表示二元分类 ]) optimizer = Adam(lr=0.001) loss_function = "binary_crossentropy" metrics_list = ['accuracy'] model.compile(optimizer=optimizer, loss=loss_function, metrics=metrics_list) ``` 设置早停机制防止过拟合，并启动训练过程: ```python early_stopping_monitor = EarlyStopping( monitor="val_loss", patience=5, restore_best_weights=True ) history = model.fit(x=x_train, y=y_train, epochs=20, batch_size=32, validation_data=(x_val, y_val), callbacks=[early_stopping_monitor]) ``` 绘制损失曲线图以便观察收敛情况： ```python plt.plot(history.history['loss'], label='Training Loss') plt.plot(history.history['val_loss'], label='Validation Loss') plt.title('Loss Curve') plt.xlabel('Epochs') plt.ylabel('Loss Value') plt.legend() plt.show() plt.plot(history.history['accuracy'], label='Training Accuracy') plt.plot(history.history['val_accuracy'], label='Validation Accuracy') plt.title('Accuracy Curve') plt.xlabel('Epochs') plt.ylabel('Accuracy Rate') plt.legend() plt.show() ``` 保存训练完成后的模型权重供后续部署或推理阶段调用： ```python save_path = "./human_detection_cnn.h5" model.save(save_path) print(f'Model saved to {save_path}') ``` 上述代码展示了如何利用Keras搭建简单CNN架构来进行人体检测的任务[^1]。需要注意的是这只是一个基础示范，在真实项目里还需要考虑更多因素比如数据增强、超参数调整以及性能优化等方面的工作。