From 8eebfa09227abc8f22699325fd3b8a5e326a857b Mon Sep 17 00:00:00 2001
From: Josh Klontz <josh@rankone.io>
Date: Thu, 21 Jan 2016 17:12:07 -0500
Subject: [PATCH] support storing floating-point data

---
 openbr/plugins/gallery/lmdb.cpp | 14 +++++++++++++-
 1 file changed, 13 insertions(+), 1 deletion(-)

diff --git a/openbr/plugins/gallery/lmdb.cpp b/openbr/plugins/gallery/lmdb.cpp
index 84a1176..6bef3ff 100644
--- a/openbr/plugins/gallery/lmdb.cpp
+++ b/openbr/plugins/gallery/lmdb.cpp
@@ -112,7 +112,19 @@ class lmdbGallery : public Gallery
             foreach(const Template &t, working) {
                 // add current image to transaction
                 caffe::Datum datum;
-                caffe::CVMatToDatum(t.m(), &datum);
+
+                const cv::Mat &m = t.m();
+                if (m.depth() == CV_32F) {
+                    datum.set_channels(m.channels());
+                    datum.set_height(m.rows);
+                    datum.set_width(m.cols);
+                    for (int i=0; i<m.channels(); i++) // Follow Caffe's channel-major ordering convention
+                        for (int j=0; j<m.rows; j++)
+                            for (int k=0; k<m.cols; k++)
+                                datum.add_float_data(m.ptr<float>(j)[k*m.channels()+i]);
+                } else {
+                    caffe::CVMatToDatum(m, &datum);
+                }
 
                 QVariant base_label = t.file.value("Label");
                 QString label_str = base_label.toString();
--
libgit2 0.21.4