mirror of
https://github.com/lordmathis/CUDANet.git
synced 2025-11-06 01:34:22 +00:00
Add imagenet class map
This commit is contained in:
@@ -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
1016
include/utils/imagenet.hpp
Normal file
File diff suppressed because it is too large
Load Diff
Reference in New Issue
Block a user