A TensorFlow implementation of paper Object Discovery with a Copy-Pasting GAN.
- Python 3.7
- TensorFlow 1.13.1
-
Prepare dataset
- Prepare foreground and background images(
jpg
format only); - Put foreground images into
data/plane_sky/plane
; - Put background images into
data/plane_sky/sky
; - Resize foreground and background images to
$240\times240$ ;(You can useimg_utils.py
inutils
folder);
- You can change image format in
train.py
andutils/img_utils.py
; - You can change image folder in
cfgs.py
;
- Prepare foreground and background images(
-
Train with
python main.py
. -
You can find checkpoints and saved images in
logs
. :)