diff options
Diffstat (limited to 'imagenet.h')
-rw-r--r-- | imagenet.h | 22 |
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 | |||
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 | private: | ||
18 | |||
19 | std::filesystem::path path_; | ||
20 | }; | ||
21 | |||
22 | #endif /* end of include guard: IMAGENET_H_41BE424F */ | ||