Skip to content

Commit 38b1d07

Browse files
benizmergify[bot]
authored andcommitted
fix: NCNN chain with images and actions
1 parent 6b5dde7 commit 38b1d07

File tree

1 file changed

+23
-0
lines changed

1 file changed

+23
-0
lines changed

src/backends/ncnn/ncnnlib.cc

Lines changed: 23 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -415,6 +415,29 @@ namespace dd
415415
out.add("multibox_rois", false);
416416
tout.finalize(ad.getobj("parameters").getobj("output"), out,
417417
static_cast<MLModel *>(&this->_mlmodel));
418+
419+
// chain compliance
420+
if (ad.has("chain") && ad.get("chain").get<bool>())
421+
{
422+
if (typeid(inputc) == typeid(ImgNCNNInputFileConn))
423+
{
424+
APIData chain_input;
425+
if (!reinterpret_cast<ImgNCNNInputFileConn *>(&inputc)
426+
->_orig_images.empty())
427+
chain_input.add("imgs",
428+
reinterpret_cast<ImgNCNNInputFileConn *>(&inputc)
429+
->_orig_images);
430+
else
431+
chain_input.add(
432+
"imgs",
433+
reinterpret_cast<ImgNCNNInputFileConn *>(&inputc)->_images);
434+
chain_input.add("imgs_size",
435+
reinterpret_cast<ImgNCNNInputFileConn *>(&inputc)
436+
->_images_size);
437+
out.add("input", chain_input);
438+
}
439+
}
440+
418441
out.add("status", 0);
419442
return 0;
420443
}

0 commit comments

Comments
 (0)