Merge pull request #11 from roboflow/cursor/add-resnet-models-for-ios-classification-9247

Cursor/add resnet models for ios classification 9247

Author: MadeWithStone

.github/workflows/swift.yml

@@ -62,6 +62,10 @@ jobs:
     - name: Run Swift package tests
       run: |
         swift test
+
+    - name: Run Xcode tests
+      run: |
+        xcodebuild test -scheme RoboflowTests -destination 'platform=iOS Simulator,OS=18.5,name=iPhone 16'
         
     - name: Validate package structure
-      run: swift package describe
+      run: swift package describe

.gitignore

@@ -90,3 +90,6 @@ fastlane/test_output
 iOSInjectionProject/
 
 .DS_Store
+
+**/*.mlpackage
+**/*.mlmodel

Package.swift

@@ -20,6 +20,13 @@ let package = Package(
             name: "Roboflow",
             dependencies: [], path: "Sources/Roboflow"
         ),
-        .testTarget(name:"RoboflowTests", dependencies: ["Roboflow"])
+        .testTarget(
+            name: "RoboflowTests", 
+            dependencies: ["Roboflow"],
+            path: "Tests/RoboflowTests",
+            resources: [
+                .copy("assets")
+            ]
+        )
     ]
 )

README.md

@@ -178,7 +178,11 @@ If you've previously installed the Roboflow SDK via Cocoapods, you'll need to up
 The SDK includes a comprehensive test suite that validates model loading and inference functionality. To run the tests:
 
 ```bash
+# for swift only tests
 swift test
+
+# for iOS simulator tests
+xcodebuild test -scheme RoboflowTests -destination 'platform=iOS Simulator,arch=arm64,OS=18.5,name=iPhone 16'
 ```
 
 The test suite includes:

Sources/Roboflow/Classes/Roboflow.docc/Roboflow.md

@@ -12,7 +12,7 @@ import UIKit
 
 extension RFModel {
     /// Run image through model and return detections
-    public func detect(image: UIImage, completion: @escaping (([RFObjectDetectionPrediction]?, Error?) -> Void)) {
+    public func detect(image: UIImage, completion: @escaping (([RFPrediction]?, Error?) -> Void)) {
         let size = image.size
         let attrs = [
             kCVPixelBufferCGImageCompatibilityKey: kCFBooleanTrue!,
@@ -71,7 +71,6 @@ extension RFModel {
         }
     }
 }
-
 #endif
 
 func hexStringToCGColor (hex:String) -> CGColor {

Sources/Roboflow/Classes/Roboflow.swift

@@ -24,3 +24,45 @@ public struct UnsupportedOSError: Error, LocalizedError, CustomStringConvertible
         "[UnsupportedOSError] \(message)"
     }
 }
+
+public struct DetectionError: Error, LocalizedError, CustomStringConvertible {
+
+    // MARK: Stored properties
+    public let message: String
+
+    // MARK: Initialiser
+    public init(
+        message: String = "Error performing detection."
+    ) {
+        self.message     = message
+    }
+
+    // MARK: LocalizedError
+    public var errorDescription: String? { message }
+
+    // MARK: CustomStringConvertible
+    public var description: String {
+        "[DetectionError] \(message)"
+    }
+}
+
+public struct ModelLoadError: Error, LocalizedError, CustomStringConvertible {
+
+    // MARK: Stored properties
+    public let message: String
+
+    // MARK: Initialiser
+    public init(
+        message: String = "Error loading model."
+    ) {
+        self.message     = message
+    }
+
+    // MARK: LocalizedError
+    public var errorDescription: String? { message }
+
+    // MARK: CustomStringConvertible
+    public var description: String {
+        "[ModelLoadError] \(message)"
+    }
+}