Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -344,25 +344,6 @@ def draw_mask(v, d, evt: gr.EventData):
|
|
344 |
|
345 |
d["layers"][0] = cv2.fillPoly(d["layers"][0], [pts], (0,0,0,255))
|
346 |
d["composite"] = cv2.fillPoly(d["composite"], [pts], (0,0,0,255))
|
347 |
-
|
348 |
-
|
349 |
-
scale = 1
|
350 |
-
delta = 0
|
351 |
-
ddepth = cv2.CV_16S
|
352 |
-
|
353 |
-
gray = cv2.cvtColor(d["composite"], cv2.COLOR_RGB2GRAY)
|
354 |
-
grad_x = cv2.Sobel(gray, ddepth, 2, 0, ksize=5, scale=scale, delta=delta, borderType=cv2.BORDER_DEFAULT)
|
355 |
-
# Gradient-Y
|
356 |
-
# grad_y = cv2.Scharr(gray,ddepth,0,1)
|
357 |
-
grad_y = cv2.Sobel(gray, ddepth, 0, 2, ksize=5, scale=scale, delta=delta, borderType=cv2.BORDER_DEFAULT)
|
358 |
-
|
359 |
-
abs_grad_x = cv2.convertScaleAbs(grad_x)
|
360 |
-
abs_grad_y = cv2.convertScaleAbs(grad_y)
|
361 |
-
|
362 |
-
grad = cv2.addWeighted(abs_grad_x, 0.5, abs_grad_y, 0.5, 0)
|
363 |
-
|
364 |
-
d["layers"][0] = cv2.cvtColor(grad, cv2.COLOR_GRAY2RGB)
|
365 |
-
d["composite"] = cv2.cvtColor(grad, cv2.COLOR_GRAY2RGB)
|
366 |
|
367 |
print(d["composite"])
|
368 |
|
|
|
344 |
|
345 |
d["layers"][0] = cv2.fillPoly(d["layers"][0], [pts], (0,0,0,255))
|
346 |
d["composite"] = cv2.fillPoly(d["composite"], [pts], (0,0,0,255))
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
347 |
|
348 |
print(d["composite"])
|
349 |
|