1
0
Fork 0
mirror of synced 2024-06-26 18:20:26 +12:00
Commit graph

68 commits

Author SHA1 Message Date
nagadomi 0fe21eef70 Update random erasing; Support x padding 2018-10-03 18:30:22 +09:00
nagadomi 93cd40a53c merge random erasing 2018-10-03 18:29:05 +09:00
nagadomi fa6ee00624 merge 2018-01-26 08:09:40 +09:00
nagadomi 4ec0b4530d merge 2018-01-26 08:08:52 +09:00
nagadomi cba96e36fb Add -invert_x option for user_method 2017-05-17 08:51:07 +09:00
nagadomi bb0fc3a1d3 Add -validation_filename_split option 2017-04-15 16:29:38 +09:00
nagadomi f54dd37848 Merge branch 'dev' 2017-04-10 20:22:37 +09:00
nagadomi b7e116de54 Add support for multi GPU training (data parallel)
train.lua -gpu 1,3,4
When use multi GPU mode, nccl.torch is required.
2017-04-10 20:20:17 +09:00
nagadomi f0fc2c89d1 Add support for grayscale data 2017-04-09 20:53:53 +09:00
nagadomi 71a34393b8 Add support for padding in convert_data.lua 2017-04-08 22:01:13 +09:00
nagadomi eb3926b61c Fix to work on Lua52
web.lua does not work on Lua52 because turbo does not support Lua52 FFI.
2017-03-02 14:47:49 +09:00
nagadomi cdafbf00ae Add BCE(binary cross entropy) loss for segmentation
Sigmoid() output is required.
2017-02-23 08:48:00 +09:00
nagadomi b111901cbb add -update_criterion option for back compatible 2017-02-12 01:56:03 +09:00
nagadomi 0eccbc6555 Add support for MSE loss 2017-01-09 13:00:04 +09:00
nagadomi d2cfb8f104 Add L1 criterion. Change the criterion of updating model 2016-12-05 10:32:26 +09:00
nagadomi 9a13392f91 Change the default -random_blur_sigma_max 0.75->1.0 2016-11-06 18:14:27 +09:00
nagadomi bdaca16c67 perfomance tuning 2016-11-02 23:42:14 +09:00
nagadomi b066761cdc Add data augmentation for user method 2016-10-21 18:34:48 +09:00
nagadomi 50fd999c38 Add -gpu option 2016-10-18 19:54:00 +09:00
nagadomi 5a3d012f4e Add random blur method for data augmentation 2016-09-24 05:32:33 +09:00
nagadomi f11108cba1 Add comment about RNG seed 2016-09-14 21:19:59 +09:00
nagadomi 81df729a8a Remove -loss option 2016-07-09 15:05:11 +09:00
nagadomi 06e073089b Fix for edac608f 2016-07-06 08:04:24 +09:00
nagadomi d8fcde46f3 Change default -validation_crops to 200. 2016-06-20 15:56:26 +09:00
nagadomi f61d59d28a Change default params. enabling jpeg_chroma_subsampling by default 2016-06-18 20:27:34 +09:00
nagadomi 3b1b2d03fe Change default -nr_rate to 0.65 2016-06-13 12:44:32 +09:00
nagadomi b8ff8c6787 Remove -gamma_correction option 2016-06-10 07:37:39 +09:00
nagadomi 01b2e6d441 Remove -upsampling_filter option 2016-06-10 07:34:11 +09:00
nagadomi e5cfd3dfce Add -resume option 2016-06-09 02:39:52 +09:00
nagadomi 307ae40883 Add noise_scale training 2016-06-08 06:39:36 +09:00
nagadomi d0630d3a20 individual filters and box-only support 2016-06-06 14:04:13 +09:00
nagadomi c89fd7249a Add learning_rate_decay 2016-06-02 10:11:15 +09:00
nagadomi 8088460a20 Add oracle_rate option 2016-05-27 16:54:29 +09:00
nagadomi 7814691cbf Add resize_blur parameter
latest graphicsmagick is required
2016-05-21 09:54:12 +09:00
nagadomi 145b47dbf5 Add use_transparent_png option 2016-05-19 23:02:02 +09:00
nagadomi 8d3950b90a Change the default parameter (epoch, downsampling_filters) 2016-05-15 11:33:34 +09:00
nagadomi 51ae485cd1 Add new models
upconv_7 is 2.3x faster than previous model
2016-05-13 09:49:53 +09:00
nagadomi 7af5c9443d Add model option and 12 layers net 2016-04-23 09:19:03 +09:00
nagadomi 8d6451a51b Add -max_training_image_size option 2016-04-10 23:06:39 +09:00
nagadomi 30fe5db735 Add upsampling_filter option 2016-04-02 22:03:27 +09:00
nagadomi 13f702b968 Add support for resizing with gamma correction 2016-03-28 19:07:09 +09:00
nagadomi 7f448a98c4 Change default validation_crops(160) 2016-03-22 10:19:52 +09:00
nagadomi cef98be88f Change default downsampling_filters (Box,Catrom) 2016-03-22 10:18:59 +09:00
nagadomi 8a799e2d56 Add support for noise level 3 2016-03-18 15:47:53 +09:00
nagadomi d4833160c7 Optionalize downsampling filters 2016-03-17 17:58:37 +09:00
nagadomi 4d115e4bdb Add support for plotting loss chart 2016-03-14 05:06:14 +09:00
nagadomi 947746b32b Change default parameter 2015-12-11 16:31:24 +09:00
nagadomi 9f935835dd Add -save_history option 2015-12-04 18:49:34 +09:00
nagadomi aaac6ed6e5 Refactor training loop
more shuffle
2015-11-30 17:18:52 +09:00
nagadomi c72ec3112b Add -random_unsharp_mask_rate option for photo 2015-11-27 18:36:36 +09:00