James Betker
|
31cf1ac98d
|
Retrofit full_image_dataset to work with new arch.
|
2020-10-27 10:26:19 -06:00 |
|
James Betker
|
ff58c6484a
|
Fixes to unified chunk datasets to support stereoscopic training
|
2020-10-26 11:12:22 -06:00 |
|
James Betker
|
9c3d059ef0
|
Updates to be able to train flownet2 in ExtensibleTrainer
Only supports basic losses for now, though.
|
2020-10-24 11:56:39 -06:00 |
|
James Betker
|
8e5b6682bf
|
Add PairedFrameDataset
|
2020-10-23 20:58:07 -06:00 |
|
James Betker
|
f9dc472f63
|
Misc nonfunctional mods to datasets
|
2020-10-22 10:16:17 -06:00 |
|
James Betker
|
aba83e7497
|
Don't apply jpeg corruption & noise corruption together
This causes some severe noise.
|
2020-10-20 12:56:35 -06:00 |
|
James Betker
|
111450f4e7
|
Use areal interpolate for multiscale_dataset
|
2020-10-19 15:30:25 -06:00 |
|
James Betker
|
9b9a6e5925
|
Add get_paths() to base_unsupervised_image_dataset
|
2020-10-19 11:30:06 -06:00 |
|
James Betker
|
ffad0e0422
|
Allow image corruption in multiscale dataset
|
2020-10-19 10:10:27 -06:00 |
|
James Betker
|
9ead2c0a08
|
Multiscale training in!
|
2020-10-17 22:54:12 -06:00 |
|
James Betker
|
b008a27d39
|
Spinenet should allow bypassing the initial conv
This makes feeding in references for recurrence easier.
|
2020-10-17 20:16:47 -06:00 |
|
James Betker
|
c7f3fc4dd9
|
Enable chunk_with_reference to work without centers
Moving away from this so it doesn't matter too much. Also fixes an issue
with the "ignore" flag.
|
2020-10-17 20:09:08 -06:00 |
|
James Betker
|
b45e132a9d
|
Allow first n tiles to be ignored
Helps zoom in with chunked dataset
|
2020-10-17 09:45:03 -06:00 |
|
James Betker
|
1ba01d69b5
|
Move datasets to INTER_AREA interpolation for downsizing
Looks **FAR** better visually
|
2020-10-15 17:18:23 -06:00 |
|
James Betker
|
d56745b2ec
|
JPEG-broad adjustment
|
2020-10-15 10:14:51 -06:00 |
|
James Betker
|
1dc0b05428
|
Add multiscale dataset
|
2020-10-15 10:12:50 -06:00 |
|
James Betker
|
0f4e03183f
|
New image corruptor gradations
|
2020-10-15 10:12:25 -06:00 |
|
James Betker
|
24792bdb4f
|
Codebase cleanup
Removed a lot of legacy stuff I have no intent on using again.
Plan is to shape this repo into something more extensible (get it? hah!)
|
2020-10-13 20:56:39 -06:00 |
|
James Betker
|
e7cf337dba
|
Fix bug with chunk_with_reference
|
2020-10-12 10:23:03 -06:00 |
|
James Betker
|
92cb83958a
|
Return zeros rather than None when image cant be read
|
2020-10-11 08:33:18 -06:00 |
|
James Betker
|
b3d0baaf17
|
Improve multiframe dataset memory usage
|
2020-10-09 08:40:00 -06:00 |
|
James Betker
|
b36ba0460c
|
Fix multi-frame dataset OBO error
|
2020-10-08 12:21:04 -06:00 |
|
James Betker
|
efbf6b737b
|
Update validate_data to work with SingleImageDataset
|
2020-10-02 08:58:34 -06:00 |
|
James Betker
|
57814f18cf
|
More features for multi-frame-dataset
|
2020-09-28 14:26:15 -06:00 |
|
James Betker
|
7dff802144
|
Add MultiFrameDataset
Retrieves video sequence patches rather than single images.
|
2020-09-27 11:13:06 -06:00 |
|
James Betker
|
d8c3fc9327
|
Fix random noise corruptor
It was functioning as a color shift
|
2020-09-27 11:12:24 -06:00 |
|
James Betker
|
c85da79697
|
Move many dataset functions into a base class
|
2020-09-27 11:11:58 -06:00 |
|
James Betker
|
5a27187c59
|
More mods to accomodate new dataset
|
2020-09-25 22:45:57 -06:00 |
|
James Betker
|
254cb1e915
|
More dataset integration work
|
2020-09-25 22:19:38 -06:00 |
|
James Betker
|
ce4613ecb9
|
Finish up single_image_dataset work
Sweet!
|
2020-09-25 16:37:54 -06:00 |
|
James Betker
|
1cf73c2cce
|
Fix dataset for a val set that includes lq
|
2020-09-24 18:01:07 -06:00 |
|
James Betker
|
dab8ab8a8f
|
Offer option to configure the size of the normal distribution that the target size is drawn from
|
2020-09-20 11:59:31 -06:00 |
|
James Betker
|
e2a146abc7
|
Add in experiments hook
|
2020-09-19 10:05:25 -06:00 |
|
James Betker
|
4f75cf0f02
|
Revert "Full image dataset operates on lists"
Going with an entirely new dataset instead..
This reverts commit 36ec32bf11.
|
2020-09-18 09:50:43 -06:00 |
|
James Betker
|
36ec32bf11
|
Full image dataset operates on lists
|
2020-09-18 09:50:26 -06:00 |
|
James Betker
|
3cb2a9a9d3
|
New dataset, initial work
|
2020-09-18 09:49:13 -06:00 |
|
James Betker
|
f5cd23e2d5
|
Further patch size adjustments
|
2020-09-16 16:50:35 -06:00 |
|
James Betker
|
0b047e5f80
|
Increase scale of the patch selector random distribution
This will cause larger slices of an image to appear more frequently,
increasing the difficulty of the generator.
|
2020-09-16 08:27:42 -06:00 |
|
James Betker
|
94deab2792
|
Fix error serving gt_fullsize_ref in full_image_dataset
|
2020-09-14 15:05:44 -06:00 |
|
James Betker
|
5189b11dac
|
Add combined dataset for training across multiple datasets
|
2020-09-11 08:44:06 -06:00 |
|
James Betker
|
3027e6e27d
|
Enable amp to be disabled
|
2020-09-09 10:45:59 -06:00 |
|
James Betker
|
a5c2388368
|
Use lower LQ image size when it is being fed in
|
2020-09-06 17:26:32 -06:00 |
|
James Betker
|
17aa205e96
|
New dataset that reads from lmdb
|
2020-09-04 17:32:57 -06:00 |
|
James Betker
|
8580490a85
|
Reduce usage of resize operations when not needed in dataloaders.
|
2020-09-04 15:31:24 -06:00 |
|
James Betker
|
6226b52130
|
Pin memory in dataloaders by default
|
2020-09-04 15:30:46 -06:00 |
|
James Betker
|
03eb29a4d9
|
Fix LQGT dataset
|
2020-08-25 11:57:25 -06:00 |
|
James Betker
|
a65b07607c
|
Reference network
|
2020-08-25 11:56:59 -06:00 |
|
James Betker
|
f224907603
|
Fix LQGT_dataset, add full_image_dataset
|
2020-08-24 17:12:43 -06:00 |
|
James Betker
|
5ec04aedc8
|
Let noise be configurable
LQ noise is not currently configurable for some reason..
|
2020-08-24 15:00:14 -06:00 |
|
James Betker
|
dffc15184d
|
More ExtensibleTrainer work
It runs now, just need to debug it to reach performance parity with SRGAN. Sweet.
|
2020-08-23 17:22:45 -06:00 |
|