Views
No views yet
| Model | Params | Size | Input | Embedding | Accuracy |
|---|---|---|---|---|---|
| PE-Core-G14-448-ANE | 2.4B | 3.5GB | 448x448 | 1280 | 1.0000 |
| PE-Core-L-14-336-ANE | 300M | 604MB | 336x336 | 1024 | 1.0000 |
| PE-Core-B-16-ANE | 86M | 178MB | 224x224 | 768 | 0.9998 |
| PE-Core-S-16-384-ANE | 22M | 45MB | 384x384 | 384 | 1.0000 |
| PE-Core-T-16-384-ANE | 6M | 12MB | 384x384 | 192 | 0.9999 |
| Model | ANE Latency | MPS Latency | Speedup |
|---|---|---|---|
| PE-Core-bigG-14-448 | 783ms | 1049ms | 1.34x |
| PE-Core-L-14-336 | ~180ms | ~280ms | ~1.5x |
| PE-Core-B-16 | ~50ms | ~80ms | ~1.6x |
1import coremltools as ct
2import numpy as np
3
4# Load model
5model = ct.models.MLModel("PE-Core-B-16-ANE.mlpackage")
6
7# Prepare image (1, 3, 224, 224) normalized
8image = np.random.randn(1, 3, 224, 224).astype(np.float32)
9
10# Get embedding
11output = model.predict({"image": image})
12embedding = output["embedding"] # (1, 768)
13
14# Normalize for similarity search
15embedding = embedding / np.linalg.norm(embedding)1import CoreML
2
3let model = try MLModel(contentsOf: modelURL)
4let input = try MLDictionaryFeatureProvider(dictionary: ["image": pixelBuffer])
5let output = try model.prediction(from: input)
6let embedding = output.featureValue(for: "embedding")!.multiArrayValue!