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

fghfdg

大佬能把训练时模型概要中文显示也搬运到原版上吗？

nde2024

感谢楼主，非常好用，赞。

pasanonic

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

Model.zip (21.1 KB, 下载次数: 6)

2024-9-16 23:55 上传

点击文件名下载附件

改好了神龙ME   论坛 [    i    ]  不显示  所以掉了
                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}')

pasanonic

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

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

论坛打 [    i    ]  不显示  所以掉了
                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

pasanonic 发表于 2024-9-17 00:20
论坛打 [    i    ]  不显示  所以掉了
                S这里= label_face_filename(S这里, filenames[0 ...

没注意，哈哈~
原版基本上比较完美了。。。

毕加索

pasanonic 发表于 2024-9-16 23:55
改好了神龙ME   论坛 [    i    ]  不显示  所以掉了
                S这里= label_face_filename(S这里 ...

感谢，亲试成功了！！

我还有个问题求教
怎么改字体，神农最新版的太糊了

pasanonic

毕加索 发表于 2024-9-17 20:57
感谢，亲试成功了！！

我还有个问题求教

楼主那有教程啊  
utils\label_face.py

这是我的
import cv2
from PIL import Image, ImageDraw, ImageFont
import numpy as np

def label_face_filename(face, filename):
    org = (5, face.shape[0] - 22)
    thickness = 1
    fontScale = 1
    color = (255, 165, 0)
    face = face.copy() # numpy array issue
    #cv2.putText(face, filename, org, font, fontScale, color, thickness, cv2.LINE_AA)
    face_uint8 = (face*255).astype('uint8')
    imgPIL = Image.fromarray(cv2.cvtColor(face_uint8, cv2.COLOR_BGR2RGB))
    drawPIL = ImageDraw.Draw(imgPIL)
    textSize = 23
    fontText = ImageFont.truetype("font/simsun.ttc", textSize, encoding="utf-8")
    text = filename
    drawPIL.text(org, text, color, font=fontText)
    imgPutText = cv2.cvtColor(np.asarray(imgPIL), cv2.COLOR_RGB2BGR)
    img_float32 = imgPutText.astype(np.float32)/ 255.0
    return img_float32

毕加索

pasanonic 发表于 2024-9-17 21:08
楼主那有教程啊  
utils\label_face.py

感谢，很好用！

wtxx8888

毕加索 发表于 2024-9-17 21:41
感谢，很好用！

祝贺

nicky9999

毕加索 发表于 2024-9-17 21:41
感谢，很好用！

    drawPIL.text(org, text, color, font=fontText, stroke_width=2, stroke_fill=(0, 0, 0))
这一行後面可以加外框，应该可以看的清楚一点