|
|
@@ -1,5 +1,6 @@
|
|
|
#include "nodes/base/base.hpp"
|
|
|
#include "nodes/infer/inferNode.hpp"
|
|
|
+#include "common/image.hpp"
|
|
|
#include <unordered_map>
|
|
|
#include <random>
|
|
|
namespace Node
|
|
|
@@ -31,7 +32,7 @@ void InferNode::work()
|
|
|
// int h = rand() % (height - y);
|
|
|
// metaData->boxes.push_back(data::Box(x, y, x + w, y + h, 0.9, 0));
|
|
|
|
|
|
- auto res = model_->forward(image);
|
|
|
+ auto res = model_->forward(tensor::cvimg(image), image.cols, image.rows, 0.0f, 0.0f);;
|
|
|
for (auto& r : res)
|
|
|
{
|
|
|
metaData->boxes.push_back(r);
|
