Add messages for SIFT features
Change-Id: I05a76480ce93cc3d108624621ebe8d5a6b3c9fd6
diff --git a/y2020/vision/sift/BUILD b/y2020/vision/sift/BUILD
index d3bafc6..ec06126 100644
--- a/y2020/vision/sift/BUILD
+++ b/y2020/vision/sift/BUILD
@@ -1,4 +1,5 @@
load(":fast_gaussian.bzl", "fast_gaussian")
+load("@com_github_google_flatbuffers//:build_defs.bzl", "flatbuffer_cc_library", "flatbuffer_py_library")
cc_binary(
name = "fast_gaussian_generator",
@@ -169,3 +170,37 @@
"@com_github_google_glog//:glog",
],
)
+
+flatbuffer_py_library(
+ name = "sift_fbs_python",
+ srcs = [
+ "sift.fbs",
+ "sift_training.fbs",
+ ],
+ namespace = "frc971.vision.sift",
+ tables = [
+ "Feature",
+ "Match",
+ "ImageMatch",
+ "TransformationMatrix",
+ "CameraPose",
+ "ImageMatchResult",
+ "TrainingImage",
+ "TrainingData",
+ ],
+)
+
+flatbuffer_cc_library(
+ name = "sift_fbs",
+ srcs = ["sift.fbs"],
+ gen_reflections = True,
+ visibility = ["//visibility:public"],
+)
+
+flatbuffer_cc_library(
+ name = "sift_training_fbs",
+ srcs = ["sift_training.fbs"],
+ gen_reflections = True,
+ includes = [":sift_fbs_includes"],
+ visibility = ["//visibility:public"],
+)