weather_effect_generator / gen_rain_nst.sh
qninhdt's picture
Upload 35 files
1a41d53 verified
raw
history blame contribute delete
272 Bytes
python Neural_Style_Transfer.py \
--content-imgs ./data/images/ \
--style-imgs ./data/styles/rain_images \
--save-folder ./data/nst/rain_images \
--vgg ./checkpoints/rain_vgg_512 \
--min-step 10 \
--max-step 10 \
--style-weight 10 \
--cuda