1 files changed, 24 insertions, 0 deletions
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 */

diff --git a/imagenet.h b/imagenet.h new file mode 100644 index 0000000..919bb04 --- /dev/null +++ b/imagenet.h
@@ -0,0 +1,24 @@
	1	#ifndef IMAGENET_H_41BE424F
	2	#define IMAGENET_H_41BE424F
	3
	4	#include <string>
	5	#include <filesystem>
	6	#include <random>
	7	#include <tuple>
	8
	9	class imagenet {
	10	public:
	11
	12	explicit imagenet(std::string path);
	13
	14	// returns bytedata, extension
	15	std::tuple<std::string, std::string> getImageForNotion(int notion_id, std::mt19937& rng) const;
	16
	17	std::vector<std::tuple<std::string, std::string>> getImagesForNotion(int notion_id, std::mt19937& rng, int num) const;
	18
	19	private:
	20
	21	std::filesystem::path path_;
	22	};
	23
	24	#endif /* end of include guard: IMAGENET_H_41BE424F */