Update app.py

app.py

@@ -342,7 +342,7 @@ def plt_to_image():
     img = Image.open(buf)
     return img
 
-def calculate_slice_statistics(one_mask, slice_size=128):
+def calculate_slice_statistics(one_mask, slice_size=256):
     """计算每个切片的材质占比"""
     num_rows, num_cols = one_mask.shape[0] // slice_size, one_mask.shape[1] // slice_size
     slice_stats = {}
@@ -386,7 +386,7 @@ def find_top_slices(slice_stats, exclusion_list, min_percent=0.7, min_slices=1,
 
     return valid_top_slices
 
-def extract_and_visualize_top_slices(image, top_slices, slice_size=128):    
+def extract_and_visualize_top_slices(image, top_slices, slice_size=256):    
     fig, axs = plt.subplots(nrows=len(top_slices), ncols=3, figsize=(15, 5 * len(top_slices)))
     image=Image.fromarray(image)
     if len(top_slices) == 1:
@@ -436,7 +436,7 @@ def process_image(image_path):
     # 重新命名 DataFrame 为 percentage_df 以清楚表达其内容
     percentage_df = counts_df.rename(columns={'计数': 'pixels', '百分比': 'percentage (%)'})
 
-    slice_size = 64
+    slice_size = 256
     exclusion_list = [38]
     slice_stats = calculate_slice_statistics(one_mask, slice_size=slice_size)
     top_slices = find_top_slices(slice_stats, exclusion_list=exclusion_list, min_percent=0.5, min_slices=1)