Commit Graph

183 Commits

Author SHA1 Message Date
torzdf 7d613cfbb3 bugfix: Make sure stored input shapes are json serializable 2019-07-24 18:20:06 +00:00
torzdf 5c17417d4d Training: Add Multi-Output Support
- Name output nodes of models
- Add support for multiple outputs in models and in training
- Update loss output format for cli and gui
- Sort graphs tabs on training
- Fix analysis graph for new loss names
- Multi output support in convert
2019-07-24 12:32:21 +00:00
torzdf 40103b9c4a Model Structure Changes
Standardize inputs for AutoEncoders
Name outputs from AutoEncoders
Name layers in nn_blocks.py
2019-07-18 17:04:47 +00:00
torzdf 5418fba726 Add Convolutional Aware Initialization (#795)
* Training: Add Convolutional Aware Initialization config option

* Centralize Conv2D layer for handling initializer

* Add 'is-output' to NNMeta to indicate that network is an output to the Model
2019-07-16 10:09:29 +01:00
torzdf 03d419c505 Training improvements
Add Optimizer Savings option to offload some optimizer calculations to the CPU
Add output_shape to NNMeta class for referencing when building models
2019-07-10 22:35:41 +00:00
torzdf 79ece7af2b bugfix: Train Posix Path to String 2019-07-08 22:17:21 +00:00
torzdf a757a4aa88 GUI: Preview for convert
Display preview when outputting to a stream container (e.g. video, gif)
2019-07-08 11:29:17 +00:00
torzdf ce45f01e4a Error Handling
Preview Tool: Catch no faces found Error
Training: Raise unhandled ValueErrors
2019-07-08 11:28:44 +00:00
torzdf 541498970e Training Fixups
Only show model folder deprecation warning for existing models
Gracefully exit FixedProducerDispatcher on error
2019-07-07 12:23:01 +00:00
Artem Ivanov 568bd7048a RealFace Plugin: Encoder complexity fix (#782)
- fixed a bug that prevented altering default value of `complexity_encoder`
2019-07-07 12:19:08 +01:00
torzdf 957fb137ed Filter small faces bugfix 2019-07-06 21:54:37 +00:00
torzdf 2f0fe704af Fixups
GUI: Fix thumbnail calculation + add borders
cv2-dnn aligner: Remove debug code
Convert: FFMpeg plugin - retry muxing audio on failure and properly error handle
2019-07-06 16:27:11 +00:00
torzdf 8e9c901aeb Bugfix: Extract - Remove BoundingBox object and replace with dict 2019-07-06 09:44:01 +00:00
torzdf ae74e80483 Deprecation Warnings, Error Handling and Bugfix
Deprecation Warning: Rotation in Extract
Deprecation Warning: Multiple models within a single folder
Error Handling: Useful message for training size assertion error
Error Handling: Useful message for fewer images than batch size
Bugfix: BoundingBox object sometimes not available inside spawned process
2019-07-05 10:28:56 +00:00
torzdf 6a8161983e Move CV2-DNN Detector from pool_process to spawn_process
cv2-dnn handles multiprocessing internally so no speedup to be had
Hopefully resolves issue with cuDNN error with on-the-fly conversion
2019-07-03 17:56:37 +00:00
torzdf e1fca14716 Add Restore Model Tool
Tool: Add tool to restore models from backup
Snapshot: Create snapshot based on total iterations rather than session iterations
Models: Move backup/snapshot functions to lib/model
Training: Output average loss since last save at each save iteration
GUI: display_page.py: minor logging update
2019-07-01 23:28:31 +00:00
torzdf be98d0147f Fix typo 2019-06-30 23:49:56 +01:00
torzdf d151722487 Handle corrupted model error 2019-06-30 16:04:16 +00:00
torzdf 864d58f0be Useful logging message for OOM 2019-06-30 11:41:42 +00:00
torzdf 541dce1771 Config Update 2019-06-26 10:54:21 +00:00
torzdf 0e76422805 Expose Augmentation Options to config 2019-06-26 10:48:38 +00:00
torzdf 0db645ae41 Explicit string in snapshot_models 2019-06-26 01:31:48 +01:00
torzdf 677903b13b Change ffmpeg macro_block_size to 8 2019-06-26 00:06:36 +00:00
torzdf 88fbd433a8 Fixups
- Training: Replace -ac (augment-color) option with -nac (no-augment-color) option (color augmentation on by default)
    - Analysis: Correctly display currently loaded session
    - Analysis: Cursor events in graph popup
2019-06-25 23:46:39 +00:00
torzdf e6653dd027 Disable parallel processing when non-amd user uses a Keras extractor + non Keras extractor 2019-06-25 09:58:02 +00:00
torzdf b11ef60a2b Update parallel process checking for plaidML 2019-06-23 18:04:39 +01:00
torzdf 240751aff0 PlaidML Updates
- Add fan-amd details to Cli
- Update Extract Parallel Processing rules for plaidML
- Handle missing plaidML lib on amd flag
- Pass through correct loglevel from GPUStats to plaidml_tools
2019-06-23 17:57:09 +01:00
kilroythethird 67d4ff9df0 Added keras version of FAN for plaidml backends (#756)
* Added keras version of FAN for plaidml backends

* Added supports_plaidml flag to fan_amd
2019-06-22 18:58:36 +01:00
torzdf 11f35fa798 Fix Active Devices for plaidML. Add Supports PlaidML to Extractors 2019-06-20 18:25:28 +01:00
torzdf 5917cc40b1 Allow customization of number of preview images to display 2019-06-16 13:55:27 +01:00
torzdf d5b64ef839 Remove batchsize limit on preview images 2019-06-16 00:21:22 +00:00
torzdf 3c80794ff8 GUI: Show more previews for extract and convert 2019-06-14 12:25:29 +00:00
torzdf 6ac422ac27 Extraction Improvements
Default to parallel processing
Add Image normalization options for aligners
2019-06-13 15:22:56 +00:00
torzdf 2324ceecd6 tf.__version__ bugfix 2019-06-12 22:05:01 +01:00
torzdf b03f44827c Full RTX Support
Remove FFMPEG from Conda Requirements
Fix Tensorboard logging for Tensorflow 1.13.1
Bump default Conda Tensorflow to 1.13.1
Update max python version to 3.7
2019-06-12 19:20:53 +01:00
torzdf f117846c22 Extended Mask. Extends eyebrow points up the forehead 2019-06-12 11:05:48 +00:00
torzdf 71bcd863e1 Bugfixes and minor changes
Fix: Occasional memory leak in convert
Fix: Dynamic config loading in Windows
Change: Better folder naming for snapshots
2019-06-11 17:26:43 +00:00
torzdf 0496bb7806 Bugfix: Stop defaults from entering available plugins list 2019-06-11 11:09:26 +00:00
torzdf 0eff0a1719 Config Changes + Bugfixes
Bugfix: Fully disable keypress monitor for GUI
Bugfix: Preview - Handle missing alignments file
Config changes:
    - Separate plugin defaults into their own files
    - Move mask_type to global training config
    - Add ability to pass in custom config files
2019-06-11 01:13:42 +00:00
torzdf d447c7d535 Bugfixes
- Set previews to minimum of batchsize and number of images
- Add manual balance help text
- Remove illegal character from convert _config.py
2019-06-09 11:13:42 +00:00
torzdf 98db63c585 Bugfixes
- Switch to correct Tools tab when loading recent file.
- Fix on-the-fly conversion cuDNN error
2019-06-08 15:39:14 +00:00
torzdf 0da3b5ef82 Color Augmentation method (#752)
* Color Augmentation: Implement testing code

* Tweak Contrast and Lighting augmentation amounts

* Prevent color augmentation from entering timelapse

* Remove all augmentations from preview images
2019-06-08 11:52:17 +01:00
torzdf a5d1d60992 Minor changes
Convert - Force prediction batchsize to 1 when converting on CPU
Training: Move snapshot saving to outside of the save threads to avoid race condition
GUI Summary stats. Show iteration count for current training session (fixed to last save amount).
2019-06-04 12:43:49 +01:00
torzdf 2071c6df45 Add save snapshot backup option for training 2019-06-03 01:43:35 +00:00
torzdf 26836a4907 Bugfixes
Facefilter - Variable reference before assignment
masks - Facehull initialize to 255.0 instead of 1.0
Alignments tool - Stop tool from ignoring upper case extensions
pep8 fixes
2019-05-31 17:48:53 +01:00
torzdf 6b7efa436f Convert: Add manual balance color adjustment plugin 2019-05-31 11:32:09 +01:00
torzdf 236c35f11a Convert settings tool (#737)
* Initial prediction pipeline and display

* Patch faces from convert pipeline. Load configs

* Add tkinter canvas for image display

* Radio buttons to config gui. Standardise config gui. Remove dssim loss from realface

* Add action frame

* Live update cli options

* Live update config changes

* Quicker frame loading. Refresh faces button callback

* remove debug code

* Add saving and reloading config buttons

* Reduce lag with threading

* Busy indicator + geometry updates

* Fix options canvas resizing

* Add global config saving and resetting

* tools to own tab

* Rename tool to 'preview'

* geometry tweak

* pep8 fixes
2019-05-30 18:15:25 +01:00
torzdf 40ed94f200 Bugfixes
sysinfo,py: CUDNN Check fix for unsupported OSes
cli.py: More useful information
scripts/convert.py: Set processes to 1 if no images are passed in
cv2_dnn aligner: Fix bounding box for negative values
scripts/convert.py: Fix for incorrect frame ranges passed in
2019-05-20 12:57:52 +01:00
torzdf 7b12021dc1 Fixups
vgg_face: to own module
vgg_face: correct output and normalization
Extract: more aggressive queue sizes
sys_info: Fix cudnn check error on Linux
2019-05-14 16:06:15 +00:00
torzdf 12d83c6e68 Remove dlib (#725)
* Sort to vgg_face. Remove Face Recognition requirements

* Remove dlib components from setup.py

* Remove dlib from Windows Installer script

* Remove dlib and cmake from requirements.txt

* Update INSTALL.md to remove all dlib requirements

* Remove dlib and cmake from dockerfiles

* Windows installer. Output cli to details Window. Setup.py bugfix

* Skip FaceFilter if invalid paths passed in
2019-05-12 23:24:25 +01:00