summary refs log tree commit diff stats
path: root/imagenet.h
diff options
context:
space:
mode:
Diffstat (limited to 'imagenet.h')
-rw-r--r--imagenet.h22
1 files changed, 22 insertions, 0 deletions
diff --git a/imagenet.h b/imagenet.h new file mode 100644 index 0000000..e2c57de --- /dev/null +++ b/imagenet.h
@@ -0,0 +1,22 @@
1#ifndef IMAGENET_H_41BE424F
2#define IMAGENET_H_41BE424F
3
4#include <string>
5#include <filesystem>
6#include <random>
7#include <tuple>
8
9class imagenet {
10public:
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
17private:
18
19 std::filesystem::path path_;
20};
21
22#endif /* end of include guard: IMAGENET_H_41BE424F */