zhiweili commited on
Commit
991954d
1 Parent(s): 0a87234

change linet detect

Browse files
Files changed (1) hide show
  1. app_haircolor_inpaint_15.py +2 -2
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, 768, 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, 768, generate_size)
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)