Skip to content

Latest commit

 

History

History
11 lines (6 loc) · 398 Bytes

readme.md

File metadata and controls

11 lines (6 loc) · 398 Bytes
Raw

train a single image

python train.py --model_dir=test/

use pretrained model to train a single image (inference)

*pretrained model was trained on image fake_and_real_peppers_ms from CAVE dataset. And It is not trained to converge.

python train.py --model_dir=test/ --start_epoch=499 --end_epoch=500

for better performance, you can set --end_epoch= to a larger number, like 1000