Spaces:
Running
on
Zero
Running
on
Zero
zhiweili
commited on
Commit
•
991954d
1
Parent(s):
0a87234
change linet detect
Browse files
app_haircolor_inpaint_15.py
CHANGED
@@ -86,9 +86,9 @@ def image_to_image(
|
|
86 |
time_cost_str = ''
|
87 |
run_task_time, time_cost_str = get_time_cost(run_task_time, time_cost_str)
|
88 |
# canny_image = canny_detector(input_image, int(generate_size*1), generate_size)
|
89 |
-
lineart_image = lineart_detector(input_image,
|
90 |
run_task_time, time_cost_str = get_time_cost(run_task_time, time_cost_str)
|
91 |
-
pidiNet_image = pidiNet_detector(input_image,
|
92 |
control_image = [lineart_image, pidiNet_image]
|
93 |
|
94 |
generator = torch.Generator(device=DEVICE).manual_seed(seed)
|
|
|
86 |
time_cost_str = ''
|
87 |
run_task_time, time_cost_str = get_time_cost(run_task_time, time_cost_str)
|
88 |
# canny_image = canny_detector(input_image, int(generate_size*1), generate_size)
|
89 |
+
lineart_image = lineart_detector(input_image, 384, generate_size)
|
90 |
run_task_time, time_cost_str = get_time_cost(run_task_time, time_cost_str)
|
91 |
+
pidiNet_image = pidiNet_detector(input_image, 512, generate_size)
|
92 |
control_image = [lineart_image, pidiNet_image]
|
93 |
|
94 |
generator = torch.Generator(device=DEVICE).manual_seed(seed)
|