裁剪并调整图像大小以训练支持向量机

时间:2018-05-07 11:57:57

标签: python opencv svm

我使用OpenCV(Python)使用HOG + SVM检测同一类的对象。我跟随Udacitiy项目的一个例子来检测车辆:访问https://github.com/TusharChugh/Vehicle-Detection-HOG/blob/master/src/vehicle-detection.ipynb。在该示例中,他们使用准备好的车辆数据集,其包含裁剪和调整大小的图像。

我问是否有人想要裁剪和调整图像大小,其中包含同一图像中的对象和背景?或者如何标记图像并将它们集成到上面提到的例子中? 我使用的是Windows 10和Python 3.6。

提前致谢

1 个答案:

答案 0 :(得分:0)

Read a negative input and convert to a Histogram of oriented gradients (HOG) 
    samples = []
    labels = []   
    for filename in glob.glob(os.path.join(negative_path, '*.jpg')):
        img = cv2.imread(filename, 1) 
        hist = hog(img)
        samples.append(hist)
        labels.append(1)

For the positive samples use the same code as above but with:
    labels.append(0)

then just train the svm modell with:
    svm.train(samples, cv2.ml.ROW_SAMPLE, labels)