added wrapped load function for python as suggested by gat3way
This commit is contained in:
@@ -1398,6 +1398,16 @@ public:
|
|||||||
Note that the train method has optional flags: ANN_MLP::TrainFlags.
|
Note that the train method has optional flags: ANN_MLP::TrainFlags.
|
||||||
*/
|
*/
|
||||||
CV_WRAP static Ptr<ANN_MLP> create();
|
CV_WRAP static Ptr<ANN_MLP> create();
|
||||||
|
|
||||||
|
/** @brief Loads and creates a serialized ANN from a file
|
||||||
|
*
|
||||||
|
* Use ANN::save to serialize and store an ANN to disk.
|
||||||
|
* Load the ANN from this file again, by calling this function with the path to the file.
|
||||||
|
*
|
||||||
|
* @param filepath path to serialized ANN
|
||||||
|
*/
|
||||||
|
CV_WRAP static Ptr<ANN_MLP> load(const String& filepath);
|
||||||
|
|
||||||
};
|
};
|
||||||
|
|
||||||
/****************************************************************************************\
|
/****************************************************************************************\
|
||||||
|
@@ -1317,6 +1317,18 @@ Ptr<ANN_MLP> ANN_MLP::create()
|
|||||||
return makePtr<ANN_MLPImpl>();
|
return makePtr<ANN_MLPImpl>();
|
||||||
}
|
}
|
||||||
|
|
||||||
}}
|
Ptr<ANN_MLP> ANN_MLP::load(const String& filepath)
|
||||||
|
{
|
||||||
|
FileStorage fs;
|
||||||
|
fs.open(filepath, FileStorage::READ);
|
||||||
|
|
||||||
|
Ptr<ANN_MLP> ann = makePtr<ANN_MLPImpl>();
|
||||||
|
|
||||||
|
((ANN_MLPImpl*)ann.get())->read(fs.getFirstTopLevelNode());
|
||||||
|
return ann;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
}}
|
||||||
|
|
||||||
/* End of file. */
|
/* End of file. */
|
||||||
|
Reference in New Issue
Block a user