diff --git a/openbr/core/core.cpp b/openbr/core/core.cpp index 53b034c..e9782e4 100644 --- a/openbr/core/core.cpp +++ b/openbr/core/core.cpp @@ -160,7 +160,7 @@ struct AlgorithmCore data.removeAt(i); const int numFiles = data.size(); - enroll(data); + data >> *transform; g->writeBlock(data); const FileList newFiles = data.files(); @@ -184,9 +184,10 @@ struct AlgorithmCore return fileList; } - void enroll(TemplateList &data) + void enroll(Template &data) { - data >> *transform; + if (transform.isNull()) qFatal("Null transform."); + data = (data >> *transform); } void retrieveOrEnroll(const File &file, QScopedPointer &gallery, FileList &galleryFiles) @@ -376,12 +377,9 @@ FileList br::Enroll(const File &input, const File &gallery) return AlgorithmManager::getAlgorithm(gallery.get("algorithm"))->enroll(input, gallery); } -void br::Enroll(const Template &tmpl) +void br::Enroll(Template &tmpl) { - QList