Move preprocessing out of main

2025-11-07 02:04:26 +00:00 · 2024-09-02 22:09:06 +02:00
parent 0c8ca44680
commit 0da98e2742
4 changed files with 54 additions and 29 deletions
--- a/examples/inception_v3/inception_v3.hpp
+++ b/examples/inception_v3/inception_v3.hpp
@@ -2,6 +2,10 @@
 #define INCEPTION_V3_HPP

 #include <cudanet.cuh>
+#include <vector>
+
+std::vector<float>
+readAndNormalizeImage(const std::string &imagePath, int resizeSize, int cropSize);

 class BasicConv2d : public CUDANet::Module {
  public: