| { | |
| "files": { | |
| "dino": "dino_student.onnx", | |
| "vae": "vae_student.onnx", | |
| "matte": "matte_student.onnx" | |
| }, | |
| "shapes": { | |
| "dino": { | |
| "input": [ | |
| 1, | |
| 3, | |
| 1024, | |
| 1024 | |
| ], | |
| "output": [ | |
| 1, | |
| 4101, | |
| 1280 | |
| ] | |
| }, | |
| "vae": { | |
| "input": [ | |
| 1, | |
| 3, | |
| 1024, | |
| 1024 | |
| ], | |
| "output": [ | |
| 1, | |
| 4096, | |
| 128 | |
| ] | |
| }, | |
| "matte": { | |
| "input": [ | |
| 1, | |
| 3, | |
| "h", | |
| "w" | |
| ], | |
| "output": [ | |
| 1, | |
| 1, | |
| "h", | |
| "w" | |
| ] | |
| } | |
| }, | |
| "params": { | |
| "dino": 19.38, | |
| "vae": 7.83, | |
| "matte": 2.76 | |
| }, | |
| "preprocess": { | |
| "resize": "shortest side 1024, dims rounded to /32", | |
| "matte_input": "rgb / 255", | |
| "dino_input": "canvas rgb / 255", | |
| "vae_input": "canvas rgb / 255 * 2 - 1", | |
| "canonical": 1024 | |
| }, | |
| "tokens": { | |
| "dino": 4101, | |
| "vae": 4096 | |
| } | |
| } |