DanteLiu commited on
Commit
0961764
·
verified ·
1 Parent(s): f58c4fb

Upload metadata.json with huggingface_hub

Browse files
Files changed (1) hide show
  1. metadata.json +40 -0
metadata.json ADDED
@@ -0,0 +1,40 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model": "DEIMv2-X",
3
+ "framework": "ONNX Runtime",
4
+ "description": "2-class pill + needle detector for Android, 100% exact match on 87 val images (fp32), 15/16 test images match (fp16)",
5
+ "classes": {"0": "pill", "1": "needle"},
6
+ "input": {
7
+ "name": "image",
8
+ "shape": [1, 3, 640, 640],
9
+ "dtype": "float16",
10
+ "layout": "NCHW",
11
+ "normalization": {
12
+ "mean": [0.485, 0.456, 0.406],
13
+ "std": [0.229, 0.224, 0.225],
14
+ "note": "ImageNet stats. Formula: (pixel / 255.0 - mean) / std"
15
+ }
16
+ },
17
+ "outputs": {
18
+ "labels": {"shape": [1, 300], "dtype": "int32", "description": "0=pill, 1=needle"},
19
+ "boxes": {"shape": [1, 300, 4], "dtype": "float16", "description": "xyxy pixel coords (0-640)"},
20
+ "scores": {"shape": [1, 300], "dtype": "float16", "description": "confidence 0-1"}
21
+ },
22
+ "postprocessing": {
23
+ "confidence_threshold": 0.5,
24
+ "nms": false,
25
+ "note": "DEIMv2 is end-to-end, no NMS needed. Always outputs 300 candidates, filter by score."
26
+ },
27
+ "files": {
28
+ "fp16": {"name": "deimv2_x_pill_fp16.onnx", "size_mb": 101},
29
+ "fp32": {"name": "deimv2_x_pill.onnx + .data", "size_mb": 199}
30
+ },
31
+ "performance": {
32
+ "val_images": 87,
33
+ "exact_match_fp32": "100%",
34
+ "exact_match_fp16": "99% (15/16 test images)",
35
+ "within_2": "100%",
36
+ "gpu_inference_ms": 24,
37
+ "parameters": "50.3M"
38
+ },
39
+ "gradle_dependency": "com.microsoft.onnxruntime:onnxruntime-android:1.21.0"
40
+ }