mirror/opencv
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2024-11-24 03:00:14 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

ml: update checks

Browse Source
This commit is contained in:
Alexander Alekhin 2021-04-13 11:09:14 +00:00
parent b9b19185bc
commit 0bdbc745c4
2 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
modules/ml/src/rtrees.cpp
View File

@ -479,7 +479,7 @@ public:
float predict( InputArray samples, OutputArray results, int flags ) const CV_OVERRIDE float predict( InputArray samples, OutputArray results, int flags ) const CV_OVERRIDE
{ {
CV_TRACE_FUNCTION(); CV_TRACE_FUNCTION();
CV_Assert( samples.cols() == getVarCount() && samples.type() == CV_32F ); CV_CheckEQ(samples.cols(), getVarCount(), "");
return impl.predict(samples, results, flags); return impl.predict(samples, results, flags);
} }

3
modules/ml/src/tree.cpp
View File

@ -1701,6 +1701,9 @@ void DTreesImpl::readParams( const FileNode& fn )
/*int cat_var_count = (int)fn["cat_var_count"]; /*int cat_var_count = (int)fn["cat_var_count"];
int ord_var_count = (int)fn["ord_var_count"];*/ int ord_var_count = (int)fn["ord_var_count"];*/
if (varAll <= 0)
CV_Error(Error::StsParseError, "The field \"var_all\" of DTree classifier is missing or non-positive");
FileNode tparams_node = fn["training_params"]; FileNode tparams_node = fn["training_params"];
TreeParams params0 = TreeParams(); TreeParams params0 = TreeParams();