CodeMirror/waifu2x
1
0
Fork 0
mirror of synced 2024-06-26 10:10:49 +12:00
Code Issues Releases Wiki Activity

Add missing training script

Browse source
This commit is contained in:
nagadomi 2015-12-04 23:33:30 +09:00
parent e4b239ec14
commit acdc29c25c
1 changed files with 12 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
train_photo.sh Executable file
View file

@ -0,0 +1,12 @@
#!/bin/sh
th convert_data.lua -style photo -data_dir ./data/photo -model_dir models/photo
th train.lua -style photo -method scale -data_dir ./data/photo -model_dir models/photo_uk -test work/scale_test_photo.png -color rgb -thread 4 -backend cudnn -random_unsharp_mask_rate 0.1 -validation_crops 160
th tools/cleanup_model.lua -model models/photo/scale2.0x_model.t7 -oformat ascii
th train.lua -style photo -method noise -noise_level 1 -data_dir ./data/photo -model_dir models/photo -test work/noise_test_photo.jpg -color rgb -thread 4 -backend cudnn -random_unsharp_mask_rate 0.5 -validation_crops 160 -nr_rate 0.6 -epoch 33
th tools/cleanup_model.lua -model models/photo/noise1_model.t7 -oformat ascii
th train.lua -style photo -method noise -noise_level 2 -data_dir ./data/photo -model_dir models/photo -test work/noise_test_photo.jpg -color rgb -thread 4 -backend cudnn -random_unsharp_mask_rate 0.5 -validation_crops 160 -nr_rate 0.8 -epoch 38
th tools/cleanup_model.lua -model models/photo/noise2_model.t7 -oformat ascii