颜色分类leetcode车辆检测最终输出定向梯度直方图 (HOG) 车辆与非车辆示例 我创建了一个名为 get_hog_features 的函数。经过一番研究,我发现可以使用 cv2.HOGDescriptor,并为其提供特征空间。为了获取图像的特征空间,这里有一个代码片段:


def get_feature_space(img, cspace):

    if cspace != 'RGB':

        if cspace == 'HLS':

            features = cv2.cvtColor(img, cv2.COLOR_RGB2HLS)

        elif cspace == 'YCrCb':

            features = cv2.cvtColor(img, cv2.COLOR_RGB2YCrCb)

        elif cspace == 'HSV':

            features = cv2.cvtColor(img, cv2.COLOR_RGB2HSV)

        elif cspace == 'LUV':

            features = cv2.cvtColor(img, cv2.COLOR_RGB2LUV)