xsl/change_hair
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

asdf

Browse Source
This commit is contained in:
xsl
2026-07-17 22:13:56 +08:00
parent 2606aca7c5
commit b2738fba7d
28 changed files with 126 additions and 126 deletions
Download Patch File Download Diff File Expand all files Collapse all files
gen_hairline_mask.py
+3 -3
View File
@@ -8,7 +8,7 @@ import sys
import cv2
import numpy as np
HAIR_SERVICE_DIR = "/home/xsl/change_hair/project/hair_service_sd"
HAIR_SERVICE_DIR = "/home/ubuntu/change_hair/project/hair_service_sd"
sys.path.insert(0, HAIR_SERVICE_DIR)
os.environ.setdefault("HF_HUB_OFFLINE", "1")
@@ -87,7 +87,7 @@ def gen_hairline_mask(img_path, mask_path):
if __name__ == "__main__":
img_path = sys.argv[1] if len(sys.argv) > 1 else \
"/home/xsl/change_hair/project/data/userImage/8488902485_20250630055547.jpg"
"/home/ubuntu/change_hair/project/data/userImage/8488902485_20250630055547.jpg"
mask_path = sys.argv[2] if len(sys.argv) > 2 else \
"/home/xsl/change_hair/project/logs/hairgrow_test_mask.png"
"/home/ubuntu/change_hair/project/logs/hairgrow_test_mask.png"
gen_hairline_mask(img_path, mask_path)