【发布时间】:2014-09-08 23:18:57
【问题描述】:
我正在尝试使用 this 手动标记的数据库来熟悉 CvSVM,该数据库包含 590 张图像,从 0-5 分级(0 表示模糊,5 表示完美)。如果等级 =3,我将其标记为 1(清晰)。
对于功能,我只是使用五种不同的通用指标进行模糊评估。每个都通过它们在训练数据中的均值和标准差进行标准化。同样的训练均值和标准差也用于标准化测试数据。
出于某种原因,我的 SVM 只能预测整数。我已经检查了 int casts 和其他愚蠢的错误,但无法弄清楚。我意识到我的特征可能不是很健壮,因为不同图像之间存在很大差异(标准化不是很有帮助,因为标准化测试特征的范围最终大于训练特征的范围),但仍然我觉得我应该得到一些小数预测,即使它们不准确。
培训:
// data format is [ img1 grade feature1 feature2 ... feature5, img2... ]
void train_svm(CvSVM& svm, const Mat& data)
{
CvSVMParams params;
params.svm_type = CvSVM::EPS_SVR;
params.kernel_type = CvSVM::RBF;
params.term_crit = cvTermCriteria(CV_TERMCRIT_ITER+CV_TERMCRIT_EPS, (int)1e8, FLT_EPSILON);
CvParamGrid Cgrid(.01, 100, exp(1));
CvParamGrid gammaGrid(.01, 10, exp(.05));
CvParamGrid pGrid(.01, 1.8, exp(.01));
params.C = Cgrid.min_val;
params.gamma = gammaGrid.min_val;
params.p = pGrid.min_val;
// split features from grades
Mat features = data.colRange(2, data.cols);
Mat grades = data.colRange(1, 2);
try
{
svm.train_auto(features, grades, Mat(), Mat(), params, 10,
Cgrid,
gammaGrid,
pGrid,
CvSVM::get_default_grid(CvSVM::NU),
CvSVM::get_default_grid(CvSVM::COEF),
CvSVM::get_default_grid(CvSVM::DEGREE),
false);
}
catch (Exception e)
{
params = svm.get_params();
qDebug() << params.C << params.gamma << params.p;
}
params = svm.get_params();
svm.train(features, grades, Mat(), Mat(), params);
}
测试:
void test_svm(const CvSVM& svm, const Mat& data)
{
Mat features = data.colRange(2, data.cols);
Mat grades = data.colRange(1, 2);
int num_test = features.rows;
assert(features.rows == grades.rows);
Mat results(num_test, 1, CV_32FC1);
svm.predict(features, results);
qDebug() << "Act\t\tPred";
for (int i = 0; i < num_test; i++)
{
float actual = grades.at<float>(i, 0);
float predicted = results.at<float>(i, 0);
qDebug() << actual << "\t" << predicted;
}
}
预测总是 0 或 1。没有小数。
谁能弄清楚我做错了什么?
【问题讨论】: