mirror of
https://github.com/iperov/DeepFaceLab.git
synced 2025-07-06 21:12:07 -07:00
OpenCL support: fix bug, adding new requirements files
This commit is contained in:
parent
72ba6b103c
commit
b939fe9d85
4 changed files with 18 additions and 6 deletions
|
@ -1,6 +1,10 @@
|
|||
### **Features**:
|
||||
|
||||
- Windows binary containing pre-compiled dependencies, including CUDA libraries.
|
||||
- works on AMD, NVIDIA, IntelHD graphics and all OpenCL1.2-compatible videocards with at least 512M video memory
|
||||
|
||||
- CPU-only mode [`--cpu-mode`]. 8th gen Intel core CPU able to train H64 model in 2 days.
|
||||
|
||||
- Windows build is standalone ready to work program and contains all dependencies (CUDA, OpenCL, ffmpeg, .bat script etc) to start working
|
||||
|
||||
- New models expanding upon the original faceswap model.
|
||||
|
||||
|
@ -8,10 +12,6 @@
|
|||
|
||||
- Face metadata embedded into extracted JPG files.
|
||||
|
||||
- CPU-only mode [`--cpu-mode`]. 8th gen Intel core CPU able to train H64 model in 2 days.
|
||||
|
||||
- Preview window
|
||||
|
||||
- Extractor and Converter run in parallel.
|
||||
|
||||
- Debug mode option for all stages: [`--debug`]
|
||||
|
@ -23,7 +23,6 @@
|
|||
|
||||
Predicts faces more uniformly than dlib, resulting in a less jittered aligned output. However, MTCNN extraction will produce more false positives.
|
||||
|
||||
|
||||
Comparison dlib (at left) vs mtcnn on hard case:
|
||||

|
||||
|
||||
|
|
|
@ -270,6 +270,8 @@ try:
|
|||
ctx = plaidml.Context()
|
||||
for d in plaidml.devices(ctx, return_all=True)[0]:
|
||||
details = json.loads(d.details)
|
||||
if details['type'] == 'CPU': #skipping opencl-CPU
|
||||
continue
|
||||
if 'nvidia' in details['vendor'].lower():
|
||||
has_nvidia_device = True
|
||||
plaidML_devices += [ {'id':d.id,
|
||||
|
|
11
requirements-opencl.txt
Normal file
11
requirements-opencl.txt
Normal file
|
@ -0,0 +1,11 @@
|
|||
numpy==1.16.1
|
||||
pathlib==1.0.1
|
||||
scandir==1.6
|
||||
h5py==2.7.1
|
||||
Keras==2.2.4
|
||||
opencv-python==4.0.0.21
|
||||
tensorflow==1.11.0
|
||||
plaidml-keras==0.5.0
|
||||
scikit-image
|
||||
tqdm
|
||||
git+https://www.github.com/keras-team/keras-contrib.git
|
Loading…
Add table
Add a link
Reference in a new issue