Hugging Face's logo

Spaces:
kornia
/
edge_detector
Running on CPU Upgrade

App Files Community
2
edgarriba commited on
Commit
909dc14
•
1 Parent(s): 773b526

Image is rgb already

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -8,8 +8,7 @@ def edge_detection(filepath, detector):
8
  img: Tensor = K.io.load_image(filepath, K.io.ImageLoadType.RGB32)
9
  img = img[None]
10
 
11
- x_rgb: Tensor = K.color.bgr_to_rgb(img)
12
- x_gray = K.color.rgb_to_grayscale(x_rgb)
13
 
14
 
15
  if detector == '1st order derivates in x':
 
8
  img: Tensor = K.io.load_image(filepath, K.io.ImageLoadType.RGB32)
9
  img = img[None]
10
 
11
+ x_gray = K.color.rgb_to_grayscale(img)
 
12
 
13
 
14
  if detector == '1st order derivates in x':