update app.py file
Browse files
app.py
CHANGED
@@ -24,7 +24,7 @@ def segment_image(img):
|
|
24 |
outputs = model(**img)
|
25 |
|
26 |
predicted_semantic_map = processor.post_process_semantic_segmentation(
|
27 |
-
outputs, target_sizes=[
|
28 |
)[0]
|
29 |
|
30 |
fig, ax = plt.subplots(figsize=(5, 5))
|
|
|
24 |
outputs = model(**img)
|
25 |
|
26 |
predicted_semantic_map = processor.post_process_semantic_segmentation(
|
27 |
+
outputs, target_sizes=[img.size[::-1]]
|
28 |
)[0]
|
29 |
|
30 |
fig, ax = plt.subplots(figsize=(5, 5))
|