halcon 使用svm分类

isyoungboy

于 2024-06-20 16:36:59 发布

阅读量199

点赞数 3

文章标签： svm

本文链接：https://blog.csdn.net/isyoungboy/article/details/139836745

版权

计算特征

1.选取特征
2.添加数据
3.训练
4.预测

NumFeatures:=200
DataPath:='G:/pj/test/train'
ValPath:='G:/pj/test/val'
list_files(DataPath, 'directories', Dirs)
NumClasses:=|Dirs|
NumData:=10

create_class_svm (NumFeatures, 'rbf', 0.01, 0.01, NumClasses,\
                  'one-versus-all', 'normalization', NumFeatures,\
                  SVMHandle)

for Index := 1 to NumClasses by 1
    classPath:=Dirs[Index-1]
    list_image_files(classPath, 'default', [], ImageFiles)
    for Index1 := 1 to |ImageFiles| by 1
        read_image(Image, ImageFiles[Index1-1])
        scale_image_range(Image, ImageScaled, 20, 200)
        get_domain (Image, Domain)
        calc_feature_gray_proj(Domain, ImageScaled, 'hor', NumFeatures, Feature)
        add_sample_class_svm(SVMHandle, Feature, Index-1)
    endfor
endfor


* Train the SVM
train_class_svm (SVMHandle, 0.001, 'default')

list_files(ValPath, 'directories', DirsVal)
NumVal:=|DirsVal|


for Index := 1 to NumVal by 1
    classPath:=DirsVal[Index-1]
    list_image_files(classPath, 'default', [], ImageFilesVal)
    for Index1 := 1 to |ImageFilesVal| by 1
        read_image(Image, ImageFilesVal[Index1-1])
        scale_image_range(Image, ImageScaled, 20, 200)
        get_domain (Image, Domain)
        calc_feature_gray_proj(Domain, ImageScaled, 'hor', NumFeatures, Feature)
        classify_class_svm(SVMHandle, Feature, 1, Class)
        stop()
    endfor
endfor