react-native-yolo 0.0.5 → 0.0.8

package/ios/Utils/Yuv420ToNv12Converter.swift

@@ -0,0 +1,71 @@
+import Foundation
+import AVFoundation
+import NitroModules
+import VisionCamera
+import CoreVideo
+
+public enum Yuv420ToNv12Converter {
+    public static func convert(frame: any HybridFrameSpec, width: Int, height: Int) -> [UInt8] {
+        do {
+            let nativeBuffer = try frame.getNativeBuffer()
+            guard let rawPointer = nativeBuffer.pointer else { return [] }
+
+            let pixelBuffer = Unmanaged<CVPixelBuffer>
+                .fromOpaque(rawPointer)
+                .takeUnretainedValue()
+
+            CVPixelBufferLockBaseAddress(pixelBuffer, .readOnly)
+            defer {
+                CVPixelBufferUnlockBaseAddress(pixelBuffer, .readOnly)
+            }
+
+            guard
+                let yBaseAddress = CVPixelBufferGetBaseAddressOfPlane(pixelBuffer, 0),
+                let uvBaseAddress = CVPixelBufferGetBaseAddressOfPlane(pixelBuffer, 1)
+            else {
+                return []
+            }
+
+            let yRowStride = CVPixelBufferGetBytesPerRowOfPlane(pixelBuffer, 0)
+            let uvRowStride = CVPixelBufferGetBytesPerRowOfPlane(pixelBuffer, 1)
+
+            let ySize = width * height
+            let uvSize = ySize / 2
+
+            var nv12 = [UInt8](repeating: 0, count: ySize + uvSize)
+
+            nv12.withUnsafeMutableBytes { dstBuffer in
+                guard let dstBase = dstBuffer.baseAddress else { return }
+
+                let ySrc = yBaseAddress.assumingMemoryBound(to: UInt8.self)
+                let uvSrc = uvBaseAddress.assumingMemoryBound(to: UInt8.self)
+                let dst = dstBase.assumingMemoryBound(to: UInt8.self)
+
+                // 1. Copy Y Plane row-by-row to safely discard padding/stride
+                for row in 0..<height {
+                    memcpy(
+                        dst.advanced(by: row * width),
+                        ySrc.advanced(by: row * yRowStride),
+                        width
+                    )
+                }
+
+                // 2. Copy UV Plane row-by-row safely
+                let uvDstStart = ySize
+                let chromaHeight = height / 2
+
+                for row in 0..<chromaHeight {
+                    memcpy(
+                        dst.advanced(by: uvDstStart + (row * width)), // Fixed layout math
+                        uvSrc.advanced(by: row * uvRowStride),
+                        width
+                    )
+                }
+            }
+
+            return nv12
+        } catch {
+            return []
+        }
+    }
+}

package/lib/commonjs/index.js

@@ -7,12 +7,13 @@ exports.Yolo = void 0;
 var _reactNativeNitroModules = require("react-native-nitro-modules");
 var _reactNative = require("react-native");
 const NativeYolo = _reactNativeNitroModules.NitroModules.createHybridObject('Yolo');
-const Yolo = exports.Yolo = Object.assign(NativeYolo, {
-  loadModelTest(modelAssetId) {
+const Yolo = exports.Yolo = {
+  ...NativeYolo,
+  loadModel(modelAssetId) {
     const {
       uri
     } = _reactNative.Image.resolveAssetSource(modelAssetId);
     return NativeYolo.loadModel(uri);
   }
-});
+};
 //# sourceMappingURL=index.js.map

package/lib/commonjs/index.js.map

@@ -1 +1 @@
-{"version":3,"names":["_reactNativeNitroModules","require","_reactNative","NativeYolo","NitroModules","createHybridObject","Yolo","exports","Object","assign","loadModelTest","modelAssetId","uri","Image","resolveAssetSource","loadModel"],"sourceRoot":"..\\..\\src","sources":["index.ts"],"mappings":";;;;;;AAAA,IAAAA,wBAAA,GAAAC,OAAA;AAEA,IAAAC,YAAA,GAAAD,OAAA;AAEA,MAAME,UAAU,GAAGC,qCAAY,CAACC,kBAAkB,CAAW,MAAM,CAAC;AAE7D,MAAMC,IAAI,GAAAC,OAAA,CAAAD,IAAA,GAAGE,MAAM,CAACC,MAAM,CAACN,UAAU,EAAE;EAC5CO,aAAaA,CAACC,YAAoB,EAAC;IACjC,MAAM;MAAEC;IAAI,CAAC,GAAGC,kBAAK,CAACC,kBAAkB,CAACH,YAAY,CAAC;IACtD,OAAOR,UAAU,CAACY,SAAS,CAACH,GAAG,CAAC;EAClC;AACF,CAAC,CAAC","ignoreList":[]}
+{"version":3,"names":["_reactNativeNitroModules","require","_reactNative","NativeYolo","NitroModules","createHybridObject","Yolo","exports","loadModel","modelAssetId","uri","Image","resolveAssetSource"],"sourceRoot":"..\\..\\src","sources":["index.ts"],"mappings":";;;;;;AAAA,IAAAA,wBAAA,GAAAC,OAAA;AAEA,IAAAC,YAAA,GAAAD,OAAA;AAEA,MAAME,UAAU,GAAGC,qCAAY,CAACC,kBAAkB,CAAW,MAAM,CAAC;AAG7D,MAAMC,IAAI,GAAAC,OAAA,CAAAD,IAAA,GAAG;EAClB,GAAGH,UAAU;EAEbK,SAASA,CAACC,YAAoB,EAAE;IAC9B,MAAM;MAAEC;IAAI,CAAC,GAAGC,kBAAK,CAACC,kBAAkB,CAACH,YAAY,CAAC;IACtD,OAAON,UAAU,CAACK,SAAS,CAACE,GAAG,CAAC;EAClC;AACF,CAAC","ignoreList":[]}

package/lib/module/index.js

@@ -3,12 +3,13 @@
 import { NitroModules } from 'react-native-nitro-modules';
 import { Image } from 'react-native';
 const NativeYolo = NitroModules.createHybridObject('Yolo');
-export const Yolo = Object.assign(NativeYolo, {
-  loadModelTest(modelAssetId) {
+export const Yolo = {
+  ...NativeYolo,
+  loadModel(modelAssetId) {
     const {
       uri
     } = Image.resolveAssetSource(modelAssetId);
     return NativeYolo.loadModel(uri);
   }
-});
+};
 //# sourceMappingURL=index.js.map

package/lib/module/index.js.map

@@ -1 +1 @@
-{"version":3,"names":["NitroModules","Image","NativeYolo","createHybridObject","Yolo","Object","assign","loadModelTest","modelAssetId","uri","resolveAssetSource","loadModel"],"sourceRoot":"..\\..\\src","sources":["index.ts"],"mappings":";;AAAA,SAASA,YAAY,QAAQ,4BAA4B;AAEzD,SAASC,KAAK,QAAQ,cAAc;AAEpC,MAAMC,UAAU,GAAGF,YAAY,CAACG,kBAAkB,CAAW,MAAM,CAAC;AAEpE,OAAO,MAAMC,IAAI,GAAGC,MAAM,CAACC,MAAM,CAACJ,UAAU,EAAE;EAC5CK,aAAaA,CAACC,YAAoB,EAAC;IACjC,MAAM;MAAEC;IAAI,CAAC,GAAGR,KAAK,CAACS,kBAAkB,CAACF,YAAY,CAAC;IACtD,OAAON,UAAU,CAACS,SAAS,CAACF,GAAG,CAAC;EAClC;AACF,CAAC,CAAC","ignoreList":[]}
+{"version":3,"names":["NitroModules","Image","NativeYolo","createHybridObject","Yolo","loadModel","modelAssetId","uri","resolveAssetSource"],"sourceRoot":"..\\..\\src","sources":["index.ts"],"mappings":";;AAAA,SAASA,YAAY,QAAQ,4BAA4B;AAEzD,SAASC,KAAK,QAAQ,cAAc;AAEpC,MAAMC,UAAU,GAAGF,YAAY,CAACG,kBAAkB,CAAW,MAAM,CAAC;AAGpE,OAAO,MAAMC,IAAI,GAAG;EAClB,GAAGF,UAAU;EAEbG,SAASA,CAACC,YAAoB,EAAE;IAC9B,MAAM;MAAEC;IAAI,CAAC,GAAGN,KAAK,CAACO,kBAAkB,CAACF,YAAY,CAAC;IACtD,OAAOJ,UAAU,CAACG,SAAS,CAACE,GAAG,CAAC;EAClC;AACF,CAAC","ignoreList":[]}

package/lib/typescript/src/index.d.ts

@@ -1,5 +1,14 @@
-import type { Yolo as YoloSpec } from './specs/yolo.nitro';
-export declare const Yolo: YoloSpec & {
-    loadModelTest(modelAssetId: number): void;
+export declare const Yolo: {
+    loadModel(modelAssetId: number): import("./specs/yolo.nitro").YoloModel;
+    frameToBase64(frame: import("react-native-vision-camera").Frame): string;
+    __type?: string;
+    name: string;
+    toString(): string;
+    equals(other: import("react-native-nitro-modules").HybridObject<{
+        ios: "swift";
+        android: "kotlin";
+    }>): boolean;
+    dispose(): void;
 };
+export type { Detection, BoundingBox, YoloModel, } from './specs/yolo.nitro';
 //# sourceMappingURL=index.d.ts.map

package/lib/typescript/src/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/index.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,IAAI,IAAI,QAAQ,EAAE,MAAM,oBAAoB,CAAA;AAK1D,eAAO,MAAM,IAAI;gCACa,MAAM;CAIlC,CAAA"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/index.ts"],"names":[],"mappings":"AAOA,eAAO,MAAM,IAAI;4BAGS,MAAM;;;;;;;;;;CAI/B,CAAA;AACD,YAAY,EACV,SAAS,EACT,WAAW,EACX,SAAS,GACV,MAAM,oBAAoB,CAAA"}

package/lib/typescript/src/specs/yolo.nitro.d.ts

@@ -1,24 +1,28 @@
 import type { HybridObject } from 'react-native-nitro-modules';
 import type { Frame } from 'react-native-vision-camera';
-type Detection = {
+export type Detection = {
     classId: number;
     score: number;
     boundingBox: BoundingBox;
 };
-type BoundingBox = {
+export type BoundingBox = {
     x1: number;
     y1: number;
     x2: number;
     y2: number;
 };
+export interface YoloModel extends HybridObject<{
+    ios: 'swift';
+    android: 'kotlin';
+}> {
+    detect(frame: Frame): Detection[];
+    close(): void;
+}
 export interface Yolo extends HybridObject<{
     ios: 'swift';
     android: 'kotlin';
 }> {
-    sum(num1: number, num2: number): number;
-    loadModel(modelPath: string): void;
+    loadModel(modelPath: string): YoloModel;
     frameToBase64(frame: Frame): string;
-    detect(frame: Frame): Detection[];
 }
-export {};
 //# sourceMappingURL=yolo.nitro.d.ts.map

package/lib/typescript/src/specs/yolo.nitro.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"yolo.nitro.d.ts","sourceRoot":"","sources":["../../../../src/specs/yolo.nitro.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,4BAA4B,CAAA;AAC9D,OAAO,KAAK,EAAE,KAAK,EAAE,MAAM,4BAA4B,CAAA;AAEvD,KAAK,SAAS,GAAG;IACf,OAAO,EAAE,MAAM,CAAA;IACf,KAAK,EAAE,MAAM,CAAA;IACb,WAAW,EAAE,WAAW,CAAA;CACzB,CAAA;AACD,KAAK,WAAW,GAAG;IACjB,EAAE,EAAE,MAAM,CAAA;IACV,EAAE,EAAE,MAAM,CAAA;IACV,EAAE,EAAE,MAAM,CAAA;IACV,EAAE,EAAE,MAAM,CAAA;CACX,CAAA;AAED,MAAM,WAAW,IAAK,SAAQ,YAAY,CAAC;IACzC,GAAG,EAAE,OAAO,CAAA;IACZ,OAAO,EAAE,QAAQ,CAAA;CAClB,CAAC;IACA,GAAG,CAAC,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,GAAG,MAAM,CAAA;IACvC,SAAS,CAAC,SAAS,EAAE,MAAM,GAAG,IAAI,CAAA;IAClC,aAAa,CAAC,KAAK,EAAE,KAAK,GAAG,MAAM,CAAA;IACnC,MAAM,CAAC,KAAK,EAAE,KAAK,GAAG,SAAS,EAAE,CAAA;CAClC"}
+{"version":3,"file":"yolo.nitro.d.ts","sourceRoot":"","sources":["../../../../src/specs/yolo.nitro.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,4BAA4B,CAAA;AAC9D,OAAO,KAAK,EAAE,KAAK,EAAE,MAAM,4BAA4B,CAAA;AAEvD,MAAM,MAAM,SAAS,GAAG;IACtB,OAAO,EAAE,MAAM,CAAA;IACf,KAAK,EAAE,MAAM,CAAA;IACb,WAAW,EAAE,WAAW,CAAA;CACzB,CAAA;AACD,MAAM,MAAM,WAAW,GAAG;IACxB,EAAE,EAAE,MAAM,CAAA;IACV,EAAE,EAAE,MAAM,CAAA;IACV,EAAE,EAAE,MAAM,CAAA;IACV,EAAE,EAAE,MAAM,CAAA;CACX,CAAA;AAED,MAAM,WAAW,SAAU,SAAQ,YAAY,CAAC;IAC9C,GAAG,EAAE,OAAO,CAAA;IACZ,OAAO,EAAE,QAAQ,CAAA;CAClB,CAAC;IACA,MAAM,CAAC,KAAK,EAAE,KAAK,GAAG,SAAS,EAAE,CAAA;IACjC,KAAK,IAAI,IAAI,CAAA;CAEd;AAED,MAAM,WAAW,IAAK,SAAQ,YAAY,CAAC;IACzC,GAAG,EAAE,OAAO,CAAA;IACZ,OAAO,EAAE,QAAQ,CAAA;CAClB,CAAC;IACA,SAAS,CAAC,SAAS,EAAE,MAAM,GAAG,SAAS,CAAA;IACvC,aAAa,CAAC,KAAK,EAAE,KAAK,GAAG,MAAM,CAAA;CACpC"}

package/nitrogen/generated/android/Yolo+autolinking.cmake

@@ -33,8 +33,10 @@ target_sources(
   # Autolinking Setup
   ../nitrogen/generated/android/YoloOnLoad.cpp
   # Shared Nitrogen C++ sources
+  ../nitrogen/generated/shared/c++/HybridYoloModelSpec.cpp
   ../nitrogen/generated/shared/c++/HybridYoloSpec.cpp
   # Android-specific Nitrogen C++ sources
+  ../nitrogen/generated/android/c++/JHybridYoloModelSpec.cpp
   ../nitrogen/generated/android/c++/JHybridYoloSpec.cpp
 )
 

package/nitrogen/generated/android/YoloOnLoad.cpp

@@ -15,6 +15,7 @@
 #include <fbjni/fbjni.h>
 #include <NitroModules/HybridObjectRegistry.hpp>
 
+#include "JHybridYoloModelSpec.hpp"
 #include "JHybridYoloSpec.hpp"
 #include <NitroModules/DefaultConstructableObject.hpp>
 
@@ -40,6 +41,7 @@ void registerAllNatives() {
   using namespace margelo::nitro::yolo;
 
   // Register native JNI methods
+  margelo::nitro::yolo::JHybridYoloModelSpec::CxxPart::registerNatives();
   margelo::nitro::yolo::JHybridYoloSpec::CxxPart::registerNatives();
 
   // Register Nitro Hybrid Objects

package/nitrogen/generated/android/c++/JHybridYoloModelSpec.cpp

@@ -0,0 +1,78 @@
+///
+/// JHybridYoloModelSpec.cpp
+/// This file was generated by nitrogen. DO NOT MODIFY THIS FILE.
+/// https://github.com/mrousavy/nitro
+/// Copyright © Marc Rousavy @ Margelo
+///
+
+#include "JHybridYoloModelSpec.hpp"
+
+// Forward declaration of `Detection` to properly resolve imports.
+namespace margelo::nitro::yolo { struct Detection; }
+// Forward declaration of `BoundingBox` to properly resolve imports.
+namespace margelo::nitro::yolo { struct BoundingBox; }
+// Forward declaration of `HybridFrameSpec` to properly resolve imports.
+namespace margelo::nitro::camera { class HybridFrameSpec; }
+
+#include "Detection.hpp"
+#include <vector>
+#include "JDetection.hpp"
+#include "BoundingBox.hpp"
+#include "JBoundingBox.hpp"
+#include <memory>
+#include <VisionCamera/HybridFrameSpec.hpp>
+#include <VisionCamera/JHybridFrameSpec.hpp>
+
+namespace margelo::nitro::yolo {
+
+  std::shared_ptr<JHybridYoloModelSpec> JHybridYoloModelSpec::JavaPart::getJHybridYoloModelSpec() {
+    auto hybridObject = JHybridObject::JavaPart::getJHybridObject();
+    auto castHybridObject = std::dynamic_pointer_cast<JHybridYoloModelSpec>(hybridObject);
+    if (castHybridObject == nullptr) [[unlikely]] {
+      throw std::runtime_error("Failed to downcast JHybridObject to JHybridYoloModelSpec!");
+    }
+    return castHybridObject;
+  }
+
+  jni::local_ref<JHybridYoloModelSpec::CxxPart::jhybriddata> JHybridYoloModelSpec::CxxPart::initHybrid(jni::alias_ref<jhybridobject> jThis) {
+    return makeCxxInstance(jThis);
+  }
+
+  std::shared_ptr<JHybridObject> JHybridYoloModelSpec::CxxPart::createHybridObject(const jni::local_ref<JHybridObject::JavaPart>& javaPart) {
+    auto castJavaPart = jni::dynamic_ref_cast<JHybridYoloModelSpec::JavaPart>(javaPart);
+    if (castJavaPart == nullptr) [[unlikely]] {
+      throw std::runtime_error("Failed to cast JHybridObject::JavaPart to JHybridYoloModelSpec::JavaPart!");
+    }
+    return std::make_shared<JHybridYoloModelSpec>(castJavaPart);
+  }
+
+  void JHybridYoloModelSpec::CxxPart::registerNatives() {
+    registerHybrid({
+      makeNativeMethod("initHybrid", JHybridYoloModelSpec::CxxPart::initHybrid),
+    });
+  }
+
+  // Properties
+  
+
+  // Methods
+  std::vector<Detection> JHybridYoloModelSpec::detect(const std::shared_ptr<margelo::nitro::camera::HybridFrameSpec>& frame) {
+    static const auto method = _javaPart->javaClassStatic()->getMethod<jni::local_ref<jni::JArrayClass<JDetection>>(jni::alias_ref<margelo::nitro::camera::JHybridFrameSpec::JavaPart> /* frame */)>("detect");
+    auto __result = method(_javaPart, std::dynamic_pointer_cast<margelo::nitro::camera::JHybridFrameSpec>(frame)->getJavaPart());
+    return [&](auto&& __input) {
+      size_t __size = __input->size();
+      std::vector<Detection> __vector;
+      __vector.reserve(__size);
+      for (size_t __i = 0; __i < __size; __i++) {
+        auto __element = __input->getElement(__i);
+        __vector.push_back(__element->toCpp());
+      }
+      return __vector;
+    }(__result);
+  }
+  void JHybridYoloModelSpec::close() {
+    static const auto method = _javaPart->javaClassStatic()->getMethod<void()>("close");
+    method(_javaPart);
+  }
+
+} // namespace margelo::nitro::yolo

package/nitrogen/generated/android/c++/JHybridYoloModelSpec.hpp

@@ -0,0 +1,64 @@
+///
+/// HybridYoloModelSpec.hpp
+/// This file was generated by nitrogen. DO NOT MODIFY THIS FILE.
+/// https://github.com/mrousavy/nitro
+/// Copyright © Marc Rousavy @ Margelo
+///
+
+#pragma once
+
+#include <NitroModules/JHybridObject.hpp>
+#include <fbjni/fbjni.h>
+#include "HybridYoloModelSpec.hpp"
+
+
+
+
+namespace margelo::nitro::yolo {
+
+  using namespace facebook;
+
+  class JHybridYoloModelSpec: public virtual HybridYoloModelSpec, public virtual JHybridObject {
+  public:
+    struct JavaPart: public jni::JavaClass<JavaPart, JHybridObject::JavaPart> {
+      static constexpr auto kJavaDescriptor = "Lcom/margelo/nitro/yolo/HybridYoloModelSpec;";
+      std::shared_ptr<JHybridYoloModelSpec> getJHybridYoloModelSpec();
+    };
+    struct CxxPart: public jni::HybridClass<CxxPart, JHybridObject::CxxPart> {
+      static constexpr auto kJavaDescriptor = "Lcom/margelo/nitro/yolo/HybridYoloModelSpec$CxxPart;";
+      static jni::local_ref<jhybriddata> initHybrid(jni::alias_ref<jhybridobject> jThis);
+      static void registerNatives();
+      using HybridBase::HybridBase;
+    protected:
+      std::shared_ptr<JHybridObject> createHybridObject(const jni::local_ref<JHybridObject::JavaPart>& javaPart) override;
+    };
+
+  public:
+    explicit JHybridYoloModelSpec(const jni::local_ref<JHybridYoloModelSpec::JavaPart>& javaPart):
+      HybridObject(HybridYoloModelSpec::TAG),
+      JHybridObject(javaPart),
+      _javaPart(jni::make_global(javaPart)) {}
+    ~JHybridYoloModelSpec() override {
+      // Hermes GC can destroy JS objects on a non-JNI Thread.
+      jni::ThreadScope::WithClassLoader([&] { _javaPart.reset(); });
+    }
+
+  public:
+    inline const jni::global_ref<JHybridYoloModelSpec::JavaPart>& getJavaPart() const noexcept {
+      return _javaPart;
+    }
+
+  public:
+    // Properties
+    
+
+  public:
+    // Methods
+    std::vector<Detection> detect(const std::shared_ptr<margelo::nitro::camera::HybridFrameSpec>& frame) override;
+    void close() override;
+
+  private:
+    jni::global_ref<JHybridYoloModelSpec::JavaPart> _javaPart;
+  };
+
+} // namespace margelo::nitro::yolo

package/nitrogen/generated/android/c++/JHybridYoloSpec.cpp

@@ -7,20 +7,15 @@
 
 #include "JHybridYoloSpec.hpp"
 
-// Forward declaration of `Detection` to properly resolve imports.
-namespace margelo::nitro::yolo { struct Detection; }
-// Forward declaration of `BoundingBox` to properly resolve imports.
-namespace margelo::nitro::yolo { struct BoundingBox; }
+// Forward declaration of `HybridYoloModelSpec` to properly resolve imports.
+namespace margelo::nitro::yolo { class HybridYoloModelSpec; }
 // Forward declaration of `HybridFrameSpec` to properly resolve imports.
 namespace margelo::nitro::camera { class HybridFrameSpec; }
 
-#include <string>
-#include "Detection.hpp"
-#include <vector>
-#include "JDetection.hpp"
-#include "BoundingBox.hpp"
-#include "JBoundingBox.hpp"
 #include <memory>
+#include "HybridYoloModelSpec.hpp"
+#include "JHybridYoloModelSpec.hpp"
+#include <string>
 #include <VisionCamera/HybridFrameSpec.hpp>
 #include <VisionCamera/JHybridFrameSpec.hpp>
 
@@ -57,33 +52,15 @@ namespace margelo::nitro::yolo {
   
 
   // Methods
-  double JHybridYoloSpec::sum(double num1, double num2) {
-    static const auto method = _javaPart->javaClassStatic()->getMethod<double(double /* num1 */, double /* num2 */)>("sum");
-    auto __result = method(_javaPart, num1, num2);
-    return __result;
-  }
-  void JHybridYoloSpec::loadModel(const std::string& modelPath) {
-    static const auto method = _javaPart->javaClassStatic()->getMethod<void(jni::alias_ref<jni::JString> /* modelPath */)>("loadModel");
-    method(_javaPart, jni::make_jstring(modelPath));
+  std::shared_ptr<HybridYoloModelSpec> JHybridYoloSpec::loadModel(const std::string& modelPath) {
+    static const auto method = _javaPart->javaClassStatic()->getMethod<jni::local_ref<JHybridYoloModelSpec::JavaPart>(jni::alias_ref<jni::JString> /* modelPath */)>("loadModel");
+    auto __result = method(_javaPart, jni::make_jstring(modelPath));
+    return __result->getJHybridYoloModelSpec();
   }
   std::string JHybridYoloSpec::frameToBase64(const std::shared_ptr<margelo::nitro::camera::HybridFrameSpec>& frame) {
     static const auto method = _javaPart->javaClassStatic()->getMethod<jni::local_ref<jni::JString>(jni::alias_ref<margelo::nitro::camera::JHybridFrameSpec::JavaPart> /* frame */)>("frameToBase64");
     auto __result = method(_javaPart, std::dynamic_pointer_cast<margelo::nitro::camera::JHybridFrameSpec>(frame)->getJavaPart());
     return __result->toStdString();
   }
-  std::vector<Detection> JHybridYoloSpec::detect(const std::shared_ptr<margelo::nitro::camera::HybridFrameSpec>& frame) {
-    static const auto method = _javaPart->javaClassStatic()->getMethod<jni::local_ref<jni::JArrayClass<JDetection>>(jni::alias_ref<margelo::nitro::camera::JHybridFrameSpec::JavaPart> /* frame */)>("detect");
-    auto __result = method(_javaPart, std::dynamic_pointer_cast<margelo::nitro::camera::JHybridFrameSpec>(frame)->getJavaPart());
-    return [&](auto&& __input) {
-      size_t __size = __input->size();
-      std::vector<Detection> __vector;
-      __vector.reserve(__size);
-      for (size_t __i = 0; __i < __size; __i++) {
-        auto __element = __input->getElement(__i);
-        __vector.push_back(__element->toCpp());
-      }
-      return __vector;
-    }(__result);
-  }
 
 } // namespace margelo::nitro::yolo

package/nitrogen/generated/android/c++/JHybridYoloSpec.hpp

@@ -54,10 +54,8 @@ namespace margelo::nitro::yolo {
 
   public:
     // Methods
-    double sum(double num1, double num2) override;
-    void loadModel(const std::string& modelPath) override;
+    std::shared_ptr<HybridYoloModelSpec> loadModel(const std::string& modelPath) override;
     std::string frameToBase64(const std::shared_ptr<margelo::nitro::camera::HybridFrameSpec>& frame) override;
-    std::vector<Detection> detect(const std::shared_ptr<margelo::nitro::camera::HybridFrameSpec>& frame) override;
 
   private:
     jni::global_ref<JHybridYoloSpec::JavaPart> _javaPart;

package/nitrogen/generated/android/kotlin/com/margelo/nitro/yolo/HybridYoloModelSpec.kt

@@ -0,0 +1,59 @@
+///
+/// HybridYoloModelSpec.kt
+/// This file was generated by nitrogen. DO NOT MODIFY THIS FILE.
+/// https://github.com/mrousavy/nitro
+/// Copyright © Marc Rousavy @ Margelo
+///
+
+package com.margelo.nitro.yolo
+
+import androidx.annotation.Keep
+import com.facebook.jni.HybridData
+import com.facebook.proguard.annotations.DoNotStrip
+import com.margelo.nitro.camera.HybridFrameSpec
+import com.margelo.nitro.core.HybridObject
+
+/**
+ * A Kotlin class representing the YoloModel HybridObject.
+ * Implement this abstract class to create Kotlin-based instances of YoloModel.
+ */
+@DoNotStrip
+@Keep
+@Suppress(
+  "KotlinJniMissingFunction", "unused",
+  "RedundantSuppression", "RedundantUnitReturnType", "SimpleRedundantLet",
+  "LocalVariableName", "PropertyName", "PrivatePropertyName", "FunctionName"
+)
+abstract class HybridYoloModelSpec: HybridObject() {
+  // Properties
+  
+
+  // Methods
+  @DoNotStrip
+  @Keep
+  abstract fun detect(frame: com.margelo.nitro.camera.HybridFrameSpec): Array<Detection>
+  
+  @DoNotStrip
+  @Keep
+  abstract fun close(): Unit
+
+  // Default implementation of `HybridObject.toString()`
+  override fun toString(): String {
+    return "[HybridObject YoloModel]"
+  }
+
+  // C++ backing class
+  @DoNotStrip
+  @Keep
+  protected open class CxxPart(javaPart: HybridYoloModelSpec): HybridObject.CxxPart(javaPart) {
+    // C++ JHybridYoloModelSpec::CxxPart::initHybrid(...)
+    external override fun initHybrid(): HybridData
+  }
+  override fun createCxxPart(): CxxPart {
+    return CxxPart(this)
+  }
+
+  companion object {
+    protected const val TAG = "HybridYoloModelSpec"
+  }
+}

package/nitrogen/generated/android/kotlin/com/margelo/nitro/yolo/HybridYoloSpec.kt

@@ -31,19 +31,11 @@ abstract class HybridYoloSpec: HybridObject() {
   // Methods
   @DoNotStrip
   @Keep
-  abstract fun sum(num1: Double, num2: Double): Double
-  
-  @DoNotStrip
-  @Keep
-  abstract fun loadModel(modelPath: String): Unit
+  abstract fun loadModel(modelPath: String): HybridYoloModelSpec
   
   @DoNotStrip
   @Keep
   abstract fun frameToBase64(frame: com.margelo.nitro.camera.HybridFrameSpec): String
-  
-  @DoNotStrip
-  @Keep
-  abstract fun detect(frame: com.margelo.nitro.camera.HybridFrameSpec): Array<Detection>
 
   // Default implementation of `HybridObject.toString()`
   override fun toString(): String {

package/nitrogen/generated/ios/Yolo-Swift-Cxx-Bridge.cpp

@@ -8,6 +8,7 @@
 #include "Yolo-Swift-Cxx-Bridge.hpp"
 
 // Include C++ implementation defined types
+#include "HybridYoloModelSpecSwift.hpp"
 #include "HybridYoloSpecSwift.hpp"
 #include "Yolo-Swift-Cxx-Umbrella.hpp"
 #include <NitroModules/NitroDefines.hpp>
@@ -25,6 +26,22 @@ namespace margelo::nitro::yolo::bridge::swift {
     return margelo::nitro::camera::bridge::swift::get_std__shared_ptr_HybridFrameSpec_(cppType);
   }
   
+  // pragma MARK: std::shared_ptr<HybridYoloModelSpec>
+  std::shared_ptr<HybridYoloModelSpec> create_std__shared_ptr_HybridYoloModelSpec_(void* NON_NULL swiftUnsafePointer) noexcept {
+    Yolo::HybridYoloModelSpec_cxx swiftPart = Yolo::HybridYoloModelSpec_cxx::fromUnsafe(swiftUnsafePointer);
+    return std::make_shared<margelo::nitro::yolo::HybridYoloModelSpecSwift>(swiftPart);
+  }
+  void* NON_NULL get_std__shared_ptr_HybridYoloModelSpec_(std__shared_ptr_HybridYoloModelSpec_ cppType) {
+    std::shared_ptr<margelo::nitro::yolo::HybridYoloModelSpecSwift> swiftWrapper = std::dynamic_pointer_cast<margelo::nitro::yolo::HybridYoloModelSpecSwift>(cppType);
+    #ifdef NITRO_DEBUG
+    if (swiftWrapper == nullptr) [[unlikely]] {
+      throw std::runtime_error("Class \"HybridYoloModelSpec\" is not implemented in Swift!");
+    }
+    #endif
+    Yolo::HybridYoloModelSpec_cxx& swiftPart = swiftWrapper->getSwiftPart();
+    return swiftPart.toUnsafe();
+  }
+  
   // pragma MARK: std::shared_ptr<HybridYoloSpec>
   std::shared_ptr<HybridYoloSpec> create_std__shared_ptr_HybridYoloSpec_(void* NON_NULL swiftUnsafePointer) noexcept {
     Yolo::HybridYoloSpec_cxx swiftPart = Yolo::HybridYoloSpec_cxx::fromUnsafe(swiftUnsafePointer);