在colab上，如何將Xseg訓練結果寫入aligned

chiefinkame

星级打分

• 1
• 2
• 3
• 4
• 5

平均分:NAN  参与人数:0  我的评分:未评

如題，
在colab的[color=var(--colab-anchor-color)]托尼是塔克腳本上，如何將Xseg訓練結果寫入aligned？

87951763

不懂看看

dsyrock

!python main.py xseg apply --input-dir ../workspace/data_dst/xseg --model-dir ../workspace/model
input-dir 和 model-dir 的路径改成你的路径

ammo

虛心學習

AKERSHUS

我一般把dst的faceset文件下载到本地，解包去看错误识别图片和画遮罩。画完再打包上传百度云，用multcloud同步到谷歌网盘。谷歌网盘操作延迟太高。

weblink

我自己在用的CODE 你參考一下。我也是用東尼的腳本自己改的。

下面請用CODE方式貼上。



#@title 對面部應用或移除 XSeg 蒙版
Mode = "Apply mask" #@param ["Apply mask", "Remove mask"]
Data = "data_src_3d" #@param {type:"string"}

from pathlib import Path
mode_arg = 'apply' if Mode == "Apply mask" else 'remove'

#移到您的目錄下
% cd $deepfacelab_cloab

data_path = f'../workspace/{Data}/aligned'
model_path = '../workspace/model'

cmd = f'main.py xseg {mode_arg} --input-dir {data_path} '
cmd += f'--model-dir {model_path}' if mode_arg == 'apply' else ''

!python $cmd

chiefinkame

weblink 发表于 2022-3-2 13:05
我自己在用的CODE 你參考一下。我也是用東尼的腳本自己改的。

下面請用CODE方式貼上。

大大，我貼上後，就出現這樣的錯誤，
可以告訴我該修正哪些地方嗎？

我用的model是圖左側那些xseg
然後想要寫進data_src中的 aligned

weblink

chiefinkame 发表于 2022-3-2 23:46
大大，我貼上後，就出現這樣的錯誤，
可以告訴我該修正哪些地方嗎？

你吧 data_src_3d 改成 data_src

chenchangqing

兄弟我能问问如何在colab里训练xseg吗