add image output2

app.py

@@ -8,14 +8,14 @@ from cell_segmentation.inference.inference_cellvit_experiment_monuseg import Inf
 
 
 ## local |  remote
-RUN_MODE = "remote"
+RUN_MODE = "local"
 if RUN_MODE != "local":
-    os.system("wget https://huggingface.co/xiazhi/LKCell-demo/resolve/main/model_best.pth")
+    os.system("wget https://huggingface.co/xiazhi/LKCell/resolve/main/model_best.pth")
     ## examples
-    os.system("wget https://huggingface.co/xiazhi/LKCell-demo/resolve/main/1.png")
-    os.system("wget https://huggingface.co/xiazhi/LKCell-demo/resolve/main/2.png")
-    os.system("wget https://huggingface.co/xiazhi/LKCell-demo/resolve/main/3.png")
-    os.system("wget https://huggingface.co/xiazhi/LKCell-demo/resolve/main/4.png")
+    os.system("wget https://huggingface.co/xiazhi/LKCell/resolve/main/1.png")
+    os.system("wget https://huggingface.co/xiazhi/LKCell/resolve/main/2.png")
+    os.system("wget https://huggingface.co/xiazhi/LKCell/resolve/main/3.png")
+    os.system("wget https://huggingface.co/xiazhi/LKCell/resolve/main/4.png")
 
 ## step 1: set up model
 
@@ -62,10 +62,12 @@ def click_process(image_input , type_dataset):
         resize_shape = (512,512)
         image_input = cv2.resize(image_input, resize_shape)
         monuseg_inf.run_single_image_inference(monuseg_inf.model, image_input)
-        
-    image_output = cv2.imread("pred_img.png")
-    image_output = cv2.cvtColor(image_output, cv2.COLOR_BGR2RGB)   
-    return image_output
+
+    image_output = cv2.imread("raw_pred.png")
+    image_output = cv2.cvtColor(image_output, cv2.COLOR_BGR2RGB)     
+    image_output2 = cv2.imread("pred_img.png")
+    image_output2 = cv2.cvtColor(image_output2, cv2.COLOR_BGR2RGB)   
+    return image_output,image_output2
 
 
 demo = gr.Blocks(title="LkCell")
@@ -81,13 +83,14 @@ with demo:
                 Type_dataset = gr.Radio(choices=["pannuke", "monuseg"], label=" input image's dataset type",value="pannuke")
                                             
         with gr.Column():
-            with gr.Row():
-                image_output = gr.Image(type="numpy", label="Output",height=480)
+            image_output = gr.Image(type="numpy", label="image prediction",height=480,width=480)
+            image_output2 = gr.Image(type="numpy", label="all predictions",height=480)
+            
     with gr.Row():
         Button_run = gr.Button("🚀 Submit (发送) ")
-        clear_button = gr.ClearButton(components=[Image_input,Type_dataset,image_output],value="🧹 Clear (清除)")
+        clear_button = gr.ClearButton(components=[Image_input,Type_dataset,image_output,image_output2],value="🧹 Clear (清除)")
 
-    Button_run.click(fn=click_process, inputs=[Image_input,  Type_dataset ], outputs=[image_output])
+    Button_run.click(fn=click_process, inputs=[Image_input,  Type_dataset ], outputs=[image_output,image_output2])
     
     ## guiline
     gr.Markdown(value="""    
@@ -103,7 +106,7 @@ with demo:
                 ['3.png', "monuseg"],
                 ['4.png', "monuseg"],
                 ], 
-                inputs=[Image_input, Type_dataset], outputs=[image_output], label="Examples")
+                inputs=[Image_input, Type_dataset], outputs=[image_output,image_output2], label="Examples")
     gr.HTML(value="""
                 <p style="text-align:center; color:orange"> <a href='https://github.com/ziwei-cui/LKCellv1' target='_blank'>Github Repo</a></p>
                     """)

cell_segmentation/inference/inference_cellvit_experiment_monuseg.py

@@ -874,7 +874,7 @@ class MoNuSegInference:
             for poly, c in zip(pred_contours_polygon, pred_contour_colors_polygon)
         ]
         placeholder[:  h, 3 * w : 4 * w, :3] = np.asarray(pred_cell_image) / 255
-
+        pred_cell_image.save("raw_pred.png")
         # plotting
         fig, axs = plt.subplots(figsize=(3, 2), dpi=1200)
         axs.imshow(placeholder)
@@ -901,7 +901,7 @@ class MoNuSegInference:
         for y_seg in grid_y:
             axs.axhline(y_seg, color="black")
 
-        fig.suptitle(f"Patch Predictions for input image", fontsize=6)
+        fig.suptitle(f"All Predictions for input image", fontsize=6)
         fig.tight_layout()
         fig.savefig("pred_img.png")
         plt.close()

cell_segmentation/inference/inference_cellvit_experiment_pannuke.py

@@ -1088,7 +1088,7 @@ class InferenceCellViT:
             for y_seg in grid_y:
                 axs.axhline(y_seg, color="black")
 
-            fig.suptitle(f"Predictions for input image")
+            fig.suptitle(f"All Predictions for input image")
             fig.tight_layout()
             fig.savefig("pred_img.png")
             plt.close()