zhiweili commited on
Commit
cef300b
·
1 Parent(s): 1d8e854

test convert to gray

Browse files
app_haircolor_inpaint_adapter_15.py CHANGED
@@ -92,8 +92,8 @@ def image_to_image(
92
  run_task_time = 0
93
  time_cost_str = ''
94
  run_task_time, time_cost_str = get_time_cost(run_task_time, time_cost_str)
95
- # canny_image = canny_detector(input_image, int(generate_size*1), generate_size)
96
- canny_image = custom_canny_detector(input_image)
97
  # lineart_image = lineart_detector(input_image, int(generate_size*1), generate_size)
98
  # run_task_time, time_cost_str = get_time_cost(run_task_time, time_cost_str)
99
  pidiNet_image = pidiNet_detector(input_image, int(generate_size*1), generate_size)
 
92
  run_task_time = 0
93
  time_cost_str = ''
94
  run_task_time, time_cost_str = get_time_cost(run_task_time, time_cost_str)
95
+ canny_image = canny_detector(input_image, int(generate_size*1), generate_size)
96
+ canny_image = canny_image.convert("L")
97
  # lineart_image = lineart_detector(input_image, int(generate_size*1), generate_size)
98
  # run_task_time, time_cost_str = get_time_cost(run_task_time, time_cost_str)
99
  pidiNet_image = pidiNet_detector(input_image, int(generate_size*1), generate_size)