summary refs log tree commit diff stats
path: root/imagenet.h
blob: e2c57de23a26afe265f1b20b4857ff417d286a37 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
#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;

private:

  std::filesystem::path path_;
};

#endif /* end of include guard: IMAGENET_H_41BE424F */