Spaces:
Configuration error
Configuration error
xierui.0097
commited on
Commit
Β·
4662946
1
Parent(s):
e3dae60
Add application file
Browse files
README.md
CHANGED
|
@@ -1,106 +1,15 @@
|
|
| 1 |
-
|
| 2 |
-
|
| 3 |
-
|
| 4 |
-
|
| 5 |
-
|
| 6 |
-
|
| 7 |
-
|
| 8 |
-
|
| 9 |
-
|
| 10 |
-
|
| 11 |
-
|
| 12 |
-
|
| 13 |
-
|
| 14 |
-
|
| 15 |
-
|
| 16 |
-
</div>
|
| 17 |
-
<div>
|
| 18 |
-
<sup>1</sup>Nanjing University, <sup>2</sup>ByteDance,  <sup>3</sup>Southwest University
|
| 19 |
-
</div>
|
| 20 |
-
<div>
|
| 21 |
-
<h4 align="center">
|
| 22 |
-
<a href="https://nju-pcalab.github.io/projects/STAR" target='_blank'>
|
| 23 |
-
<img src="https://img.shields.io/badge/π-Project%20Page-blue">
|
| 24 |
-
</a>
|
| 25 |
-
<a href="https://arxiv.org/abs/2407.07667" target='_blank'>
|
| 26 |
-
<img src="https://img.shields.io/badge/arXiv-2312.06640-b31b1b.svg">
|
| 27 |
-
</a>
|
| 28 |
-
<a href="https://youtu.be/hx0zrql-SrU" target='_blank'>
|
| 29 |
-
<img src="https://img.shields.io/badge/Demo%20Video-%23FF0000.svg?logo=YouTube&logoColor=white">
|
| 30 |
-
</a>
|
| 31 |
-
</h4>
|
| 32 |
-
</div>
|
| 33 |
-
</div>
|
| 34 |
-
|
| 35 |
-
|
| 36 |
-
### π Updates
|
| 37 |
-
- **2024.12.01** The pretrained STAR model (I2VGen-XL version) and inference code have been released.
|
| 38 |
-
|
| 39 |
-
|
| 40 |
-
## π Method Overview
|
| 41 |
-

|
| 42 |
-
|
| 43 |
-
|
| 44 |
-
## π· Results Display
|
| 45 |
-

|
| 46 |
-

|
| 47 |
-
π More visual results can be found in our [Project Page](https://nju-pcalab.github.io/projects/STAR) and [Video Demo](https://youtu.be/hx0zrql-SrU).
|
| 48 |
-
|
| 49 |
-
|
| 50 |
-
## βοΈ Dependencies and Installation
|
| 51 |
-
```
|
| 52 |
-
## git clone this repository
|
| 53 |
-
git clone https://github.com/NJU-PCALab/STAR.git
|
| 54 |
-
cd STAR
|
| 55 |
-
|
| 56 |
-
## create an environment
|
| 57 |
-
conda create -n star python=3.10
|
| 58 |
-
conda activate star
|
| 59 |
-
pip install -r requirements.txt
|
| 60 |
-
sudo apt-get update && apt-get install ffmpeg libsm6 libxext6 -y
|
| 61 |
-
```
|
| 62 |
-
|
| 63 |
-
## π Inference
|
| 64 |
-
#### Step 1: Download the pretrained model STAR from [HuggingFace](https://huggingface.co/SherryX/STAR).
|
| 65 |
-
We provide two verisions, `heavy_deg.pt` for heavy degraded videos and `light_deg.pt` for light degraded videos (e.g., the low-resolution video downloaded from video websites).
|
| 66 |
-
|
| 67 |
-
You can put the weight into `pretrained_weight/`.
|
| 68 |
-
|
| 69 |
-
|
| 70 |
-
#### Step 2: Prepare testing data
|
| 71 |
-
You can put the testing videos in the `input/video/`.
|
| 72 |
-
|
| 73 |
-
As for the prompt, there are three options: 1. No prompt. 2. Automatically generate a prompt [using Pllava](https://github.com/hpcaitech/Open-Sora/tree/main/tools/caption#pllava-captioning). 3. Manually write the prompt. You can put the txt file in the `input/text/`.
|
| 74 |
-
|
| 75 |
-
|
| 76 |
-
#### Step 3: Change the path
|
| 77 |
-
You need to change the paths in `video_super_resolution/scripts/inference_sr.sh` to your local corresponding paths, including `video_folder_path`, `txt_file_path`, `model_path`, and `save_dir`.
|
| 78 |
-
|
| 79 |
-
|
| 80 |
-
#### Step 4: Running inference command
|
| 81 |
-
```
|
| 82 |
-
bash video_super_resolution/scripts/inference_sr.sh
|
| 83 |
-
```
|
| 84 |
-
|
| 85 |
-
|
| 86 |
-
## β€οΈ Acknowledgments
|
| 87 |
-
This project is based on [I2VGen-XL](https://github.com/ali-vilab/VGen), [VEnhancer](https://github.com/Vchitect/VEnhancer) and [CogVideoX](https://github.com/THUDM/CogVideo). Thanks for their awesome works.
|
| 88 |
-
|
| 89 |
-
|
| 90 |
-
## πCitations
|
| 91 |
-
If our project helps your research or work, please consider citing our paper:
|
| 92 |
-
|
| 93 |
-
```
|
| 94 |
-
@misc{xie2024addsr,
|
| 95 |
-
title={AddSR: Accelerating Diffusion-based Blind Super-Resolution with Adversarial Diffusion Distillation},
|
| 96 |
-
author={Rui Xie and Ying Tai and Kai Zhang and Zhenyu Zhang and Jun Zhou and Jian Yang},
|
| 97 |
-
year={2024},
|
| 98 |
-
eprint={2404.01717},
|
| 99 |
-
archivePrefix={arXiv},
|
| 100 |
-
primaryClass={cs.CV}
|
| 101 |
-
}
|
| 102 |
-
```
|
| 103 |
-
|
| 104 |
-
|
| 105 |
-
## π§ Contact
|
| 106 |
-
If you have any inquiries, please don't hesitate to reach out via email at `ruixie0097@gmail.com`
|
|
|
|
| 1 |
+
title: STAR
|
| 2 |
+
emoji: π
|
| 3 |
+
colorFrom: red
|
| 4 |
+
colorTo: yellow
|
| 5 |
+
sdk: gradio
|
| 6 |
+
sdk_version: 4.37.2
|
| 7 |
+
app_file: app.py
|
| 8 |
+
pinned: false
|
| 9 |
+
disable_embedding: true
|
| 10 |
+
tags:
|
| 11 |
+
- Video Super-Resolution
|
| 12 |
+
- Video Restoration
|
| 13 |
+
- Video-to-Video
|
| 14 |
+
- Text-to-Video Model
|
| 15 |
+
short_description: Video super-resolution with text-to-video model
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|