Update app.py

app.py

@@ -185,10 +185,11 @@ def predict_and_visualize(image):
     height,width,_=image_np.shape
     maxhl=max(height,width)
     image = tf.convert_to_tensor(image_np, dtype=tf.float32) 
-    if maxhl==height:
-        image=tf.image.resize(image,(1500,int(1500*width/height)))
-    if maxhl==width:
-        image=tf.image.resize(image,(int(1500*height/width),1500))
+    if maxhl>1500:
+        if maxhl==height:
+            image=tf.image.resize(image,(1500,int(1500*width/height)))
+        if maxhl==width:
+            image=tf.image.resize(image,(int(1500*height/width),1500))
     #image = tf.image.resize_with_pad(image, 1500, 1500)
     image = tf.cast(image, tf.float32) / 255.0
     image = tf.transpose(image, perm=[2, 0, 1])
@@ -219,12 +220,13 @@ def predict_and_visualize(image):
     color_mask = color_mask.astype(np.uint8)
     result_image = Image.fromarray(color_mask)
     draw = ImageDraw.Draw(result_image)
-    font = ImageFont.truetype("arial.ttf", int(height/40))  # 尝试加载Arial字体，大小为12
+    font = ImageFont.truetype("arial.ttf", int(height/30))  # 尝试加载Arial字体，大小为12
 
     for key, position in label_positions.items():
         if key in id2label:
             # 绘制文本，您可能需要调整文本位置和字体大小
-            draw.text((position[0], position[1]), str(key), font=font, fill='white')
+            material=id2material[key]
+            draw.text((position[0], position[1]), str(material), font=font, fill='white')
 
     return pred_mask,result_image,counts_dict
 
@@ -334,7 +336,7 @@ def plot_material_color_palette_grid(material_dict, materials_per_row=4):
 # 将matplotlib图转换为图像
 def plt_to_image():
     buf = io.BytesIO()
-    plt.savefig(buf, format='png')
+    plt.savefig(buf, format='png',dpi=300)
     plt.close()
     buf.seek(0)
     img = Image.open(buf)