https://github.com/causallearning/2022_neurips_picmm
Science Score: 26.0%
This score indicates how likely this project is to be science-related based on various indicators:
-
○CITATION.cff file
-
✓codemeta.json file
Found codemeta.json file -
✓.zenodo.json file
Found .zenodo.json file -
○DOI references
-
○Academic publication links
-
○Academic email domains
-
○Institutional organization owner
-
○JOSS paper metadata
-
○Scientific vocabulary similarity
Low similarity (8.9%) to scientific vocabulary
Repository
Basic Info
- Host: GitHub
- Owner: CausalLearning
- Language: Python
- Default Branch: main
- Size: 10.6 MB
Statistics
- Stars: 115
- Watchers: 0
- Forks: 0
- Open Issues: 0
- Releases: 0
Metadata Files
README.md
Pluralistic Image Completion with Gaussian Mixture Models (PICMM)
Paper
This repository is the official pytorch implementation of our NeurIPS 2022 paper, Pluralistic Image Completion with Gaussian Mixture Models.
Xiaobo Xia1*, Wenhao Yang2*, Jie Ren3, Yewen Li4, Yibing Zhan5, Bo Han6, Tongliang Liu1
1The University of Sydney, 2Nanjing University, 3The University of Edinburgh, 4Nanyang Technological University, 5JD Explore Academy, 6Hong Kong Baptist University
* Equal contributions
Prerequisites
- Python >=3.7
- NVIDIA GPU + CUDA cuDNN
bash pip install -r requirements.txt
Pipeline
Training
python train.py --name [exp_name] \
--k [numbers_of_distributions] \
--img_file [training_image_path]
Notes of training process: + Our code supports
Inference
python test.py --name [exp_name] \
--k [numbers_of_distributions] \
--which_iter [which_iterations_to_load] \
--img_file [training_image_path] \
--sample_num [number_of_diverse_results_to_sample]
Notes of inference:
+ --sample_num: How many completion results do you want?
Citation
If you find our work useful for your research, please consider citing the following papers :)
bibtex
@inproceedings{xia2022pluralistic,
title={Pluralistic Image Completion with Gaussian Mixture Models},
author={Xia, Xiaobo and Yang, Wenhao and Ren, Jie and Li, Yewen and Zhan, Yibing and Han, Bo and Liu, Tongliang},
booktitle={NeurIPS},
year={2022}
}
Acknowledgments
The authors would give special thanks to Mingrui Zhu (Xidian University), Zihan Ding (Princeton University), and Chenlai Qian (Southeast University) for helpful discussions and comments.
Contact
This repo is currently maintained by Xiaobo Xia and Wenhao Yang, which is only for academic research use. If you have any problems about the implementation of our code, feel free to contact wenhaoyang.alpha@gmail.com. Discussions and questions about our paper are welcome via xiaoboxia.uni@gmail.com.
Owner
- Name: CausalLearning
- Login: CausalLearning
- Kind: organization
- Repositories: 1
- Profile: https://github.com/CausalLearning
GitHub Events
Total
- Watch event: 67
Last Year
- Watch event: 67
Dependencies
- dominate ==2.7.0
- imageio ==2.22.0
- numpy ==1.23.2
- opencv-python ==4.6.0.66
- torch ==1.11.0
- torchvision ==0.12.0