zhiweili commited on
Commit
049f1dd
1 Parent(s): 7dc06b4

adjust detect size

Browse files
Files changed (1) hide show
  1. app_haircolor_inpaint_15.py +2 -2
app_haircolor_inpaint_15.py CHANGED
@@ -85,9 +85,9 @@ def image_to_image(
85
  time_cost_str = ''
86
  run_task_time, time_cost_str = get_time_cost(run_task_time, time_cost_str)
87
  # canny_image = canny_detector(input_image, int(generate_size*1), generate_size)
88
- lineart_image = lineart_detector(input_image, int(generate_size*1), generate_size)
89
  run_task_time, time_cost_str = get_time_cost(run_task_time, time_cost_str)
90
- pidiNet_image = pidiNet_detector(input_image, int(generate_size*1), generate_size)
91
  control_image = [lineart_image, pidiNet_image]
92
 
93
  generator = torch.Generator(device=DEVICE).manual_seed(seed)
 
85
  time_cost_str = ''
86
  run_task_time, time_cost_str = get_time_cost(run_task_time, time_cost_str)
87
  # canny_image = canny_detector(input_image, int(generate_size*1), generate_size)
88
+ lineart_image = lineart_detector(input_image, 384, generate_size)
89
  run_task_time, time_cost_str = get_time_cost(run_task_time, time_cost_str)
90
+ pidiNet_image = pidiNet_detector(input_image, 512, generate_size)
91
  control_image = [lineart_image, pidiNet_image]
92
 
93
  generator = torch.Generator(device=DEVICE).manual_seed(seed)