Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -55,7 +55,7 @@ def process_image(image, mask, strategy, sampler, fx=1, fy=1):
|
|
55 |
def create_mask(image, prediction):
|
56 |
mask = Image.new("RGBA", image.size, (0, 0, 0, 255)) # Black background
|
57 |
draw = ImageDraw.Draw(mask)
|
58 |
-
scale = 1
|
59 |
for polygons in prediction['polygons']:
|
60 |
for _polygon in polygons:
|
61 |
_polygon = np.array(_polygon).reshape(-1, 2)
|
|
|
55 |
def create_mask(image, prediction):
|
56 |
mask = Image.new("RGBA", image.size, (0, 0, 0, 255)) # Black background
|
57 |
draw = ImageDraw.Draw(mask)
|
58 |
+
scale = 1
|
59 |
for polygons in prediction['polygons']:
|
60 |
for _polygon in polygons:
|
61 |
_polygon = np.array(_polygon).reshape(-1, 2)
|