1855 Commits

Author SHA1 Message Date
torzdf
81b6002c35 Bugfixes:
Some checks failed
ci/build / conda (ubuntu-latest, cpu 3.11) (push) Has been cancelled
ci/build / conda (ubuntu-latest, cpu 3.12) (push) Has been cancelled
ci/build / conda (ubuntu-latest, cpu 3.13) (push) Has been cancelled
ci/build / conda (windows-latest, cpu 3.11) (push) Has been cancelled
ci/build / conda (windows-latest, cpu 3.12) (push) Has been cancelled
ci/build / conda (windows-latest, cpu 3.13) (push) Has been cancelled
ci/build / conda (ubuntu-latest, nvidia 3.11) (push) Has been cancelled
ci/build / conda (ubuntu-latest, nvidia 3.12) (push) Has been cancelled
ci/build / conda (ubuntu-latest, nvidia 3.13) (push) Has been cancelled
ci/build / conda (windows-latest, nvidia 3.11) (push) Has been cancelled
ci/build / conda (windows-latest, nvidia 3.12) (push) Has been cancelled
ci/build / conda (windows-latest, nvidia 3.13) (push) Has been cancelled
ci/build / conda (ubuntu-latest, rocm 3.11) (push) Has been cancelled
ci/build / conda (ubuntu-latest, rocm 3.12) (push) Has been cancelled
ci/build / conda (ubuntu-latest, rocm 3.13) (push) Has been cancelled
ci/build / pip (ubuntu-latest, cpu 3.11) (push) Has been cancelled
ci/build / pip (ubuntu-latest, cpu 3.12) (push) Has been cancelled
ci/build / pip (ubuntu-latest, cpu 3.13) (push) Has been cancelled
ci/build / pip (windows-latest, cpu 3.11) (push) Has been cancelled
ci/build / pip (windows-latest, cpu 3.12) (push) Has been cancelled
ci/build / pip (windows-latest, cpu 3.13) (push) Has been cancelled
- Convert: Fix error when 'learn mask' has been selected
  - Bisenet-FP mask: Correctly store mask for face or head centering
2025-12-31 14:24:16 +00:00
torzdf
c53dcca287 Bugfixes:
- Keras 2 to 3, remove 'group' parameter from Conv2DTranspose
  - Suppress Keras warning about backend padding differences
  - Flatten non-list Fully Connected inputs to silence Keras warnings
2025-12-27 14:01:17 +00:00
torzdf
030b4fa879 bugfix: Keras 2to3, store dtypes in Keras3 friendly manner 2025-12-23 15:10:48 +00:00
torzdf
837bc2d51d Faceswap 3 (#1516)
* FaceSwap 3 (#1515)
* Update extract pipeline
* Update requirements + setup for nvidia
* Remove allow-growth option
* tf.keras to keras updates
* lib.model.losses - Port + fix all loss functions for Keras3
* lib.model - port initializers, layers. normalization to Keras3
* lib.model.autoclip to Keras 3
* Update mixed precision layer storage
* model file to .keras format
* Restructure nn_blocks to initialize layers in __init__
* Tensorboard
  - Trainer: Add Torch compatible Tensorboard callbacks
  - GUI event reader remove TF dependency
* Loss logging
  - Flush TB logs on save
  - Replace TB live iterator for GUI
* Backup models on total loss drop rather than per side
* Update all models to Keras3 Compat
* Remove lib.model.session
* Update clip ViT to Keras 3
* plugins.extract.mask.unet-dfl - Fix for Keras3/Torch backend
* Port AdaBelief to Keras 3
* setup.py:
  - Add --dev flag for dev tool install
* Fix Keras 3 syntax
* Fix LR Finder for Keras 3
* Fix mixed precision switching for Keras  3
* Add more optimizers + open up config setting
* train: Remove updating FS1 weights to FS2 models
* Alignments: Remove support for legacy .json files
* tools.model:
  - Remove TF Saved Format saving
  - Fix Backup/Restore + Nan-Scan
* Fix inference model creation for Keras 3
* Preview tool: Fix for Keras3
* setup.py: Configure keras backend
* train: Migration of FS2 models to FS3
* Training: Default coverage to 100%
* Remove DirectML backend
* Update setup for MacOS
* GUI: Force line reading to UTF-8
* Remove redundant Tensorflow references
* Remove redundant code
* Legacy model loading: Fix TFLamdaOp scalar ops and DepthwiseConv2D
* Add vertical offset option for training
* Github actions: Add more python versions
* Add python version to workflow names
* Github workflow: Exclude Python 3.12 for macOS
* Implement custom training loop
* Fs3 - Add RTX5xxx and ROCm 6.1-6.4 support (#1511)
* setup.py: Add Cuda/ROCm version select options
* bump minimum python version to 3.11
* Switch from setup.cgf to pyproject.toml
* Documentation: Update all docs to use automodapi
* Allow sysinfo to run with missing packages + correctly install tk under Linux
* Bugfix: dot naming convention in clip models
* lib.config: Centralise globally rather than passing as object
- Add torch DataParallel for multi-gpu training
  - GUI: Group switches together when generating cli args
  - CLI: Remove deprecated multi-character argparse args
  - Refactor:
    - Centralise tensorboard reading/writing + unit tests
    - Create trainer plugin interfaces + add original + distributed
* Update installers
2025-12-21 02:45:11 +00:00
torzdf
d115fcbabf bugfux: GUI session data. Prevent crash on missing session data 2025-11-23 15:10:40 +00:00
torzdf
102c8b4f90 bugfix: Correctly set learning rate from learning rate finder on restart 2025-11-21 22:51:20 +00:00
torzdf
381b06f6d3 training: Add Learning Rate Warmup 2025-11-21 17:52:45 +00:00
torzdf
05857b63e1 bugfix: Correctly set learning rate from LR Finder when resuming 2025-11-21 15:19:12 +00:00
torzdf
0370d2b77e bugfix: Update installers for conda changes 2025-11-11 15:35:21 +00:00
torzdf
02bcde1089 bugfix: setup.py
- Explicitly use Conda defaults channel
  - Prevent tkinter from pulling in incompatible libs on Linux
2025-11-11 14:59:05 +00:00
torzdf
57027092af bugfix: Correctly generate thumbnails for alignments when frames are missing 2025-11-06 17:51:56 +00:00
torzdf
bcb85b77c2 Merge branch 'staging' 2025-11-04 12:30:02 +00:00
torzdf
50f895d3c1 bugfix: FFL loss - prevent divide by zero errors 2025-11-04 12:29:25 +00:00
torzdf
fbbdf5522c Merge branch 'staging'
Some checks failed
ci/build / conda (${{ matrix.os }}, ${{ matrix.backend }}) (cpu, macos-latest) (push) Has been cancelled
ci/build / conda (${{ matrix.os }}, ${{ matrix.backend }}) (cpu, ubuntu-latest) (push) Has been cancelled
ci/build / conda (${{ matrix.os }}, ${{ matrix.backend }}) (cpu, windows-latest) (push) Has been cancelled
ci/build / conda (${{ matrix.os }}, ${{ matrix.backend }}) (directml, windows-latest) (push) Has been cancelled
ci/build / conda (${{ matrix.os }}, ${{ matrix.backend }}) (nvidia, macos-latest) (push) Has been cancelled
ci/build / conda (${{ matrix.os }}, ${{ matrix.backend }}) (nvidia, ubuntu-latest) (push) Has been cancelled
ci/build / conda (${{ matrix.os }}, ${{ matrix.backend }}) (nvidia, windows-latest) (push) Has been cancelled
ci/build / conda (${{ matrix.os }}, ${{ matrix.backend }}) (rocm, ubuntu-latest) (push) Has been cancelled
ci/build / pip (ubuntu-latest, ${{ matrix.backend }}) (cpu, 3.10) (push) Has been cancelled
ci/build / pip (windows-latest, ${{ matrix.backend }}) (cpu, 3.10) (push) Has been cancelled
ci/build / pip (windows-latest, ${{ matrix.backend }}) (directml, 3.10) (push) Has been cancelled
2025-10-23 18:18:56 +01:00
torzdf
b16b755d55 lr-finder: Log error on NaN 2025-10-23 18:18:39 +01:00
torzdf
b02075647b Merge branch 'staging'
Some checks failed
ci/build / conda (${{ matrix.os }}, ${{ matrix.backend }}) (cpu, macos-latest) (push) Has been cancelled
ci/build / conda (${{ matrix.os }}, ${{ matrix.backend }}) (cpu, ubuntu-latest) (push) Has been cancelled
ci/build / conda (${{ matrix.os }}, ${{ matrix.backend }}) (cpu, windows-latest) (push) Has been cancelled
ci/build / conda (${{ matrix.os }}, ${{ matrix.backend }}) (directml, windows-latest) (push) Has been cancelled
ci/build / conda (${{ matrix.os }}, ${{ matrix.backend }}) (nvidia, macos-latest) (push) Has been cancelled
ci/build / conda (${{ matrix.os }}, ${{ matrix.backend }}) (nvidia, ubuntu-latest) (push) Has been cancelled
ci/build / conda (${{ matrix.os }}, ${{ matrix.backend }}) (nvidia, windows-latest) (push) Has been cancelled
ci/build / conda (${{ matrix.os }}, ${{ matrix.backend }}) (rocm, ubuntu-latest) (push) Has been cancelled
ci/build / pip (ubuntu-latest, ${{ matrix.backend }}) (cpu, 3.10) (push) Has been cancelled
ci/build / pip (windows-latest, ${{ matrix.backend }}) (cpu, 3.10) (push) Has been cancelled
ci/build / pip (windows-latest, ${{ matrix.backend }}) (directml, 3.10) (push) Has been cancelled
2025-09-18 14:22:18 +01:00
torzdf
0f27fe19cd Installer: Miniconda fix TOS for Windows 2025-09-18 14:12:07 +01:00
torzdf
161c610c01 Install: Miniconda - Auto Accept TOS 2025-09-18 13:10:27 +01:00
torzdf
c3ffa603a7 Merge branch 'staging' 2025-07-11 18:20:07 +01:00
torzdf
9cda7aacae Bugfix: requirements, pin opencv to prevent numpy2.x install 2025-07-11 18:19:27 +01:00
torzdf
acd0f11749 Merge branch 'staging'
Some checks failed
ci/build / conda (${{ matrix.os }}, ${{ matrix.backend }}) (cpu, macos-latest) (push) Has been cancelled
ci/build / conda (${{ matrix.os }}, ${{ matrix.backend }}) (cpu, ubuntu-latest) (push) Has been cancelled
ci/build / conda (${{ matrix.os }}, ${{ matrix.backend }}) (cpu, windows-latest) (push) Has been cancelled
ci/build / conda (${{ matrix.os }}, ${{ matrix.backend }}) (directml, windows-latest) (push) Has been cancelled
ci/build / conda (${{ matrix.os }}, ${{ matrix.backend }}) (nvidia, macos-latest) (push) Has been cancelled
ci/build / conda (${{ matrix.os }}, ${{ matrix.backend }}) (nvidia, ubuntu-latest) (push) Has been cancelled
ci/build / conda (${{ matrix.os }}, ${{ matrix.backend }}) (nvidia, windows-latest) (push) Has been cancelled
ci/build / conda (${{ matrix.os }}, ${{ matrix.backend }}) (rocm, ubuntu-latest) (push) Has been cancelled
ci/build / pip (ubuntu-latest, ${{ matrix.backend }}) (cpu, 3.10) (push) Has been cancelled
ci/build / pip (windows-latest, ${{ matrix.backend }}) (cpu, 3.10) (push) Has been cancelled
ci/build / pip (windows-latest, ${{ matrix.backend }}) (directml, 3.10) (push) Has been cancelled
2025-05-21 17:58:33 +01:00
torzdf
92ef5aa92d bugfix: setup.py fix badly escaped delimiters for Windows 2025-05-21 17:57:00 +01:00
torzdf
6c947c44f9 Merge branch 'staging' 2025-05-19 16:51:17 +01:00
torzdf
5212589242 Pin imageio-ffmpeg to <0.6.0 2025-05-19 16:48:50 +01:00
torzdf
7d80bdbba3 bugfix: setup.py - Don't delimit package specs 2025-02-26 17:55:36 +00:00
torzdf
41b61f96a4 Update README.md
Some checks failed
ci/build / conda (${{ matrix.os }}, ${{ matrix.backend }}) (cpu, macos-latest) (push) Has been cancelled
ci/build / conda (${{ matrix.os }}, ${{ matrix.backend }}) (cpu, ubuntu-latest) (push) Has been cancelled
ci/build / conda (${{ matrix.os }}, ${{ matrix.backend }}) (cpu, windows-latest) (push) Has been cancelled
ci/build / conda (${{ matrix.os }}, ${{ matrix.backend }}) (directml, windows-latest) (push) Has been cancelled
ci/build / conda (${{ matrix.os }}, ${{ matrix.backend }}) (nvidia, macos-latest) (push) Has been cancelled
ci/build / conda (${{ matrix.os }}, ${{ matrix.backend }}) (nvidia, ubuntu-latest) (push) Has been cancelled
ci/build / conda (${{ matrix.os }}, ${{ matrix.backend }}) (nvidia, windows-latest) (push) Has been cancelled
ci/build / conda (${{ matrix.os }}, ${{ matrix.backend }}) (rocm, ubuntu-latest) (push) Has been cancelled
ci/build / pip (ubuntu-latest, ${{ matrix.backend }}) (cpu, 3.10) (push) Has been cancelled
ci/build / pip (windows-latest, ${{ matrix.backend }}) (cpu, 3.10) (push) Has been cancelled
ci/build / pip (windows-latest, ${{ matrix.backend }}) (directml, 3.10) (push) Has been cancelled
2024-11-19 23:12:57 +00:00
torzdf
cbaad146d5 Bugfix: Linux installer - pin git to < 2.45 2024-08-10 12:42:47 +01:00
torzdf
6fe300e601 pin numpy to < 2.0 2024-08-02 18:42:57 +01:00
torzdf
3f53ee9cec Bugfix: Convert - Correctly error if a valid mask has not been selected 2024-08-02 18:35:34 +01:00
torzdf
b6ac7b8039 bugfix: Manual tool. Allow working with image folders at EEN values > 1 2024-07-05 18:46:10 +01:00
torzdf
ea63f1e64a Mask tool. Add ability to output custom imported masks 2024-06-28 13:50:14 +01:00
torzdf
c8652ecaac Training preview: Correctly display blur/kernel amount on mask 2024-06-02 16:25:30 +01:00
torzdf
1f4fcff5dd bugfix: Alignments, prevent duplicate backup alignment file names 2024-05-19 11:04:48 +01:00
torzdf
8c3bc39454 bugfix: Patch writer. Correctly split frame number from the end of filenames 2024-05-10 21:48:38 +01:00
torzdf
13dd5b3a90 bugfix: Alignments - correctly update keys for multiple underscores in the filename 2024-04-26 17:39:46 +01:00
torzdf
0f947791f5 bugfix: Alignments remove version check for video file extension update
- Typofixes
2024-04-21 20:03:42 +01:00
torzdf
be42b04064 Bugfix: Alignment file video key lookup for very old alignment files 2024-04-21 19:55:45 +01:00
torzdf
696692dc08 Fixups
- Deprecations - display correct long argument
  - Minor spelling + linting
2024-04-21 16:22:42 +01:00
torzdf
dce7d98302 lib.align: Split lib.align.alignments to smaller modules:
- Move update objects to own module
  - Move Thumbnails to own module
  - docs update + linting/typing
2024-04-19 13:45:30 +01:00
torzdf
96528ee3e8 lib.align.detected_face: Split Mask objects to own aligned_mask module 2024-04-19 12:25:00 +01:00
torzdf
2bad105dc8 bugfix: Alignment tool, auto-detect alignments
- Random linting and typing
2024-04-19 11:33:52 +01:00
torzdf
d75898f718 Bugfixes for video file alignments storage:
- extract/convert: Load images with correct video extension
  - Manual tool: Cache thumbnails with correct extension
  - Mask tool + Preview tool:: Update legacy alignment keys for pre-video extension storage
2024-04-18 12:45:33 +01:00
torzdf
3f69d9feab manual tool: bugfx: don't error when getting mesh for non-existant face 2024-04-17 14:16:33 +01:00
torzdf
ec2a95adf6 tests: Fix alignment tool process_video test 2024-04-17 14:02:43 +01:00
torzdf
957734dfc0 convert: Bugfix: Update legacy .png video alignments to include video file extension 2024-04-17 12:37:47 +01:00
torzdf
3d5b962d29 linting: typo-fix 2024-04-16 11:15:54 +01:00
torzdf
1c081aea7d Add ability to export and import alignment data (#1383)
* tools.alignments - add export job

* plugins.extract: Update __repr__ for ExtractorBatch dataclass

* plugins.extract: Initial implementation of external import plugins

* plugins.extract: Disable lm masks on ROI alignment data import

* lib.align: Add `landmark_type` property to AlignedFace and return dummy data for ROI Landmarks pose estimate

* plugins.extract: Add centering config item for align import and fix filename mapping for images

* plugins.extract: Log warning on downstream plugins on limited alignment data

* tools: Fix plugins for 4 point ROI landmarks (alignments, sort, mask)

* tools.manual: Fix for 2D-4 ROI landmarks

* training: Fix for 4 point ROI landmarks

* lib.convert: Average color plugin. Avoid divide by zero errors

* extract - external:
  - Default detector to 'external' when importing alignments
  - Handle different frame origin co-ordinates

* alignments: Store video extension in alignments file

* plugins.extract.external: Handle video file keys

* plugins.extract.external: Output warning if missing data

* locales + docs

* plugins.extract.align.external: Roll the corner points to top-left for different origins

* Clean up

* linting fix
2024-04-15 12:19:15 +01:00
torzdf
118e615724 bugfix: tests - change -it switch to -i 2024-04-06 16:53:12 +01:00
torzdf
b1caa03e3f Fix tests for new cli switches 2024-04-06 15:48:54 +01:00
torzdf
95b4431c57 Deprecate multi-character cli switches 2024-04-05 13:51:57 +01:00