颜色分类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)
暂无评论