about summary refs log tree commit diff stats
path: root/imagenet.h
diff options
context:
space:
mode:
authorStar Rauchenberger <fefferburbia@gmail.com>2022-12-23 11:09:58 +0000
committerStar Rauchenberger <fefferburbia@gmail.com>2022-12-23 11:09:58 +0000
commita76e81ee8bcbad87c9eb58cdb25452cae65bd0c7 (patch)
tree4f677cd679bb7374b022238d55d49dba81b7313f /imagenet.h
parent31770500a918bccf7b624262bc7d444cc2dd8a1d (diff)
downloaddifference-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.h24
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
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
17 std::vector<std::tuple<std::string, std::string>> getImagesForNotion(int notion_id, std::mt19937& rng, int num) const;
18
19private:
20
21 std::filesystem::path path_;
22};
23
24#endif /* end of include guard: IMAGENET_H_41BE424F */