Add imagenet class map

This commit is contained in:
2024-04-28 19:03:51 +02:00
parent 1f708f3ea3
commit 3320f610db
2 changed files with 1020 additions and 1 deletions

View File

@@ -1,6 +1,7 @@
#include <conv2d.cuh>
#include <dense.cuh>
#include <iostream>
#include <imagenet.hpp>
#include <max_pooling.cuh>
#include <model.hpp>
#include <opencv2/opencv.hpp>
@@ -141,6 +142,8 @@ int main(int argc, const char *const argv[]) {
}
}
std::cout << "Prediction: " << maxIndex << std::endl;
std::string classLabel = CUDANet::Utils::IMAGENET_CLASS_MAP.at(maxIndex);
std::cout << "Prediction: " << maxIndex << " " << classLabel << std::endl;
return 0;
}

1016
include/utils/imagenet.hpp Normal file

File diff suppressed because it is too large Load Diff