Xingyi Li1,3, Zhiguo Cao1, Huiqiang Sun1, Jianming Zhang2, Ke Xian3*, Guosheng Lin3
1Huazhong University of Science and Technology, 2Adobe Research, 3S-Lab, Nanyang Technological University
This repository is the official PyTorch implementation of the CVPR 2023 paper "3D Cinemagraphy from a Single Image".
git clone https://github.com/xingyi-li/3d-cinemagraphy.git
cd 3d-cinemagraphy
bash requirements.sh
Download pretrained models from Google Drive, then unzip and put them in the directory ckpts
.
To achieve better motion estimation results and controllable animation, here we provide the controllable version.
Firstly, use labelme to specify the target regions (masks) and desired movement directions (hints):
conda activate 3d-cinemagraphy
cd demo/0/
labelme image.png
It is recommended to specify short hints rather than long hints to avoid artifacts. Please follow labelme for detailed instructions if needed.
After that, we can obtain an image.json file. Our next step is to convert the annotations stored in JSON format into datasets that can be used by our method:
labelme_json_to_dataset image.json # this will generate a folder image_json
cd ../../
python scripts/generate_mask.py --inputdir demo/0/image_json
We now can create 3D cinemagraphs according to your preference:
python demo.py -c configs/config.yaml --input_dir demo/0/ --ckpt_path ckpts/model_150000.pth --flow_scale 1.0 --ds_factor 1.0
input_dir
: input folder that contains src images.ckpt_path
: checkpoint path.flow_scale
: scale that used to control the speed of fluid, > 1.0 will slow down the fluid.ds_factor
: downsample factor for the input images.
Results will be saved to the input_dir/output
.
- Due to the limited size of the training dataset, the intermediate frame may occasionally experience flickering.
- The utilization of a fixed distance threshold in agglomerative clustering within the disparity space can occasionally result in the presence of visible boundaries between different layers.
- We may sometimes see artifacts when the fluid is moving very fast. You can either slow down the fluid by increasing the
flow_scale
or try to specify short hints rather than long hints, to avoid artifacts. - The motion estimation module occasionally generates motion fields that do not perfectly align with the desired preferences.
If you find our work useful in your research, please consider to cite our paper:
@InProceedings{li2023_3dcinemagraphy,
author = {Li, Xingyi and Cao, Zhiguo and Sun, Huiqiang and Zhang, Jianming and Xian, Ke and Lin, Guosheng},
title = {3D Cinemagraphy From a Single Image},
booktitle = {Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR)},
month = {June},
year = {2023},
pages = {4595-4605}
}
- Animating Pictures with Eulerian Motion Fields, CVPR 2021
- Controllable Animation of Fluid Elements in Still Images, CVPR 2022
- Simulating Fluids in Real-World Still Images, arXiv 2022
- 3D Photography using Context-aware Layered Depth Inpainting, CVPR 2020
- 3D Photo Stylization: Learning to Generate Stylized Novel Views from a Single Image, CVPR 2022
- 3D Moments from Near-Duplicate Photos, CVPR 2022
- 3D Video Loops from Asynchronous Input, CVPR 2023
This code borrows heavily from 3D Moments and SLR-SFS. We thank the respective authors for open sourcing their methods.