mirror of
https://github.com/iperov/DeepFaceLab.git
synced 2025-07-07 05:22:06 -07:00
54 lines
2.2 KiB
Python
54 lines
2.2 KiB
Python
import json
|
|
import os
|
|
# Sample dictionary with default answers
|
|
dictionary = {
|
|
'Extractor_continue_deleting_files' : '\n',
|
|
'choose_model_name' : '0',
|
|
'choose_mode' : '1',
|
|
'choose_mask_mode' : '4',
|
|
'enhance_details_sharpen_filters' : '0',
|
|
'Output image format':'png',
|
|
'NonInteractiveMode':'y',
|
|
'Which GPU indexes to choose?': '0',
|
|
'Face type': 'wf',
|
|
'Max number of faces from image' : '1',
|
|
'Image size' : '512',
|
|
'Jpeg quality' : '90',
|
|
'Write debug images to aligned_debug?': 'n',
|
|
'Autobackup every N hour':'2',
|
|
'Write preview history' : 'n',
|
|
'Flip SRC faces randomly':'n',
|
|
'Flip DST faces randomly':'n',
|
|
'Batch_size': '4',
|
|
'Eyes and mouth priority':'n',
|
|
'Uniform yaw distribution of samples':'y',
|
|
'Blur out mask':'n',
|
|
'Place models and optimizer on GPU' : 'y',
|
|
'Use AdaBelief optimizer?' : 'y',
|
|
'Use learning rate dropout' : 'n',
|
|
'Enable random warp of samples' : 'y',
|
|
'Random hue/saturation/light intensity' : '0.0',
|
|
'GAN power' : '0.0',
|
|
'Face style power' : '0.0',
|
|
'Background style power': '0.0',
|
|
'Color transfer for src faceset' : 'lct',
|
|
'Enable gradient clipping': 'n',
|
|
'Enable pretraining mode' : 'n',
|
|
'Use interactive merger?':'n',
|
|
'Number of workers?':'8',
|
|
'Use saved session?':'n',
|
|
'Bitrate of output file in MB/s' : '16',
|
|
'Choose erode mask modifier': '80',
|
|
'Choose blur mask modifier' : '100',
|
|
'Choose motion blur power' : '0',
|
|
'Choose output face scale modifier' : '0',
|
|
'Choose super resolution power' : '0',
|
|
'Choose image degrade by denoise power' : '0',
|
|
'Choose image degrade by bicubic rescale power' : '0',
|
|
'Degrade color power of final image' : '0',
|
|
'Color transfer to predicted face' : 'rct',
|
|
'Press enter in 2 seconds to override model settings.' : 'y',
|
|
}
|
|
os.makedirs('DeepFaceLab_Linux/workspace/interact', exist_ok = True)
|
|
with open('DeepFaceLab_Linux/workspace/interact/interact_dict.json', 'w') as handle:
|
|
json.dump(dictionary, handle)
|