Commit Graph

2145 Commits (b253da6e353f0170c3eb60fe299c41d2fa21db50)
 

Author SHA1 Message Date
James Betker 299d855b34 Enable forced learning rates 2020-06-07 16:56:05 +07:00
James Betker efb5b3d078 Add switched_conv 2020-06-07 16:45:07 +07:00
James Betker 063719c5cc Fix attention conv bugs 2020-06-06 18:31:02 +07:00
James Betker cbedd6340a Add RRDB with attention 2020-06-05 21:02:08 +07:00
James Betker ef5d8a0ed1 Misc 2020-06-05 21:01:50 +07:00
James Betker 318a604405 Allow weighting of input data
This essentially allows you to give some datasets more
importance than others for the purposes of reaching a more
refined network.
2020-06-04 10:05:21 +07:00
James Betker edf0f8582e Fix rrdb bug 2020-06-02 11:15:55 +07:00
James Betker dc17545083 Add RRDB Initial Stride
Allows downsampling immediately before processing, which reduces network complexity on
higher resolution images but keeps a higher filter count.
2020-06-02 10:47:15 +07:00
James Betker 76a38b6a53 Misc 2020-06-02 09:35:52 +07:00
James Betker 726d1913ac Allow validating in batches, remove val size limit 2020-06-02 08:41:22 +07:00
James Betker 90125f5bed Allow blurring to be specified 2020-06-02 08:40:52 +07:00
James Betker 8355f3d1b3 Only log discriminator data when gan is activated 2020-06-01 15:48:16 +07:00
James Betker f1a1fd14b1 Introduce (untested) colab mode 2020-06-01 15:09:52 +07:00
James Betker a38dd62489 Only train discriminator/gan losses when gan_w > 0 2020-06-01 15:09:10 +07:00
James Betker 1eb9c5a059 Fix grayscale 2020-05-29 22:04:50 +07:00
James Betker 2b03e40f98 Debug process_video 2020-05-29 20:44:50 +07:00
James Betker 74b313aaa9 Add grayscale downsampling option 2020-05-29 20:34:00 +07:00
James Betker b123ed8a45 Add attention resnet
Not ready for prime time, but is a first draft.
2020-05-29 20:02:10 +07:00
James Betker 5e9da65d81 Fix process_video bugs 2020-05-29 12:47:22 +07:00
James Betker beac71ad18 Allow minivids to start at any specified number 2020-05-28 20:27:42 +07:00
James Betker 57682ebee3 Separate feature extractors out, add resnet feature extractor 2020-05-28 20:26:30 +07:00
James Betker 156cee240a Remove reliance on magick, only wait for ffmpeg at last second, fix image ordering issue 2020-05-27 23:09:46 +07:00
James Betker b551e86adb Encode videos to HEVC
MP4 doesnt support 8K
2020-05-27 20:04:45 +07:00
James Betker bdeafad8c5 Only run 'convert' not magick 2020-05-27 17:24:10 +07:00
James Betker 41c1efbf56 Add dynamic video processing script 2020-05-27 17:09:11 +07:00
James Betker f745be9dea Fix vgg disc arch 2020-05-27 13:31:22 +07:00
James Betker 6962ccb306 Adjust motion blur
0 is invalid.
2020-05-27 13:09:46 +07:00
James Betker f6815df58b Misc 2020-05-27 08:04:47 +07:00
James Betker e27a49454e Enable vertical splitting on inference images to support very large resolutions. 2020-05-27 08:04:35 +07:00
James Betker 96ac26a8b7 Allow injection of random low-amplitude noise & motion blur into generator 2020-05-27 08:04:11 +07:00
James Betker 69cbfa2f0c Adjust dataset mutations a bit
Adjusts the compression artfacts to be more aggressive, and blurring to be less aggressive.
2020-05-26 13:48:34 +07:00
James Betker 2931142458 Allow multiple gt image dirs 2020-05-25 19:21:09 +07:00
James Betker 4e44b8a1aa Clean up video stuff 2020-05-25 19:20:49 +07:00
James Betker 8464cae168 HQ blurring doesnt actually work right - hq images arent the right size when they are blurred
Just revert it and blur the lq images..
2020-05-24 22:32:54 +07:00
James Betker 5fd8749cf2 More updates - need more blurring 2020-05-24 22:13:27 +07:00
James Betker 9627cc2c49 Update HR gaussian blur params 2020-05-24 18:00:31 +07:00
James Betker 2f8b0250b9 Blur HR image before downsizing, when available 2020-05-24 17:18:44 +07:00
James Betker cc4571eb8d Randomize blur effect 2020-05-24 12:35:41 +07:00
James Betker 27a548c019 Enable blurring via settings 2020-05-24 11:56:39 +07:00
James Betker 3c2e5a0250 Apply fixes to resgen 2020-05-24 07:43:23 +07:00
James Betker 446322754a Support generators that don't output intermediary values. 2020-05-23 21:09:54 +07:00
James Betker 987cdad0b6 Misc mods 2020-05-23 21:09:38 +07:00
James Betker 9b44f6f5c0 Add AssistedRRDB and remove RRDBNetXL 2020-05-23 21:09:21 +07:00
James Betker 445e7e7053 Extract subimages mod 2020-05-23 21:07:41 +07:00
James Betker 90073fc761 Update LQ_dataset to support inference on split image videos 2020-05-23 21:05:49 +07:00
James Betker 74bb0fad33 Allow dataset classes to add noise internally 2020-05-23 21:04:24 +07:00
James Betker af1968f9e5 Allow passthrough discriminator to have passthrough disabled from config 2020-05-19 09:41:16 +07:00
James Betker 67139602f5 Test modifications
Allows bifurcating large images put into the test pipeline

This code is fixed and not dynamic. Needs some fixes.
2020-05-19 09:37:58 +07:00
James Betker 6400607fc5 ONNX export support 2020-05-19 09:36:04 +07:00
James Betker 89c71293ce IDEA update 2020-05-19 09:35:26 +07:00