imbcmdth commited on
Commit
fab0b40
·
verified ·
1 Parent(s): 994020a

the weight blobs moved to imbcmdth/longpipe-weights; README as before

Browse files
README.md CHANGED
@@ -1,65 +1,38 @@
1
- # longpipe-onnx
2
-
3
- Longpipe matting models for ffrwd model pins. The network and weights
4
- are [sb2702/longpipe](https://github.com/sb2702/longpipe), MIT licensed
5
- (code and weights; see their licensing page for dataset provenance).
6
- This repo holds two kinds of file: longpipe's own weight blobs, as the
7
- project serves them, and faithful ONNX conversions of them.
8
-
9
- ## Weight blobs
10
-
11
- The files longpipe's own WebGPU code loads, byte for byte as served at
12
- cdn.longpipe.dev/models/v/0.0.5/ (its manifest.json carries the same
13
- sizes and digests). One per tier in f32, and the f16 packing of each.
14
- ffrwd/longpipe 0.3 and later pin these and run longpipe's code on the
15
- sidecar's GPU; the `tier` parameter picks one.
16
-
17
- | file | tier | bytes | sha256 |
18
- |---|---|---|---|
19
- | model_xs.bin | xs | 4300564 | 501cd8fc26a260552c62d22cde4d5ab7b32087921f42ba2539a9b5300d3f8c9d |
20
- | model_small.bin | small | 4232672 | 56430de7da6cfd6b834753f912441fbca00ad397790e308b8dc041e5c0b03394 |
21
- | model_medium.bin | medium | 14074452 | d6684cecaa1cf0570626872b89eef3690e41e3b861fc67f8916732b549d50945 |
22
- | model_large.bin | large | 14075812 | 5294fbbfd1eac70a6ee8d99dbe4de9a6d8c459d2c8a893c24402b35464e121f5 |
23
- | model_xl.bin | xl | 30463184 | c1ea11f3ff2a17eb60fe53c5d37d91c410f6d992ecb245236b21c7ef80217032 |
24
- | model_xs.f16.bin | xs | 2153062 | f1786bf9ffb03916fd265c253f4928ec4a7576855162aacc856aaf90908899e9 |
25
- | model_small.f16.bin | small | 2119038 | 8fea35e96ee8e37d1b2b6e901a2a7f8df1ce2ec14e1a1227636df6b410e2c11f |
26
- | model_medium.f16.bin | medium | 7040874 | 7d8bf5d787da24e679193e70db1e7b9ca62f473ef79d8f14db1604dda8d99dd5 |
27
- | model_large.f16.bin | large | 7041646 | 40ad8b9f538cf70b89b09a41f916ab441610c9a92ff4222a27071fa22fe92ae9 |
28
- | model_xl.f16.bin | xl | 15236450 | 4f25cb0c267d8966329ea7882afd42a64856786d5de1a276b67eb20f42043fe4 |
29
-
30
- Each blob is a u32 little-endian header length, a JSON index of the
31
- tensors, then the packed weights in the layouts longpipe's shaders read.
32
-
33
- ## ONNX conversions
34
-
35
- Used by ffrwd/longpipe 0.2, which ran the xl system through ONNX Runtime.
36
-
37
- | file | preset | input | sha256 |
38
- |---|---|---|---|
39
- | matting_xl_temporal_f32.onnx | xl system | see notes | 67f79a367c4014d9df631afb002c99bf6efb1aa4ae122df175b45f2f92d611b6 |
40
- | matting_xl_refined_f32.onnx | xl system | see notes | 54b73fcc90335f4ef5274039f7ae09bf5462b2091172695c399bd8078b6b4a69 |
41
- | flow_xl_f32.onnx | xl system | see notes | e39ee98cb8ecff36e416ad234bb90b5301faa302becf4e82fec61869aa218df3 |
42
- | face_xl_f32.onnx | xl system | see notes | fb3d2ef4f6de97cc51708ed56ec9fb6e9fff8a5a0daeebf952a0f7ef8905cbb3 |
43
- | landmark_mesh_f32.onnx | xl system | see notes | e930cc43f06288ba53ba2a3251e2fd3c42995a5c39324c56988648cc32e62edc |
44
- | matting_large_f32.onnx | large | 1x3x160x256 RGB [0,1] | 5007ee2a9fe53dbdebfd2d6c9baba270199aaee9ad1189df9227030b51f21b70 |
45
- | matting_xl_f32.onnx | xl | 1x3x192x320 RGB [0,1] | 996cf2779c5b7c690b35ee1ef5e8890f8d2497e12dea97dc13ddd18e4132d04c |
46
- | matting_small_f32.onnx | small | 1x3x112x192 RGB [0,1] | d5a8c23461c253366ef9b87c59448e3d8c93062a65e25db6591b8def997f6afc |
47
-
48
- Converted from the f32 blobs above by reconstruction in PyTorch: the
49
- blob's own JSON index locates each tensor, the WGSL mat4x4 packing
50
- inverts to dense OIHW (the inverse of longpipe's own
51
- scripts/generate_fixtures.py packer), and the export holds
52
- PyTorch-to-ONNX parity under 1e-4. Conversion scripts live with the
53
- ffrwd project.
54
-
55
- The xl system files: `matting_xl_refined_f32.onnx` is base+wrapper
56
- in one graph (frame in, refined alpha out, canvas 1280x768);
57
- `matting_xl_temporal_f32.onnx` is the whole per-frame production
58
- step with the optical-flow stabilizer, state as explicit inputs and
59
- outputs (previous frame, four encoder taps, alpha and envelope);
60
- `flow_xl_f32.onnx` is the flow net alone; `face_xl_f32.onnx` emits
61
- five facial heatmaps at base/8; `landmark_mesh_f32.onnx` regresses
62
- 478 landmarks from a 256x256 ImageNet-normalized crop (its blob is
63
- their separate model_landmark_mesh.bin, sha ca534744...3b87).
64
-
65
- MIT, matching upstream.
 
1
+ # longpipe-onnx
2
+
3
+ Longpipe matting models converted to ONNX for use as ffrwd model
4
+ pins. The network and weights are [sb2702/longpipe](https://github.com/sb2702/longpipe),
5
+ MIT licensed (code and weights; see their licensing page for dataset
6
+ provenance). This repo holds faithful conversions, nothing more.
7
+
8
+ | file | preset | input | sha256 |
9
+ |---|---|---|---|
10
+ | matting_xl_temporal_f32.onnx | xl system | see notes | 67f79a367c4014d9df631afb002c99bf6efb1aa4ae122df175b45f2f92d611b6 |
11
+ | matting_xl_refined_f32.onnx | xl system | see notes | 54b73fcc90335f4ef5274039f7ae09bf5462b2091172695c399bd8078b6b4a69 |
12
+ | flow_xl_f32.onnx | xl system | see notes | e39ee98cb8ecff36e416ad234bb90b5301faa302becf4e82fec61869aa218df3 |
13
+ | face_xl_f32.onnx | xl system | see notes | fb3d2ef4f6de97cc51708ed56ec9fb6e9fff8a5a0daeebf952a0f7ef8905cbb3 |
14
+ | landmark_mesh_f32.onnx | xl system | see notes | e930cc43f06288ba53ba2a3251e2fd3c42995a5c39324c56988648cc32e62edc |
15
+ | matting_large_f32.onnx | large | 1x3x160x256 RGB [0,1] | 5007ee2a9fe53dbdebfd2d6c9baba270199aaee9ad1189df9227030b51f21b70 |
16
+ | matting_xl_f32.onnx | xl | 1x3x192x320 RGB [0,1] | 996cf2779c5b7c690b35ee1ef5e8890f8d2497e12dea97dc13ddd18e4132d04c |
17
+ | matting_small_f32.onnx | small | 1x3x112x192 RGB [0,1] | d5a8c23461c253366ef9b87c59448e3d8c93062a65e25db6591b8def997f6afc |
18
+
19
+ Converted from the f32 weight blobs served at
20
+ cdn.longpipe.dev/models/v/0.0.5/ (small blob 56430de7...b03394,
21
+ large 5294fbbf...4e121f5, xl c1ea11f3...80217032) by
22
+ reconstruction in PyTorch: the blob's own JSON index locates each
23
+ tensor, the WGSL mat4x4 packing inverts to dense OIHW (the inverse
24
+ of longpipe's own scripts/generate_fixtures.py packer), and the
25
+ export holds PyTorch-to-ONNX parity under 1e-4. Conversion scripts
26
+ live with the ffrwd project.
27
+
28
+ The xl system files: `matting_xl_refined_f32.onnx` is base+wrapper
29
+ in one graph (frame in, refined alpha out, canvas 1280x768);
30
+ `matting_xl_temporal_f32.onnx` is the whole per-frame production
31
+ step with the optical-flow stabilizer, state as explicit inputs and
32
+ outputs (previous frame, four encoder taps, alpha and envelope);
33
+ `flow_xl_f32.onnx` is the flow net alone; `face_xl_f32.onnx` emits
34
+ five facial heatmaps at base/8; `landmark_mesh_f32.onnx` regresses
35
+ 478 landmarks from a 256x256 ImageNet-normalized crop (its blob is
36
+ their separate model_landmark_mesh.bin, sha ca534744...3b87).
37
+
38
+ MIT, matching upstream.
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
model_large.bin DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:5294fbbfd1eac70a6ee8d99dbe4de9a6d8c459d2c8a893c24402b35464e121f5
3
- size 14075812
 
 
 
 
model_large.f16.bin DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:40ad8b9f538cf70b89b09a41f916ab441610c9a92ff4222a27071fa22fe92ae9
3
- size 7041646
 
 
 
 
model_medium.bin DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:d6684cecaa1cf0570626872b89eef3690e41e3b861fc67f8916732b549d50945
3
- size 14074452
 
 
 
 
model_medium.f16.bin DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:7d8bf5d787da24e679193e70db1e7b9ca62f473ef79d8f14db1604dda8d99dd5
3
- size 7040874
 
 
 
 
model_small.bin DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:56430de7da6cfd6b834753f912441fbca00ad397790e308b8dc041e5c0b03394
3
- size 4232672
 
 
 
 
model_small.f16.bin DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:8fea35e96ee8e37d1b2b6e901a2a7f8df1ce2ec14e1a1227636df6b410e2c11f
3
- size 2119038
 
 
 
 
model_xl.bin DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:c1ea11f3ff2a17eb60fe53c5d37d91c410f6d992ecb245236b21c7ef80217032
3
- size 30463184
 
 
 
 
model_xl.f16.bin DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:4f25cb0c267d8966329ea7882afd42a64856786d5de1a276b67eb20f42043fe4
3
- size 15236450
 
 
 
 
model_xs.bin DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:501cd8fc26a260552c62d22cde4d5ab7b32087921f42ba2539a9b5300d3f8c9d
3
- size 4300564
 
 
 
 
model_xs.f16.bin DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:f1786bf9ffb03916fd265c253f4928ec4a7576855162aacc856aaf90908899e9
3
- size 2153062