zhiweili commited on
Commit
ad2fa8d
·
1 Parent(s): 9e6a15d

change adapter

Browse files
app_haircolor_inpaint_adapter_15.py CHANGED
@@ -91,16 +91,17 @@ def image_to_image(
91
  ):
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)
100
  # cond_image = [canny_image, pidiNet_image]
101
  # cond_scale = [cond_scale1, cond_scale2]
102
 
103
- cond_image = canny_image
104
  cond_scale = cond_scale1
105
 
106
  generator = torch.Generator(device=DEVICE).manual_seed(seed)
 
91
  ):
92
  run_task_time = 0
93
  time_cost_str = ''
94
+ detect_image = input_image.convert('L')
95
  run_task_time, time_cost_str = get_time_cost(run_task_time, time_cost_str)
96
  # canny_image = canny_detector(input_image, int(generate_size*1), generate_size)
97
+ # canny_image = custom_canny_detector(input_image)
98
  # lineart_image = lineart_detector(input_image, int(generate_size*1), generate_size)
99
  # run_task_time, time_cost_str = get_time_cost(run_task_time, time_cost_str)
100
+ pidiNet_image = pidiNet_detector(detect_image, int(generate_size*1), generate_size)
101
  # cond_image = [canny_image, pidiNet_image]
102
  # cond_scale = [cond_scale1, cond_scale2]
103
 
104
+ cond_image = pidiNet_image
105
  cond_scale = cond_scale1
106
 
107
  generator = torch.Generator(device=DEVICE).manual_seed(seed)