From a76e81ee8bcbad87c9eb58cdb25452cae65bd0c7 Mon Sep 17 00:00:00 2001
From: Star Rauchenberger <fefferburbia@gmail.com>
Date: Fri, 23 Dec 2022 11:09:58 +0000
Subject: Bot is now a Mastodon bot

---
 imagenet.h | 24 ++++++++++++++++++++++++
 1 file changed, 24 insertions(+)
 create mode 100644 imagenet.h

(limited to 'imagenet.h')

diff --git a/imagenet.h b/imagenet.h
new file mode 100644
index 0000000..919bb04
--- /dev/null
+++ b/imagenet.h
@@ -0,0 +1,24 @@
+#ifndef IMAGENET_H_41BE424F
+#define IMAGENET_H_41BE424F
+
+#include <string>
+#include <filesystem>
+#include <random>
+#include <tuple>
+
+class imagenet {
+public:
+
+  explicit imagenet(std::string path);
+
+  // returns bytedata, extension
+  std::tuple<std::string, std::string> getImageForNotion(int notion_id, std::mt19937& rng) const;
+
+  std::vector<std::tuple<std::string, std::string>> getImagesForNotion(int notion_id, std::mt19937& rng, int num) const;
+
+private:
+
+  std::filesystem::path path_;
+};
+
+#endif /* end of include guard: IMAGENET_H_41BE424F */
-- 
cgit 1.4.1