TensorFlow Implementation for the paper:
Patchwise Generative ConvNet: Training Energy-Based Models
from a Single Natural Image for Internal Learning
In CVPR 2021 (Oral)
This codebase is tested using Ubuntu 16.04, Python 3.5 and a single NVIDIA RTX 2080 GPU. Similar configurations are preferred.
- Clone this repo:
git clone https://github.com/zilongzheng/PatchGenCN.git
cd PatchGenCN- Install requirements
- Tensorflow 1.14+
python train.py --datapath <path to image>If you use this code for your research, please cite our paper.
@InProceedings{zheng2021patchgencn,
author = {Zheng, Zilong and Xie, Jianwen and Li, Ping},
title = {Patchwise Generative ConvNet: Training Energy-Based Models From a Single Natural Image for Internal Learning},
booktitle = {Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR)},
month = {June},
year = {2021},
pages = {2961-2970}
}