Skip to content

Commit bc75e3a

Browse files
committed
fix: post rebase fixes
1 parent e3d63c1 commit bc75e3a

2 files changed

Lines changed: 4 additions & 3 deletions

File tree

packages/react-native-executorch/common/rnexecutorch/host_objects/JsiConversions.h

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,7 @@
1010
#include <rnexecutorch/host_objects/JSTensorViewOut.h>
1111
#include <rnexecutorch/jsi/OwningArrayBuffer.h>
1212

13+
#include <rnexecutorch/TypeConcepts.h>
1314
#include <rnexecutorch/models/object_detection/Constants.h>
1415
#include <rnexecutorch/models/object_detection/Utils.h>
1516

packages/react-native-executorch/common/rnexecutorch/models/object_detection/ObjectDetection.cpp

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@ ObjectDetection::ObjectDetection(
88
const std::string &modelSource,
99
std::shared_ptr<react::CallInvoker> callInvoker)
1010
: BaseModel(modelSource, callInvoker) {
11-
auto inputTensors = getInputShape();
11+
auto inputTensors = getAllInputShapes();
1212
if (inputTensors.size() == 0) {
1313
throw std::runtime_error("Model seems to not take any input tensors.");
1414
}
@@ -68,9 +68,9 @@ ObjectDetection::postprocess(const std::vector<EValue> &tensors,
6868
std::vector<Detection> ObjectDetection::forward(std::string imageSource,
6969
double detectionThreshold) {
7070
auto [inputTensor, originalSize] =
71-
imageprocessing::readImageToTensor(imageSource, getInputShape()[0]);
71+
imageprocessing::readImageToTensor(imageSource, getAllInputShapes()[0]);
7272

73-
auto forwardResult = forwardET(inputTensor);
73+
auto forwardResult = BaseModel::forward(inputTensor);
7474
if (!forwardResult.ok()) {
7575
throw std::runtime_error(
7676
"Failed to forward, error: " +

0 commit comments

Comments
 (0)