Update README.md
Browse files
README.md
CHANGED
|
@@ -0,0 +1,38 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
license: mit
|
| 3 |
+
tags:
|
| 4 |
+
- computer-vision
|
| 5 |
+
- video-prediction
|
| 6 |
+
- spatiotemporal-prediction
|
| 7 |
+
- pytorch
|
| 8 |
+
datasets:
|
| 9 |
+
- moving-mnist
|
| 10 |
+
- taxibj
|
| 11 |
+
- kth
|
| 12 |
+
- human3.6m
|
| 13 |
+
---
|
| 14 |
+
# PFGNet
|
| 15 |
+
|
| 16 |
+
PFGNet is a fully convolutional framework for efficient spatiotemporal predictive learning.
|
| 17 |
+
|
| 18 |
+
This repository provides dataset-specific trained checkpoints of PFGNet on multiple benchmarks.
|
| 19 |
+
|
| 20 |
+
**Code:** https://github.com/fhjdqaq/PFGNet
|
| 21 |
+
**Paper:** https://huggingface.co/papers/2602.20537
|
| 22 |
+
**Provided by:** WYK Lab
|
| 23 |
+
**Project page:** https://kaimaoge.github.io
|
| 24 |
+
|
| 25 |
+
## Available checkpoints
|
| 26 |
+
|
| 27 |
+
| Dataset | Checkpoint |
|
| 28 |
+
|---|---|
|
| 29 |
+
| Moving MNIST | `pfg_mmnist.ckpt` |
|
| 30 |
+
| Moving Fashion MNIST | `pfg_mfmnist.ckpt` |
|
| 31 |
+
| TaxiBJ | `pfg_taxibj.ckpt` |
|
| 32 |
+
| KTH (10→20) | `pfg_kth20.ckpt` |
|
| 33 |
+
| KTH (10→40) | `pfg_kth40.ckpt` |
|
| 34 |
+
| Human3.6M | `pfg_human.ckpt` |
|
| 35 |
+
|
| 36 |
+
## Usage
|
| 37 |
+
|
| 38 |
+
Please refer to the official code repository for training and inference scripts.
|