244 Commits
Author SHA1 Message Date
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