### 形状特征提取 可以使用OpenCV库提取车辆的形状特征，如轮廓面积和周长。以下是示例代码： ```python import cv2 import numpy as np def extract_shape_features(image_path): """ 提取图像中的形状特征 :param image_path: 图像路径 :return: 轮廓面积和周长 """ # 读取图像 image = cv2.imread(image_path, cv2.IMREAD_GRAYSCALE) if image is None: raise FileNotFoundError("图像文件未找到") # 二值化处理 _, binary = cv2.threshold(image, 127, 255, cv2.THRESH_BINARY) # 查找轮廓 contours, _ = cv2.findContours(binary, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE) # 提取轮廓面积和周长 shape_features = [] for contour in contours: area = cv2.contourArea(contour) perimeter = cv2.arcLength(contour, True) shape_features.append((area, perimeter)) return shape_features # 示例 image_path = 'path_to_image.jpg' shape_features = extract_shape_features(image_path) # 输出结果 for area, perimeter in shape_features: print(f'轮廓面积: {area}, 轮廓周长: {perimeter}') ``` ### 纹理特征提取 可以使用`skimage`库提取车辆的纹理特征，如对比度。以下是示例代码： ```python from skimage.feature import greycomatrix, greycoprops from PIL import Image import numpy as np # 加载灰度图像 gray_img = Image.open('example.jpg').convert('L') gray_array = np.array(gray_img) # 计算灰度共生矩阵 glcm = greycomatrix(gray_array, [1], [0], 256, symmetric=True, normed=True) # 提取对比度特征 contrast = greycoprops(glcm, 'contrast')[0, 0] print("对比度特征值:", contrast) ``` ### 颜色特征提取 可以使用`numpy`库提取车辆的颜色特征，如颜色直方图。以下是示例代码： ```python from PIL import Image import numpy as np import matplotlib.pyplot as plt # 加载图像 img = Image.open('example.jpg') # 转换为RGB模式 rgb_img = img.convert('RGB') # 转换为numpy数组 img_array = np.array(rgb_img) # 提取颜色直方图 hist, bins = np.histogram(img_array.flatten(), bins=256, range=[0, 256]) # 显示直方图 plt.plot(hist) plt.xlim([0, 256]) plt.show() ``` ### 运动特征提取 可以使用OpenCV库的光流法来提取车辆的运动特征。以下是示例代码： ```python import cv2 import numpy as np # 读取视频 cap = cv2.VideoCapture('path_to_video.mp4') # 读取第一帧 ret, frame1 = cap.read() prvs = cv2.cvtColor(frame1, cv2.COLOR_BGR2GRAY) hsv = np.zeros_like(frame1) hsv[..., 1] = 255 while(1): ret, frame2 = cap.read() if not ret: break next = cv2.cvtColor(frame2, cv2.COLOR_BGR2GRAY) # 计算光流 flow = cv2.calcOpticalFlowFarneback(prvs, next, None, 0.5, 3, 15, 3, 5, 1.2, 0) # 计算光流的大小和方向 mag, ang = cv2.cartToPolar(flow[..., 0], flow[..., 1]) hsv[..., 0] = ang*180/np.pi/2 hsv[..., 2] = cv2.normalize(mag, None, 0, 255, cv2.NORM_MINMAX) bgr = cv2.cvtColor(hsv, cv2.COLOR_HSV2BGR) cv2.imshow('frame2', bgr) k = cv2.waitKey(30) & 0xff if k == 27: break elif k == ord('s'): cv2.imwrite('opticalfb.png', frame2) cv2.imwrite('opticalhsv.png', bgr) prvs = next cap.release() cv2.destroyAllWindows() ```