Fix Json input with new transition data structure

ircam-ismm · Apr 27, 2016 · b98d8ee · b98d8ee
1 parent 1865407
commit b98d8ee
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/models/hierarchical_hmm.cpp b/src/models/hierarchical_hmm.cpp
@@ -931,6 +931,7 @@ void xmm::HierarchicalHMM::from_json(JSONNode root) {
                 "Wrong type for node 'prior': was expecting "
                 "'JSON_ARRAY'",
                 root_it->name());
+        prior.resize(numModels);
         json2vector(*root_it, prior, numModels);
 
         // Get High-level Exit Probabilities
@@ -942,6 +943,7 @@ void xmm::HierarchicalHMM::from_json(JSONNode root) {
                 "Wrong type for node 'exit': was expecting "
                 "'JSON_ARRAY'",
                 root_it->name());
+        exitTransition.resize(numModels);
         json2vector(*root_it, exitTransition, numModels);
 
         // Get High-level Transition Matrix
@@ -956,9 +958,11 @@ void xmm::HierarchicalHMM::from_json(JSONNode root) {
         std::vector<double> trans(numModels * numModels);
         json2vector(*root_it, trans, numModels * numModels);
         transition.resize(numModels);
-        for (int i = 0; i < numModels; i++)
+        for (int i = 0; i < numModels; i++) {
+            transition[i].resize(numModels);
             for (int j = 0; j < numModels; j++)
                 transition[i][j] = trans[i * numModels + j];
+        }
 
     } catch (JSONException& e) {
         throw JSONException(e, root.name());