diff options
author | Star Rauchenberger <fefferburbia@gmail.com> | 2022-12-23 11:09:58 +0000 |
---|---|---|
committer | Star Rauchenberger <fefferburbia@gmail.com> | 2022-12-23 11:09:58 +0000 |
commit | a76e81ee8bcbad87c9eb58cdb25452cae65bd0c7 (patch) | |
tree | 4f677cd679bb7374b022238d55d49dba81b7313f /imagenet.h | |
parent | 31770500a918bccf7b624262bc7d444cc2dd8a1d (diff) | |
download | difference-a76e81ee8bcbad87c9eb58cdb25452cae65bd0c7.tar.gz difference-a76e81ee8bcbad87c9eb58cdb25452cae65bd0c7.tar.bz2 difference-a76e81ee8bcbad87c9eb58cdb25452cae65bd0c7.zip |
Bot is now a Mastodon bot
Diffstat (limited to 'imagenet.h')
-rw-r--r-- | imagenet.h | 24 |
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 @@ | |||
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 */ | ||