hysts HF staff commited on
Commit
ef50d9e
1 Parent(s): 27be314
Files changed (1) hide show
  1. model.py +10 -10
model.py CHANGED
@@ -136,8 +136,8 @@ class Model:
136
  )
137
  return [vis_control_image] + results.images
138
 
 
139
  def preprocess_canny(
140
- self,
141
  input_image: np.ndarray,
142
  image_resolution: int,
143
  low_threshold: int,
@@ -184,8 +184,8 @@ class Model:
184
  seed=seed,
185
  )
186
 
 
187
  def preprocess_hough(
188
- self,
189
  input_image: np.ndarray,
190
  image_resolution: int,
191
  detect_resolution: int,
@@ -244,8 +244,8 @@ class Model:
244
  seed=seed,
245
  )
246
 
 
247
  def preprocess_hed(
248
- self,
249
  input_image: np.ndarray,
250
  image_resolution: int,
251
  detect_resolution: int,
@@ -292,8 +292,8 @@ class Model:
292
  seed=seed,
293
  )
294
 
 
295
  def preprocess_scribble(
296
- self,
297
  input_image: np.ndarray,
298
  image_resolution: int,
299
  ) -> tuple[PIL.Image.Image, PIL.Image.Image]:
@@ -334,8 +334,8 @@ class Model:
334
  seed=seed,
335
  )
336
 
 
337
  def preprocess_scribble_interactive(
338
- self,
339
  input_image: np.ndarray,
340
  image_resolution: int,
341
  ) -> tuple[PIL.Image.Image, PIL.Image.Image]:
@@ -377,8 +377,8 @@ class Model:
377
  seed=seed,
378
  )
379
 
 
380
  def preprocess_fake_scribble(
381
- self,
382
  input_image: np.ndarray,
383
  image_resolution: int,
384
  detect_resolution: int,
@@ -433,8 +433,8 @@ class Model:
433
  seed=seed,
434
  )
435
 
 
436
  def preprocess_pose(
437
- self,
438
  input_image: np.ndarray,
439
  image_resolution: int,
440
  detect_resolution: int,
@@ -484,8 +484,8 @@ class Model:
484
  seed=seed,
485
  )
486
 
 
487
  def preprocess_seg(
488
- self,
489
  input_image: np.ndarray,
490
  image_resolution: int,
491
  detect_resolution: int,
@@ -532,8 +532,8 @@ class Model:
532
  seed=seed,
533
  )
534
 
 
535
  def preprocess_depth(
536
- self,
537
  input_image: np.ndarray,
538
  image_resolution: int,
539
  detect_resolution: int,
@@ -577,8 +577,8 @@ class Model:
577
  seed=seed,
578
  )
579
 
 
580
  def preprocess_normal(
581
- self,
582
  input_image: np.ndarray,
583
  image_resolution: int,
584
  detect_resolution: int,
 
136
  )
137
  return [vis_control_image] + results.images
138
 
139
+ @staticmethod
140
  def preprocess_canny(
 
141
  input_image: np.ndarray,
142
  image_resolution: int,
143
  low_threshold: int,
 
184
  seed=seed,
185
  )
186
 
187
+ @staticmethod
188
  def preprocess_hough(
 
189
  input_image: np.ndarray,
190
  image_resolution: int,
191
  detect_resolution: int,
 
244
  seed=seed,
245
  )
246
 
247
+ @staticmethod
248
  def preprocess_hed(
 
249
  input_image: np.ndarray,
250
  image_resolution: int,
251
  detect_resolution: int,
 
292
  seed=seed,
293
  )
294
 
295
+ @staticmethod
296
  def preprocess_scribble(
 
297
  input_image: np.ndarray,
298
  image_resolution: int,
299
  ) -> tuple[PIL.Image.Image, PIL.Image.Image]:
 
334
  seed=seed,
335
  )
336
 
337
+ @staticmethod
338
  def preprocess_scribble_interactive(
 
339
  input_image: np.ndarray,
340
  image_resolution: int,
341
  ) -> tuple[PIL.Image.Image, PIL.Image.Image]:
 
377
  seed=seed,
378
  )
379
 
380
+ @staticmethod
381
  def preprocess_fake_scribble(
 
382
  input_image: np.ndarray,
383
  image_resolution: int,
384
  detect_resolution: int,
 
433
  seed=seed,
434
  )
435
 
436
+ @staticmethod
437
  def preprocess_pose(
 
438
  input_image: np.ndarray,
439
  image_resolution: int,
440
  detect_resolution: int,
 
484
  seed=seed,
485
  )
486
 
487
+ @staticmethod
488
  def preprocess_seg(
 
489
  input_image: np.ndarray,
490
  image_resolution: int,
491
  detect_resolution: int,
 
532
  seed=seed,
533
  )
534
 
535
+ @staticmethod
536
  def preprocess_depth(
 
537
  input_image: np.ndarray,
538
  image_resolution: int,
539
  detect_resolution: int,
 
577
  seed=seed,
578
  )
579
 
580
+ @staticmethod
581
  def preprocess_normal(
 
582
  input_image: np.ndarray,
583
  image_resolution: int,
584
  detect_resolution: int,