Commit Graph

1814 Commits

Author SHA1 Message Date
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
torzdf
64a7b5812e bugfix: setup/sysinfo - Don't look for files in non-existant path 2024-04-03 16:43:37 +01:00
torzdf
ab8199e243 linting + typing 2024-04-03 15:30:46 +01:00
torzdf
70c064ca7d Minor fixups and linting 2024-04-03 15:14:32 +01:00
torzdf
983901466f linting: Reduce Class min-public-methods and fix spacing 2024-04-03 14:37:48 +01:00
torzdf
a9d87ae007 linting: Remove pylint hint for logger 2024-04-03 14:03:54 +01:00
torzdf
97a842112c lib.align.aligned_face - Add split_mask method 2024-04-03 13:57:19 +01:00
torzdf
4502824481 Add mask dilation/erosion option for training 2024-03-26 18:06:48 +00:00
torzdf
2b08b8d2a6 typofix: tools.manual lower loglevel for mask removal 2024-03-25 18:55:29 +00:00
torzdf
180fe97a52 lib.training - Outpu useful error message if non-faceswap training images used 2024-03-22 12:51:50 +00:00
torzdf
aaa63ff896 bugfix: logger, don't import numpy unless required 2024-03-21 11:56:01 +00:00
torzdf
9ddc838e68 Bugfix: Correct loss labels when graphing 2024-03-20 17:08:39 +00:00
torzdf
1d3c59c351 Bugfix: Extract error on rotate faces 2024-03-20 15:24:30 +00:00
torzdf
1009254c57 logging
- Add standardized class __init__ log function
  - Import logger from lib.__init__ to prevent custom log level errors when running non-fs scripts
2024-03-20 14:19:23 +00:00
torzdf
63b4d91281 Bugfix: Mask tool - correctly name imported mask 2024-03-18 19:38:10 +00:00
torzdf
af95dbb310 Bugfix: mask tool. Prevent hangiing when input is folder of frames 2024-03-13 18:21:04 +00:00
torzdf
23944c3c10 Update requirements 2024-03-12 14:42:23 +00:00
torzdf
7a16f753cc
Add Mask importing to the mask tool (#1376)
* lib.align.alignments: expose count_faces_in_frame
* tools.mask: refactor and fix frame output to display all masks in a single frame
* tools.mask: add import mask process
* manual tool: Remove NN masks on landmark edit
2024-03-12 13:17:35 +00:00
torzdf
d1dfce8a13 setup.py - fix ldconfig errors 2024-03-01 11:12:44 +00:00
torzdf
dea021cf82 bugfix - setup.py
- Install xorg-libxft for Linux users
  - Force tensorflow-cpu from pip
2024-01-12 15:14:56 +00:00
torzdf
b25f48431d bugfix: Patch writer - handle filenames which are only numeric 2024-01-10 13:05:35 +00:00
torzdf
2a3845152b bugfix: Convert - patch writer. Fix TIFF metadata 2024-01-07 14:58:48 +00:00
torzdf
b72b730adb setup.py:
- Don't try to call ldconfig when it doesn't exist
  - Adjust Cuda versions to TF compiled versions
  - Multiple package install handling
  - Use XFT version of tk under Linux
2024-01-05 13:46:07 +00:00
torzdf
a62a85c021 bugfix: setup.py - Force pip for imagio-ffmpeg 2023-11-17 02:09:55 +00:00
torzdf
4557331cb9 bugfix: Preview in GUI when converting to vid/gif 2023-10-24 23:47:08 +01:00
torzdf
8e6c6c3500 patch writer: Sort the json file by key 2023-10-04 10:47:59 +01:00
torzdf
3336772ff1 bugfix: patch writer - remove hard coded values 2023-09-30 12:19:04 +01:00
torzdf
f23c87c899 bugfix: Preview tool regression 2023-09-29 14:50:52 +01:00
torzdf
e80ae046d5
Convert - Patch writer (#1352)
* Add cli-args and config opts for patch writer
* Add supporting code to convert funcs
* writers
  - move frame count functions to _base
  - Add kwargs for pre_encode
* scripts/lib convert - add code for patch writer
* lib.image.encode_image - Add cv2.imencode args
* patch_defaults: Add face index location option
* Add patch writer with PNG support
* Send correct matrix to patch plugin
* Add canvas origin option
* Add Tiff format to Patch Writer
* Docs and locales
* Add ROI to output
* convert: choose warp border by face count
2023-09-28 11:55:38 +01:00
torzdf
a660eda8e1 Convert: Add face scale option 2023-09-09 18:40:36 +01:00
torzdf
48aca1462a bugfix: LR finder. Reset optimizer state 2023-09-08 09:48:09 +01:00
torzdf
8388241db9 bugfix: Always load images as 3 channel color 2023-09-01 22:36:31 +01:00
torzdf
4588adbd16 remove debug code 2023-09-01 19:44:57 +01:00
torzdf
460a229e79 bugfix: edge case NaN in pose estimation 2023-09-01 19:43:05 +01:00
torzdf
603fdc2a35 Phaze A - Allow outputs divisible by 16 2023-09-01 09:50:47 +01:00
torzdf
5d00025884 bugfix: error loading images with special chars 2023-08-27 02:15:55 +01:00
torzdf
cf0efff6ba
Learning Rate finder (#1341)
* Add LR Finder support structure

* Move trainer.Feeder to lib.training.generator

* Expose model.io

* Add lr_finder

* Update docs and locales

* Pre-PR fixups
  - Fix training graph not displaying
  - CI fixes
  - Switch lr finder progress to tqdm
  - Exit lr finder early on NaN
  - Display lr finder progress in GUI
2023-08-22 17:07:36 +01:00
torzdf
68a3322748 bugfix: train - fix learn_mask training 2023-08-09 19:37:24 +01:00
torzdf
0bba6ffd8b bugfix: train - fix occasional mem leak in preview 2023-08-06 13:16:32 +01:00
torzdf
81e3bf5cbe
Merge pull request #1331 from torzdf/macos
macOS Installer
2023-07-14 14:00:44 +01:00
torzdf
31eff9c096 macos fixes
- setup.py - reduce width of progress bar
  - macos bash install script
  - macos app launcher
2023-07-14 01:40:41 +01:00
torzdf
12814ce049 bugfix: Manual tool - face filtering 2023-07-12 08:23:48 +01:00