commit f64b9bdf63690116118827088e6adcd65d18cc56
author Brian Silverman <bsilver16384@gmail.com> Sat Feb 29 12:51:33 2020 -0800
committer Brian Silverman <bsilver16384@gmail.com> Wed Mar 04 19:52:30 2020 -0800
tree 45f56b989749cc2bbf216cc0c2d4f071e82982b2
parent 11d6846db37fbaf591ab5d98f2bec06fdb33df49

Store descriptor coefficients in 1 byte instead of 4

Change-Id: I26ed08a552a4452de13ed8508c3c00d68d590152

y2020/vision/sift/demo_sift_training.py

diff --git a/y2020/vision/sift/demo_sift_training.py b/y2020/vision/sift/demo_sift_training.py
index 3fa33cf..c78a44a 100644
--- a/y2020/vision/sift/demo_sift_training.py
+++ b/y2020/vision/sift/demo_sift_training.py
@@ -26,7 +26,8 @@
   for keypoint, descriptor in zip(keypoints, descriptors):
     Feature.FeatureStartDescriptorVector(fbb, len(descriptor))
     for n in reversed(descriptor):
-      fbb.PrependFloat32(n)
+      assert n == round(n)
+      fbb.PrependUint8(int(round(n)))
     descriptor_vector = fbb.EndVector(len(descriptor))
 
     Feature.FeatureStart(fbb)