【实用工具】预览界面显示样本loss值

wtxx8888

奇奇怪怪的ID 发表于 2024-9-13 22:10
现在可以再试一下了，刚更新了，我在老乡的机子上试了一遍，没问题了，但是有个选项至少要打开一次，就是 ...

额，用的我前面的代码？那个漏了全局的默认配置语句。
新的，是不需要进入参数的，没有此项的丹，会自动加默认。

haohmaru

学习中。。。

fghfdg

感谢分享

wzx945

看起来不错

wtxx8888

用我的吧，整合了你显示LOSS的代码。
文件名显示，支持中文。
嘴眼优先拆分为，MVE分支的眼部优先，嘴部优先。更精确！
DFL原版比较完美了，需求大的功能，基本都有了。
https://dfldata.cc/forum.php?mod=viewthread&tid=9843

奇奇怪怪的ID

wtxx8888 发表于 2024-9-14 00:56
用我的吧，整合了你显示LOSS的代码。
文件名显示，支持中文。
嘴眼优先拆分为，MVE分支的眼部优先，嘴部优 ...

OK，我改下这个贴的内容，废弃掉

wtxx8888

毕加索 发表于 2024-9-13 16:34
可惜了，神农用不了

神农，需要自行添加代码。

编辑  _internal\DeepFaceLab\models\Model_SAEHD\Model.py   文件。

搜索紫色的 onGetPreview，添加蓝色的代码，即可显示LOSS，就三句。


    def onGetPreview(self, samples, for_history=False, filenames=None):
        ( (warped_src, target_src, target_srcm, target_srcm_em),
          (warped_dst, target_dst, target_dstm, target_dstm_em) ) = samples

        src_loss, dst_loss = self.src_dst_train (warped_src, target_src, target_srcm, target_srcm_em, warped_dst, target_dst, target_dstm, target_dstm_em)

        S, D, SS, DD, DDM, SD, SDM = [ np.clip( nn.to_data_format(x,"NHWC", self.model_data_format), 0.0, 1.0) for x in ([target_src,target_dst] + self.AE_view (target_src, target_dst) ) ]
        DDM, SDM, = [ np.repeat (x, (3,), -1) for x in [DDM, SDM] ]

        target_srcm, target_dstm = [ nn.to_data_format(x,"NHWC", self.model_data_format) for x in ([target_srcm, target_dstm] )]

        n_samples = min(4, self.get_batch_size(), 800 // self.resolution )

        if filenames is not None and len(filenames) > 0:
            for i in range(n_samples):
                S= label_face_filename(S, filenames[0])
                SS = label_face_filename(SS, f'{src_loss:.4f}')
                D = label_face_filename(D, filenames[1])
                DD = label_face_filename(DD, f'{dst_loss:.4f}')

毕加索

wtxx8888 发表于 2024-9-14 02:18
神农，需要自行添加代码。

编辑  _internal\DeepFaceLab\models\Model_SAEHD\Model.py   文件。

谢谢大哥！
但是我用的是ME模型不是SAEHD
然后我在E:\DFL-ShenNong-V3.0.6\_internal\DeepFaceLab\models\Model_ME\Model.py 按照您给的方法改不行
未改之前：
def onGetPreview(self, samples, for_history=False, filenames=None):
        (
            (warped_src, target_src, target_srcm, target_srcm_em),
            (warped_dst, target_dst, target_dstm, target_dstm_em),
        ) = samples

        S, D, SS, SSM, DD, DDM, SD, SDM = [
            np.clip(nn.to_data_format(x, "NHWC", self.model_data_format), 0.0, 1.0)
            for x in ([target_src, target_dst] + self.AE_view(target_src, target_dst))
        ]
        SW, DW = [
            np.clip(nn.to_data_format(x, "NHWC", self.model_data_format), 0.0, 1.0)
            for x in ([warped_src, warped_dst])
        ]
        (
            SSM,
            DDM,
            SDM,
        ) = [np.repeat(x, (3,), -1) for x in [SSM, DDM, SDM]]

        target_srcm, target_dstm = [
            nn.to_data_format(x, "NHWC", self.model_data_format)
            for x in ([target_srcm, target_dstm])
        ]

        n_samples = min(self.get_batch_size(), self.options["preview_samples"])

        if filenames is not None and len(filenames) > 0:
            for i in range(n_samples):
                S = label_face_filename(S, filenames[0])
                D = label_face_filename(D, filenames[1])

wtxx8888

毕加索 发表于 2024-9-14 12:00
谢谢大哥！
但是我用的是ME模型不是SAEHD
然后我在E:\DFL-ShenNong-V3.0.6\_internal\DeepFaceLab\models ...

该不会是A卡DX通道的吧？有人反映DX通道的，会报错。
N卡的CUDA，一切正常。。。

你是不显示，还是报错呀？？
ME的参数多一个，尝试改SS，DD。变为其他的试试吧（S, D, SS, SSM, DD, DDM, SD, SDM中，除了S ,D）。

wzx945

大佬，你还是别把文件下架吧，wtxx8888大佬的版本A卡用不了，但是你之前发布的那个版本倒是可以用，可能是大佬整合MVE眼嘴优先优化的那部分A卡用不了