|
楼主 |
发表于 2024-4-29 17:36:38
|
显示全部楼层
本帖最后由 aini 于 2024-4-29 19:46 编辑
继续更,可以直接在.sh文件那加入命令来自动注释或者取消注释,需要改三个sh文件。
5_data_dst_extract_faces_S3FD.sh, 5_data_src_extract_faces_S3FD.sh, 7_merge_SAEHD.sh
更改 5_data_dst_extract_faces_S3FD.sh, 5_data_src_extract_faces_S3FD.sh(注意两个文件的保存路径不一样,不要一起改了)***********************
#!/usr/bin/env bash
2 source env.sh
3
4 #!/bin/bash
5
6 # 要操作的 Python 文件路径
7 python_file="/data/adengine/jupyter/wangjie_work/DeepFaceLab/DeepFaceLab_Linux/DeepFaceLab/core/joblib/SubprocessorBase.py"
8
9 # 要注释或取消注释的行号
10 line_number=247
11
12 is_commented=$(sed -n "${line_number}p" "$python_file" | grep -qE '^\s*#' && echo "true" || echo "false")
13
14 # 如果已经被注释,不进行任何操作;如果没有被注释,注释掉该行
15 if [ "$is_commented" = "false" ]; then
16 sed -i "${line_number}s/^/#/" "$python_file"
17 fi
18
19
20 $DFL_PYTHON "$DFL_SRC/main.py" extract \
21 --input-dir "$DFL_WORKSPACE/data_dst" \
22 --output-dir "$DFL_WORKSPACE/data_dst/aligned" \
23 --detector s3fd \
24 --max-faces-from-image 0 \
25 --output-debug
26
~
~
更改 7_merge_SAEHD.sh *********************************
1 #!/usr/bin/env bash
2 source env.sh
3
4 python_file="/data/adengine/jupyter/wangjie_work/DeepFaceLab/DeepFaceLab_Linux/DeepFaceLab/core/joblib/SubprocessorBase.py"
5 line_number=247
6
7 # 检查指定行是否已经被注释
8 is_commented=$(sed -n "${line_number}p" "$python_file" | grep -qE '^\s*#' && echo "true" || echo "false")
9
10 # 如果已经被注释,取消注释;如果没有被注释,不进行任何操作
11 if [ "$is_commented" = "true" ]; then
12 sed -i "${line_number}s/^#//" "$python_file"
13 fi
14
15
16 $DFL_PYTHON "$DFL_SRC/main.py" merge \
17 --input-dir "$DFL_WORKSPACE/data_dst" \
18 --output-dir "$DFL_WORKSPACE/data_dst/merged" \
19 --output-mask-dir "$DFL_WORKSPACE/data_dst/merged_mask" \
20 --aligned-dir "$DFL_WORKSPACE/data_dst/aligned" \
21 --model-dir "$DFL_WORKSPACE/model" \
22 --model SAEHD
23
~
|
|