Commit graph

55 commits

Author SHA1 Message Date
iperov
8294feca24 extractior: fix extract from already extracted dflimg 2019-05-11 17:36:22 +04:00
iperov
e58197ca22 initial code to extract umdfaces.io dataset and train pose estimator 2019-04-23 08:14:09 +04:00
iperov
046649e6be
update == 04.20.2019 == (#242)
* superb improved fanseg

* _

* _

* added FANseg extractor for src and dst faces to use it in training

* -

* _

* _

* update to 'partial' func

* _

* trained FANSeg_256_full_face.h5,
new experimental models: AVATAR, RecycleGAN

* _

* _

* _

* fix for TCC mode cards(tesla), was conflict with plaidML initialization.

* _

* update manuals

* _
2019-04-20 08:23:08 +04:00
iperov
9f68b25987 Extractor: added warning messages if aligned folder contains files that will be deleted. 2019-04-06 10:01:59 +04:00
iperov
de8d75b4f7 fix mask editor,
upd manuals
2019-04-04 22:19:56 +04:00
iperov
b03b147bae refactoring 2019-03-26 11:09:44 +04:00
iperov
7a0cc56603 extractor: fixes, optimizations,
manual extractor: added 'a' option to switch accuracy mode
2019-03-21 18:56:32 +04:00
iperov
d1676dea06 refactorings, optimizations 2019-03-21 11:58:54 +04:00
iperov
a3df04999c removing trailing spaces 2019-03-19 23:53:27 +04:00
iperov
1f569117c8 accurate landmark extractor now by default 2019-03-19 14:01:15 +04:00
iperov
7c1ea1a617 removing dflimg image_to_face_mat 2019-03-18 14:42:10 +04:00
iperov
00c6a0cdaf accurate landmarks extraction now as an option 2019-03-18 11:24:41 +04:00
iperov
bf831331e6 fix 2019-03-18 10:58:03 +04:00
iperov
b8efb4cbba enhancing landmarks extractor by using s3fd second pass inside second pass,
it will be x2 slower, but time will be saved due to more images will be marked properly
works on 2GB+
2019-03-18 10:25:24 +04:00
iperov
1d56585f33 fix ability for auto re-extract dst faces, after this update DFL torrent extractor scripts will not work. 2019-03-18 01:12:04 +04:00
iperov
b03d62a29b extractor: added ability to re-extract dst faces with transferring all necessary meta data 2019-03-18 00:43:07 +04:00
iperov
4ff88865ce refactoring 2019-03-16 21:29:56 +04:00
iperov
f3b343c0e5 small fixed and refactorings 2019-03-16 20:55:51 +04:00
iperov
9823421a44 added transparent mask to draw_landmarks 2019-03-14 12:16:21 +04:00
iperov
62af1b6f16 S3FD : filtering faces < 40pix. 2019-03-14 01:23:39 +04:00
iperov
893354b870 fix S3FD for plaidML, fix S3FD warnings on Lambda layers. 2019-03-11 09:32:00 +04:00
iperov
fbf39d2727 added new extractor: S3FD,
all extractors now produce less false-positive faces
2019-03-10 23:18:10 +04:00
iperov
168d48b119 extractor: increased speed of 1st pass by DLIB for >= 4GB vram cards,
added 'h' key for hide the help in manual extractor
2019-03-08 17:01:11 +04:00
iperov
46f4d6c426 fix cpu fallback for extractor 2019-02-28 14:54:59 +04:00
iperov
438213e97c manual extractor: increased FPS,
sort by final : now you can specify target number of images,
converter: fix seamless mask and exception,
huge refactoring
2019-02-28 11:56:31 +04:00
iperov
daebe10ca7 Extractor: fix aspect ratio for square images,
now we can re-extract face manually from already aligned jpg image without quality loss.
2019-02-22 17:28:08 +04:00
iperov
72ba6b103c added support of AMD videocards
added Intel's plaidML backend to use OpenCL engine. Check new requirements.
smart choosing of backend in device.py
env var 'force_plaidML' can be choosed to forced using plaidML
all tf functions transferred to pure keras
MTCNN transferred to pure keras, but it works slow on plaidML (forced to CPU in this case)
default batch size for all models and VRAMs now 4, feel free to adjust it on your own
SAE: default style options now ZERO, because there are no best values for all scenes, set them on your own.
SAE: return back option pixel_loss, feel free to enable it on your own.
SAE: added option multiscale_decoder default is true, but you can disable it to get 100% same as H,DF,LIAEF model behaviour.
fix converter output to .png
added linux fork reference to doc/doc_build_and_repository_info.md
2019-02-19 17:33:12 +04:00
iperov
5a727f4e50 fix manual extractor bugs 2019-02-13 11:02:12 +04:00
iperov
06fe1314d8 removing default yaw_value from DFLIMG files,
added better pitch/yaw estimator from 68 landmarks,
improving face yaw accuracy for sorting and trainers,
added sort by face-pitch
2019-02-12 21:31:37 +04:00
iperov
e5995630fc fix 2019-02-10 23:00:08 +04:00
iperov
0866b2c509 manual extractor: added zoom for small faces 2019-02-10 22:40:57 +04:00
iperov
bda3fb3da1 refactoring 2019-02-10 21:40:58 +04:00
iperov
5a1fb199f4 added support of non-english characters in paths 2019-02-05 19:33:55 +04:00
iperov
0fb3dd53ce fix extractor jpg quality 2019-02-05 14:08:56 +04:00
iperov
6d95dd4a99 DFL now works with JPG files by default. Old PNG files also supported. Added util convertor PNG to JPG. 2019-02-04 19:47:24 +04:00
iperov
7ed7f2bab9 fixed bug in manual extractor on windows 10 2019-01-26 06:20:47 +04:00
iperov
72646becd1 forcing all tqdm's to ascii in order to work properly on Chinese windows 2019-01-24 21:32:07 +04:00
iperov
318e6608f7 Extractor: increased speed of deleted debug files searcher 2019-01-23 09:48:45 +04:00
iperov
21460c2e13 upd manual extractor help message 2019-01-22 18:15:55 +04:00
iperov
2571f23bc4 Extractor added new feature --manual-output-debug-fix,
Performs manual reextract input-dir frames which were deleted from [output_dir]_debug\ dir.
2019-01-22 12:46:54 +04:00
iperov
3bbc008207 change jpeg quality to 50% 2019-01-22 10:05:20 +04:00
iperov
8016ee32cc change aligned_debug files from png to jpg 2019-01-22 09:55:41 +04:00
iperov
1f2b1481ef now you can train models on multiple GPU's on same workspace without cloning any folders.
Model files names will be prefixed with GPU index if GPU choosed explicitly on train/convert start.
if you leave GPU idx choice default, then best GPU idx will be choosed and model file names will not contain index prefix.
It gives you possibility to train same fake with various models or options on multiple GPUs.

H64 and H128: now you can choose 'Lighter autoencoder'. It is same as vram gb <= 4 before this update.

added archived_models.zip contains old experiments

RecycleGAN: archived

devicelib: if your system has no NVML installed (some old cards), then it will work with gpu_idx=0 as 'Generic GeForce GPU' with 2GB vram.

refactorings
2019-01-14 10:48:23 +04:00
iperov
12383570e8 Fix for systems without NVSMI 2019-01-13 16:38:17 +04:00
iperov
4625bcec1c fix multigpu extractor 2019-01-13 13:59:22 +04:00
iperov
c3f175862a fix manual extractor bug 2019-01-10 05:12:02 +04:00
iperov
b5ba7d52cb increased speed of sort by hist sim for ten thousands of faces 2019-01-04 22:54:09 +04:00
iperov
f8824f9601 refactoring. Added RecycleGAN for testing. 2018-12-28 19:38:52 +04:00
iperov
44798c2b85 refactoring 2018-12-22 19:37:25 +04:00
iperov
0d7387165a added --cpu-only mode for all stages. Upd readme. Added requirements-cpu.txt 2018-12-21 16:54:22 +04:00