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 -06:00 |
|
James Betker
|
edf0f8582e
|
Fix rrdb bug
|
2020-06-02 11:15:55 -06: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 -06:00 |
|
James Betker
|
76a38b6a53
|
Misc
|
2020-06-02 09:35:52 -06:00 |
|
James Betker
|
726d1913ac
|
Allow validating in batches, remove val size limit
|
2020-06-02 08:41:22 -06:00 |
|
James Betker
|
90125f5bed
|
Allow blurring to be specified
|
2020-06-02 08:40:52 -06:00 |
|
James Betker
|
8355f3d1b3
|
Only log discriminator data when gan is activated
|
2020-06-01 15:48:16 -06:00 |
|
James Betker
|
f1a1fd14b1
|
Introduce (untested) colab mode
|
2020-06-01 15:09:52 -06:00 |
|
James Betker
|
a38dd62489
|
Only train discriminator/gan losses when gan_w > 0
|
2020-06-01 15:09:10 -06:00 |
|
James Betker
|
1eb9c5a059
|
Fix grayscale
|
2020-05-29 22:04:50 -06:00 |
|
James Betker
|
2b03e40f98
|
Debug process_video
|
2020-05-29 20:44:50 -06:00 |
|
James Betker
|
74b313aaa9
|
Add grayscale downsampling option
|
2020-05-29 20:34:00 -06:00 |
|
James Betker
|
b123ed8a45
|
Add attention resnet
Not ready for prime time, but is a first draft.
|
2020-05-29 20:02:10 -06:00 |
|
James Betker
|
5e9da65d81
|
Fix process_video bugs
|
2020-05-29 12:47:22 -06:00 |
|
James Betker
|
beac71ad18
|
Allow minivids to start at any specified number
|
2020-05-28 20:27:42 -06:00 |
|
James Betker
|
57682ebee3
|
Separate feature extractors out, add resnet feature extractor
|
2020-05-28 20:26:30 -06: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 -06:00 |
|
James Betker
|
b551e86adb
|
Encode videos to HEVC
MP4 doesnt support 8K
|
2020-05-27 20:04:45 -06:00 |
|
James Betker
|
bdeafad8c5
|
Only run 'convert' not magick
|
2020-05-27 17:24:10 -06:00 |
|
James Betker
|
41c1efbf56
|
Add dynamic video processing script
|
2020-05-27 17:09:11 -06:00 |
|
James Betker
|
f745be9dea
|
Fix vgg disc arch
|
2020-05-27 13:31:22 -06:00 |
|
James Betker
|
6962ccb306
|
Adjust motion blur
0 is invalid.
|
2020-05-27 13:09:46 -06:00 |
|
James Betker
|
f6815df58b
|
Misc
|
2020-05-27 08:04:47 -06:00 |
|
James Betker
|
e27a49454e
|
Enable vertical splitting on inference images to support very large resolutions.
|
2020-05-27 08:04:35 -06:00 |
|
James Betker
|
96ac26a8b7
|
Allow injection of random low-amplitude noise & motion blur into generator
|
2020-05-27 08:04:11 -06: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 -06:00 |
|
James Betker
|
2931142458
|
Allow multiple gt image dirs
|
2020-05-25 19:21:09 -06:00 |
|
James Betker
|
4e44b8a1aa
|
Clean up video stuff
|
2020-05-25 19:20:49 -06: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 -06:00 |
|
James Betker
|
5fd8749cf2
|
More updates - need more blurring
|
2020-05-24 22:13:27 -06:00 |
|
James Betker
|
9627cc2c49
|
Update HR gaussian blur params
|
2020-05-24 18:00:31 -06:00 |
|
James Betker
|
2f8b0250b9
|
Blur HR image before downsizing, when available
|
2020-05-24 17:18:44 -06:00 |
|
James Betker
|
cc4571eb8d
|
Randomize blur effect
|
2020-05-24 12:35:41 -06:00 |
|
James Betker
|
27a548c019
|
Enable blurring via settings
|
2020-05-24 11:56:39 -06:00 |
|
James Betker
|
3c2e5a0250
|
Apply fixes to resgen
|
2020-05-24 07:43:23 -06:00 |
|
James Betker
|
446322754a
|
Support generators that don't output intermediary values.
|
2020-05-23 21:09:54 -06:00 |
|
James Betker
|
987cdad0b6
|
Misc mods
|
2020-05-23 21:09:38 -06:00 |
|
James Betker
|
9b44f6f5c0
|
Add AssistedRRDB and remove RRDBNetXL
|
2020-05-23 21:09:21 -06:00 |
|
James Betker
|
445e7e7053
|
Extract subimages mod
|
2020-05-23 21:07:41 -06:00 |
|
James Betker
|
90073fc761
|
Update LQ_dataset to support inference on split image videos
|
2020-05-23 21:05:49 -06:00 |
|
James Betker
|
74bb0fad33
|
Allow dataset classes to add noise internally
|
2020-05-23 21:04:24 -06:00 |
|
James Betker
|
af1968f9e5
|
Allow passthrough discriminator to have passthrough disabled from config
|
2020-05-19 09:41:16 -06: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 -06:00 |
|
James Betker
|
6400607fc5
|
ONNX export support
|
2020-05-19 09:36:04 -06:00 |
|
James Betker
|
89c71293ce
|
IDEA update
|
2020-05-19 09:35:26 -06:00 |
|
James Betker
|
9cde58be80
|
Make RRDB usable in the current iteration
|
2020-05-16 18:36:30 -06:00 |
|
James Betker
|
b95c4087d1
|
Allow an alt_path for saving models and states
|
2020-05-16 09:10:51 -06:00 |
|
James Betker
|
f911ef0d3e
|
Add corruptor_usage_probability
Governs how often a corruptor is used, vs feeding uncorrupted images.
|
2020-05-16 09:05:43 -06:00 |
|
James Betker
|
635c53475f
|
Restore swapout models just before a checkpoint
|
2020-05-16 07:45:19 -06:00 |
|
James Betker
|
877be4d88c
|
README update
|
2020-05-15 14:03:44 -06:00 |
|