Lens AI Profiler Cpp
common_types.h
1 #ifndef COMMON_TYPES_H
2 #define COMMON_TYPES_H
3 
4 #include <tuple>
5 #include <vector>
6 
7 // Define common types used in parsers
8 using YOLOOutput = std::vector<std::tuple<float, int, float, float>>; // (score, class_id, x, y)
9 using ResNetOutput = std::vector<float>; // Probabilities for each class
10 using ModelOutput = std::vector<std::pair<float, int>>; // Example definition: vector of score-class pairs
11 
12 #endif // COMMON_TYPES_H