blob: 919bb044438fe7a457a6395e3024fda66f5e8764 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
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 */
|
