torzdf
02239d4183
locales: Train settings
2023-06-11 23:30:05 +01:00
torzdf
0c1d1e3253
Bugfix: Color order on preview with learn_mask
2023-02-14 13:09:46 +00:00
torzdf
2028579756
Preview - Allow mask color/opacity configuration
...
- typing: lib.config
- docs: lib.config
- bugfix: popup_configure - keep assets on top
2023-02-13 15:03:34 +00:00
torzdf
80f63280ca
Bugfixes
...
- plugin.train.unbalanced - decoder b
- gui.stats - crash when log folder deleted
- setup.py - install gcc=12.1.0 on linux (scipy dep)
- setup.py - don't test ROCm unless valid
- typofix - cli_args extract singleprocess help
typing
- lib.gui.analysis.event_reader
- lib.gpu_stats.directml - make platform specific
unit tests
- lib.gui.analysis.event_reader
2023-01-13 23:29:51 +00:00
torzdf
28cb2fcffb
Unhide options for ROCm
2023-01-11 14:34:30 +00:00
torzdf
2f16f8aa3b
Bugfix - Phaze-A, monkeypatch EfficientNet for MixedPrecision
2023-01-07 19:19:07 +00:00
torzdf
dc94ed9cad
DirectML Support ( #1291 )
...
* DML Support
- Add requirements. Update setup.py. Update Windows Installer
- Add 'directml' backend. Update backend supporting logic
- GPUStats - Add GPU querying from Win SDK for DML backend
- Windows PlaidML - Deprecation Warning
- Remove deprecated '-d' training flag
- Logging in DML functions
- Unittest
- Windows Unit Test
- lib.utils
2023-01-06 14:45:57 +00:00
torzdf
d750af1e92
bump TF to version 2.10
2022-12-20 15:30:26 +00:00
torzdf
84ae6006bb
Bugfix - TF Version check
2022-12-19 11:26:32 +00:00
torzdf
403f981e0f
bugfix: Catch learn-mask error when no mask type selected
2022-10-12 12:03:59 +01:00
torzdf
7da2cc3dd2
Training - Use custom preview pop-out
2022-09-07 11:49:52 +01:00
torzdf
477e3e2013
bugfix: Train: Toggle mask when 'learn_mask' is selected
2022-09-02 11:50:58 +01:00
torzdf
97da6250ff
typing fix
2022-08-30 12:39:59 +01:00
torzdf
de1068b417
Bugfix: Preview - show predictions at correct size
2022-08-30 12:39:39 +01:00
torzdf
f3b88d5626
bugfix: Get correct output size for learn mask
2022-08-29 01:04:40 +01:00
torzdf
326110f09d
bugfix - timelapse image loader
...
multithreading.py - typing + docs
2022-08-25 12:38:05 +01:00
torzdf
0837273d46
bugfix: CV2 error on previews
2022-08-22 21:31:51 +01:00
torzdf
2beceffad9
Data Augmentation update ( #1263 )
...
- lib.detected_face
- Subclass Masks for Landmark based masks
- Add training mask propery + methods to DetectedFace
- lib.training_training
- subclass TrainingDataGenerator for training and preview data
- Split cache into own module
- Reduce thread count to 1 to prevent image corruption + data re-use
- Process on largest model input/output size rather than stored image size
- Size and crop masks during caching stage
- Implement ring buffer for data flow
- Fix preview reload bug
- augmentation
- typing
- switch color aug order
- better initialization
- Fix warp + landmark warp to correctly apply at different image scales
- Slightly improved warp caching
- Don't store whether image is_preview. Handle all data as training images implicitly
- plugins.trainer: Typing and fixes to work with trainingdata refactor
2022-08-21 18:59:45 +01:00
torzdf
629c02a61e
Add "custom" mask
2022-08-05 14:09:05 +01:00
torzdf
26e26c6288
Update TF to 2.9
...
- Update TF to 2.7 to 2.9
- Bump dependencies
- Remove decode from pynvml calls
- force keras predict functions to non-verbose
- update tests
- update Tensorboard logging
- Update docs
2022-08-02 17:12:26 +01:00
Chatcharin Sangbutsarakum
d5df9e45e7
Fix GUI tooltips on configuration setting window in Original/ Trainer/ Color Augmentation section ( #1252 )
...
* Fix info description of color augmentation group
* Cut sentence due to PEP8
2022-08-02 10:29:47 +01:00
torzdf
dcb436c9df
Bugfix: Handle distribution strat in convert
2022-07-20 12:44:22 +01:00
torzdf
54398559ff
Set default distribution strategy to 'default'
2022-07-20 12:29:14 +01:00
torzdf
2ea05623bd
Update Distibution Strategies:
...
- Add Central Storage Stategy
- Deprecate 'distributed' cli argument
2022-07-19 18:23:16 +01:00
torzdf
89b71f7f1e
bugfix: inference for learn mask with some models
2022-07-11 09:30:33 +01:00
torzdf
9945efb3c0
Bugfix: Phaze-A Fix learn mask for upscales in FC
2022-07-07 23:46:43 +01:00
torzdf
a6332ed324
Merge branch 'master' into staging
2022-07-07 10:06:01 +01:00
torzdf
a3fdee617a
Bugfix - Phaze-A - Correctly scale linear filters
2022-07-07 10:05:49 +01:00
torzdf
582c2ce40c
Add Flip Loss Function
...
- Add Flip for AMD and TF
- Split Perceptual Loss functions to own modules
- Fix allowed input shape for models
- Allow GUI tooltip to display at higher width
2022-07-07 01:02:11 +01:00
torzdf
c8122bc499
bugfix: Stop preview window from stealing focus
2022-07-06 11:15:51 +01:00
torzdf
7b9fc0454d
Live Preview - Replace cv2 with matplotlib viewer
2022-07-01 13:07:01 +01:00
torzdf
b1a8183ab4
scripts.train - type checking
2022-06-30 18:41:58 +01:00
torzdf
a3825dc58f
Bugfix: prevent OOM when just viewing summary
2022-06-29 02:29:22 +01:00
torzdf
d20d3f2ecb
bugfix
2022-06-26 18:21:36 +01:00
torzdf
c279ac3518
Disable saving optimizer state option
2022-06-24 13:43:26 +01:00
torzdf
8ec1e1a091
Nvidia: Add AutoClip training option
2022-06-23 18:00:22 +01:00
torzdf
6168170e76
Force saving optimizer weights on snapshot
2022-06-23 16:10:20 +01:00
torzdf
c3c3483eed
AMD Bugfixes: Save Optimizer Weights
2022-06-23 15:55:24 +01:00
torzdf
06468c97d4
Training: Add setting option to save optimizer weights
2022-06-23 14:54:09 +01:00
torzdf
4d42591ffc
Mixed Precision bugfix for AMD
2022-06-22 12:49:07 +01:00
torzdf
0d2eafa485
Enable change of precision for existing models
2022-06-22 01:58:59 +01:00
torzdf
e2fc070370
Bug fixes
...
- PhazeA tooltip spacing
- Graph live cache bug
2022-06-20 03:38:47 +01:00
torzdf
d7ffcda3b0
Add LPIPS loss function
2022-06-19 18:42:34 +01:00
torzdf
bad5025aea
Core updates
...
- Change loss loading mechanism
- Autosize tooltips based on content size
- Random linting + code modernisation
2022-06-19 12:32:39 +01:00
torzdf
1d434b73a4
Enable LogCosh Loss for AMD
2022-06-18 16:12:39 +01:00
torzdf
76cb535e82
linting
2022-06-18 15:32:02 +01:00
torzdf
308c5edf39
Loss updates
...
- Split Loss to own package
- Add Focal Frequency Loss (Nvidia only)
- Enable GMSD for AMD
2022-06-18 14:33:45 +01:00
torzdf
d9c84a5f9f
Add Laplacian Pyramid Loss
2022-06-18 02:29:19 +01:00
torzdf
04337e0c5e
SSIM Updates
...
- Standardize DSSIM Function
- Implement MSSIM function for AMD
2022-06-18 00:02:28 +01:00
torzdf
d02d51f2bb
doc fixes
2022-06-17 22:01:57 +01:00