1

app.py

@@ -43,8 +43,13 @@ def segment_image(image):
     # 색상 팔레트 적용
     result_color = colors[result]
 
+    # NumPy 배열을 PIL 이미지로 변환
     result_image = Image.fromarray(result_color.astype(np.uint8))
 
+    # 원본 이미지와 추론 결과 이미지의 크기 일치시키기
+    result_image = result_image.resize(image.size, Image.NEAREST)
+
+    # 원본 이미지와 추론 결과 이미지 결합
     combined_image = Image.blend(image.convert("RGBA"), result_image.convert("RGBA"), alpha=0.5)
 
     return combined_image