Improve model card: Add pipeline tag, library name, and optical-flow tag (#1)
Browse files- Improve model card: Add pipeline tag, library name, and optical-flow tag (0d88c87be6b14cd743bf47f8ef5c493b68783a9d)
Co-authored-by: Niels Rogge <[email protected]>
README.md
CHANGED
|
@@ -1,9 +1,12 @@
|
|
| 1 |
---
|
| 2 |
-
license: bsd-3-clause
|
| 3 |
language:
|
| 4 |
- en
|
|
|
|
| 5 |
tags:
|
| 6 |
- arxiv:2506.23151
|
|
|
|
|
|
|
|
|
|
| 7 |
---
|
| 8 |
|
| 9 |
# MEMFOF-Tartan-T-TSKH-spring
|
|
@@ -32,4 +35,4 @@ model = MEMFOF.from_pretrained("egorchistov/MEMFOF-Tartan-T-TSKH-spring").eval()
|
|
| 32 |
with torch.inference_mode():
|
| 33 |
example_input = torch.randint(0, 256, [1, 3, 3, 1080, 1920], device=device) # [B=1, T=3, C=3, H=1080, W=1920]
|
| 34 |
backward_flow, forward_flow = model(example_input).unbind(dim=1) # [B=1, C=2, H=1080, W=1920]
|
| 35 |
-
```
|
|
|
|
| 1 |
---
|
|
|
|
| 2 |
language:
|
| 3 |
- en
|
| 4 |
+
license: bsd-3-clause
|
| 5 |
tags:
|
| 6 |
- arxiv:2506.23151
|
| 7 |
+
- optical-flow
|
| 8 |
+
pipeline_tag: image-to-image
|
| 9 |
+
library_name: pytorch
|
| 10 |
---
|
| 11 |
|
| 12 |
# MEMFOF-Tartan-T-TSKH-spring
|
|
|
|
| 35 |
with torch.inference_mode():
|
| 36 |
example_input = torch.randint(0, 256, [1, 3, 3, 1080, 1920], device=device) # [B=1, T=3, C=3, H=1080, W=1920]
|
| 37 |
backward_flow, forward_flow = model(example_input).unbind(dim=1) # [B=1, C=2, H=1080, W=1920]
|
| 38 |
+
```
|