mirror of
https://github.com/iperov/DeepFaceLab.git
synced 2025-07-05 20:42:11 -07:00
01d81674fd
here new whole_face + XSeg workflow: with XSeg model you can train your own mask segmentator for dst(and/or src) faces that will be used by the merger for whole_face. Instead of using a pretrained segmentator model (which does not exist), you control which part of faces should be masked. new scripts: 5.XSeg) data_dst edit masks.bat 5.XSeg) data_src edit masks.bat 5.XSeg) train.bat Usage: unpack dst faceset if packed run 5.XSeg) data_dst edit masks.bat Read tooltips on the buttons (en/ru/zn languages are supported) mask the face using include or exclude polygon mode. repeat for 50/100 faces, !!! you don't need to mask every frame of dst only frames where the face is different significantly, for example: closed eyes changed head direction changed light the more various faces you mask, the more quality you will get Start masking from the upper left area and follow the clockwise direction. Keep the same logic of masking for all frames, for example: the same approximated jaw line of the side faces, where the jaw is not visible the same hair line Mask the obstructions using exclude polygon mode. run XSeg) train.bat train the model Check the faces of 'XSeg dst faces' preview. if some faces have wrong or glitchy mask, then repeat steps: run edit find these glitchy faces and mask them train further or restart training from scratch Restart training of XSeg model is only possible by deleting all 'model\XSeg_*' files. If you want to get the mask of the predicted face (XSeg-prd mode) in merger, you should repeat the same steps for src faceset. New mask modes available in merger for whole_face: XSeg-prd - XSeg mask of predicted face -> faces from src faceset should be labeled XSeg-dst - XSeg mask of dst face -> faces from dst faceset should be labeled XSeg-prd*XSeg-dst - the smallest area of both if workspace\model folder contains trained XSeg model, then merger will use it, otherwise you will get transparent mask by using XSeg-* modes. Some screenshots: XSegEditor: https://i.imgur.com/7Bk4RRV.jpg trainer : https://i.imgur.com/NM1Kn3s.jpg merger : https://i.imgur.com/glUzFQ8.jpg example of the fake using 13 segmented dst faces : https://i.imgur.com/wmvyizU.gifv
83 lines
2.7 KiB
Python
83 lines
2.7 KiB
Python
from PyQt5.QtCore import *
|
|
from PyQt5.QtGui import *
|
|
from PyQt5.QtWidgets import *
|
|
|
|
from localization import StringsDB
|
|
from .QXMainWindow import *
|
|
|
|
class QXIconButton(QPushButton):
|
|
"""
|
|
Custom Icon button that works through keyEvent system, without shortcut of QAction
|
|
works only with QXMainWindow as global window class
|
|
currently works only with one-key shortcut
|
|
"""
|
|
|
|
def __init__(self, icon,
|
|
tooltip=None,
|
|
shortcut=None,
|
|
click_func=None,
|
|
first_repeat_delay=300,
|
|
repeat_delay=20,
|
|
):
|
|
|
|
super().__init__(icon, "")
|
|
|
|
self.setIcon(icon)
|
|
|
|
if shortcut is not None:
|
|
tooltip = f"{tooltip} ( {StringsDB['S_HOT_KEY'] }: {shortcut} )"
|
|
|
|
self.setToolTip(tooltip)
|
|
|
|
|
|
self.seq = QKeySequence(shortcut) if shortcut is not None else None
|
|
|
|
QXMainWindow.inst.add_keyPressEvent_listener ( self.on_keyPressEvent )
|
|
QXMainWindow.inst.add_keyReleaseEvent_listener ( self.on_keyReleaseEvent )
|
|
|
|
self.click_func = click_func
|
|
self.first_repeat_delay = first_repeat_delay
|
|
self.repeat_delay = repeat_delay
|
|
self.repeat_timer = None
|
|
|
|
self.op_device = None
|
|
|
|
self.pressed.connect( lambda : self.action(is_pressed=True) )
|
|
self.released.connect( lambda : self.action(is_pressed=False) )
|
|
|
|
def action(self, is_pressed=None, op_device=None):
|
|
if self.click_func is None:
|
|
return
|
|
|
|
if is_pressed is not None:
|
|
if is_pressed:
|
|
if self.repeat_timer is None:
|
|
self.click_func()
|
|
self.repeat_timer = QTimer()
|
|
self.repeat_timer.timeout.connect(self.action)
|
|
self.repeat_timer.start(self.first_repeat_delay)
|
|
else:
|
|
if self.repeat_timer is not None:
|
|
self.repeat_timer.stop()
|
|
self.repeat_timer = None
|
|
else:
|
|
self.click_func()
|
|
if self.repeat_timer is not None:
|
|
self.repeat_timer.setInterval(self.repeat_delay)
|
|
|
|
def on_keyPressEvent(self, ev):
|
|
key = ev.key()
|
|
if ev.isAutoRepeat():
|
|
return
|
|
|
|
if self.seq is not None:
|
|
if key == self.seq[0]:
|
|
self.action(is_pressed=True)
|
|
|
|
def on_keyReleaseEvent(self, ev):
|
|
key = ev.key()
|
|
if ev.isAutoRepeat():
|
|
return
|
|
if self.seq is not None:
|
|
if key == self.seq[0]:
|
|
self.action(is_pressed=False)
|