Commit Graph

254 Commits

Author SHA1 Message Date
torzdf
98d01760e4 Overhaul sort:
- Standardize image data reading and writing
  - Optimize loading (just one pass required)
  - Make all sort groups binnable (to greater or lesser results)
  - Add sort by pitch
  - Deprecate multiple options
  - linting, docs + locales
2022-09-13 13:08:59 +01:00
torzdf
6985495902 sort - lower log level 2022-09-07 12:45:20 +01:00
torzdf
89d124c311 Sort - Add Batch Mode. Speed up yaw sort 2022-09-07 12:40:19 +01:00
torzdf
58926e9328 Typofix: tools.preview 2022-08-27 00:03:03 +01:00
torzdf
1022651eb8 Bugfix: convert - Gif Writer
- Fix non-launch error on Gif Writer
  - convert plugins - linting
  - convert/fs_media/preview/queue_manager - typing
  - Change convert items from dict to Dataclass
2022-08-26 23:56:03 +01:00
torzdf
5e73437be4 lib.align updates:
- alignments.py
    - Add typed dicts for imported alignments
    - Explicitly check for presence of thumb value in alignments dict
    - linting
  - detected_face.py
    - Typing
    - Linting
    - Legacy support for pre-aligned face
    - Update dependencies to new property names
2022-08-19 12:36:08 +01:00
torzdf
3295089737 lib.detected_face.Mask
- Add source + target offset and coverage to set_sub_crop method
2022-08-19 10:57:40 +01:00
torzdf
629c02a61e Add "custom" mask 2022-08-05 14:09:05 +01:00
torzdf
eb3c612f29 Replace restore tool with model tool 2022-06-28 14:27:23 +01:00
torzdf
9d55ade27f amd gpu-stats bugfix 2022-06-18 11:16:03 +01:00
torzdf
2298d51237 Preview Tool fixes
- Prevent errors from getting swallowed
  - Correctly handle None mask type
  - Consistent box blending
2022-06-08 12:08:18 +01:00
torzdf
f848e8ad0d bugfix - typing for python < 3.9 2022-05-25 12:50:09 +01:00
torzdf
6437cd7ab0 alignments tool - Add from-faces job
- Allows user to regenerate alignments file(s) from a folder of extracted faces
2022-05-24 12:42:52 +01:00
torzdf
a9908b46f7 Alignments tool - Replace 'extract-large' with 'min-size' 2022-05-21 17:06:12 +01:00
torzdf
71c20252c2 bugfix: Preview Tool, ensure all config items are written 2022-05-08 14:18:30 +01:00
torzdf
30872ef265 alignments tool - Don't re-analyze video if metadata in alignments 2022-03-30 15:08:41 +01:00
torzdf
635a24dc42 Remove debug code 2022-03-30 02:56:17 +01:00
torzdf
23d92c1f0d Bugfixes
- Sort - Fix rare help-text parsing bug
  - Manual - Fix issue where frame count is incorrect when een > 1 used on extract
2022-03-30 02:54:29 +01:00
Markus Karileet
abb7b004cb feat: add support for rectangle cursor shape for masking (#1189)
* feat: add support for rectangle cursor shape for masking
* formatting update
* fix bug with initial rectangular point

Co-authored-by: markus <markus@helmes.com>
2021-12-05 14:35:17 +00:00
torzdf
cf4b567cc6 Tensorflow 2.6 Support (#1182)
* lib.cli.launcher - Bump max tf version to 2.6

* Remove pathlib requirement

* Update requirements files

* Update setup.py

* bugfix - GUI: Supress errors when attempting to load previews in extract

* GUI: Suppress ptxas error messages for Windows
2021-09-05 02:25:12 +01:00
torzdf
ee92cff6fc tools.sort - Linting and languages 2021-08-07 12:39:09 +01:00
Olivier Gagnon
acf1fc5612 Add a new sorting/grouping feature based on black pixels in images (#1169)
* Add a new sorting/grouping feature based on the percentage of black pixels in images

Context:
- Faces are quite often moving and can get near borders of the image.
- Because of that, the face image has the out of bound part replaced by black pixels.
- Some are still good for training but others not. Also, one might want to keep an alignment file with the near borders faces but excluded (some of) them for training.
- Having to manually move/delete those images in a dataset of thousands of them is painful and prone to mistakes.
- Thus, having a way to sort and group those images would be helpful. At least, it is for me :-)

Added features:
- A new option in Tools/Sort for both Sort and Group By called Black-Pixels.
- Sort feature: calculates the percentage of black pixels in each images and sort them from 0 to 100%.
- Group By feature: Uses the Bins slider number selected and creates 100/Bins folders with the images percentage of black pixels. ie. a Bins of 5 will create 20 folders with the first one containing images with 0-5% of black pixels, the second one with 6-10%, etc. A static method as been added for the rounding error.

* PEP8 fixes

Co-authored-by: Olivier Gagnon <ogagnon@ludia.com>
2021-08-07 12:24:41 +01:00
torzdf
ac22d40a91 extract: mask - Delete any mask from outside of frame boundaries 2021-07-22 13:04:29 +01:00
torzdf
0775245d29 Bugfix - Manual Tool
- Fix bug when adding new face with "misaligned" filter applied
2021-06-12 17:01:02 +01:00
torzdf
104a549abb Bugfix - Manual Tool
- Fix bug when changing filter modes from a filter with no matches
2021-06-11 19:26:13 +01:00
torzdf
fb6f576e6d Bugfix - Manual Tool
- Fix non-appearing landmark annotations in face viewer
2021-06-11 15:50:27 +01:00
torzdf
cbf64decc9 Typofix 2021-06-10 19:39:12 +01:00
torzdf
44aaac09d7 Bugfixes - Manual Tool
- Update total frames count on a face count change
  - Correct navigation position when decrementing frame after edit changes the filter criteria
2021-06-10 16:22:10 +01:00
torzdf
c9a9361793 bugfix: sort tool - fix group by 2021-06-09 10:51:09 +01:00
torzdf
eb96da0346 Add Misalignment Detection
- lib.align.AlignedFace - Add average_distance property (distance from mean_face)
  - tools.manual - Add misaligned Faces filter
  - tools.sort - Add sort by distance (misaligned sort)Add "Misaligned Faces" filter to manual tool
2021-06-08 19:30:28 +01:00
torzdf
b4087cc06c Manual Tool - Bugfixes
- Fix extract box rotation
  - Fix thumbnail view size
  - Fix issue of deleted landmark meshes still displaying
2021-06-06 12:17:39 +01:00
torzdf
fbdd91ef0e Bugfix - Manual Tool
- Correctly update mesh annotations in frame viewer when there are multiple faces in frame
2021-06-03 11:19:23 +01:00
torzdf
5010466f2e bugfix: Manual tool. Fix misaligned landmarks in face viewer after adding/removing a face 2021-06-02 13:25:28 +01:00
torzdf
6321ca64e4 Bug fix - Manual Tool - Fix a first time indexing bug when EEN > 1 2021-06-01 12:30:31 +01:00
torzdf
a26bc50891 Manual Tool - Update to support masks with different centering 2021-05-31 18:58:34 +01:00
torzdf
d20b04ef84 tools.mask - Correctly output masks when generating masks with multi-config options 2021-05-20 11:52:24 +01:00
torzdf
2b896df498 mask tool - Update to support masks that support multi-centering 2021-05-18 19:36:01 +01:00
torzdf
6c439944cd Masks updates
- Add head centering support to training
  - Update helptext/tooltips
2021-05-17 18:20:08 +01:00
torzdf
40798fd27f core mask updates:
- Add support for different mask centering
  - Update legacy alignments to store mask centering
  - Bugfix: lib.image ImageIO. Ensure unique queues are created (fixes mask tool when Face is input and an output folder is provided)
2021-05-16 13:24:00 +01:00
torzdf
eb0cc9c271 alignments tool - Update alignments file tooltip 2021-05-08 11:39:39 +01:00
pfakanator
a68a9eda4f Introduce sorting by FFT filtered blur detection. (#1147)
* Update cli.py
* Update sort.py
2021-04-19 10:48:46 +01:00
torzdf
ed8a38677a tools.sort - Mask out background for sort by blur 2021-04-13 12:05:02 +00:00
torzdf
616bd7c500 GUI fixes
- Split themes and styles to own module
  - Fix border on console window
  - variable name updates to make pep8 happy
2021-03-25 19:29:29 +00:00
torzdf
2bcb7d572a tools.sort
- Add sort by size
  - Centralize some of the image loading routines
2021-03-23 16:50:37 +00:00
torzdf
5f598ca825 Locales Update 2021-03-22 18:41:55 +00:00
torzdf
a49f810e7c GUI - Make tooltips themeable 2021-03-20 14:08:07 +00:00
torzdf
544a95bc4f linting 2021-03-19 01:09:52 +00:00
Dominik Miszkiewicz
86ac596784 Update sorting by yaws (#1131) 2021-03-19 01:06:26 +00:00
torzdf
a656552204 Minor Updates
- GUI - Tweaks
    - Fix Analysis Slider background color
    - More granular theming controls
    - Standardize comboboxes and scrollbars
  - plugins.train.model._base
      - Catch more model corruption errors
2021-03-18 17:43:56 +00:00
torzdf
cff0f5698b GUI updates
- Control panel fixes
  - Basic themeing support
  - Fix tools to display more natively
  - Give all config items a group
2021-03-16 19:46:14 +00:00