Lens AI Profiler Cpp
resnet_parser.h
1 #ifndef RESNET_PARSER_H
2 #define RESNET_PARSER_H
3 
4 #include "modeloutput_parser.h"
5 #include <vector>
6 
8 public:
9  // Override the raw output parsing method for ResNet
10  std::vector<std::pair<float, int>> parseRawOutput(const void* raw_output) const override {
11  // Assuming raw_output is a std::vector<float>
12  auto raw_data = *reinterpret_cast<const std::vector<float>*>(raw_output);
13  std::vector<std::pair<float, int>> results;
14 
15  for (size_t i = 0; i < raw_data.size(); ++i) {
16  results.push_back({raw_data[i], static_cast<int>(i)});
17  }
18 
19  return results;
20  }
21 };
22 
23 #endif // RESNET_PARSER_H
Definition: modeloutput_parser.h:9
Definition: resnet_parser.h:7