Update README.md

README.md

@@ -47,14 +47,18 @@ Here is how to use this model:
 
 ```python
 
+
 from skimage import io, segmentation, morphology, measure, exposure
 from sribd_cellseg_models import MultiStreamCellSegModel,ModelConfig
 import numpy as np
 import tifffile as tif
 import requests
 import torch
+from PIL import Image
+from overlay import visualize_instances_map
+import cv2
+img_name = 'test_images/cell_00551.tiff'
 
-img_name = 'cell_00023.tiff'
 def normalize_channel(img, lower=1, upper=99):
     non_zero_vals = img[np.nonzero(img)]
     percentiles = np.percentile(non_zero_vals, [lower, upper])
@@ -86,5 +90,8 @@ my_model.__init__(ModelConfig())
 my_model.load_checkpoints(checkpoints)
 with torch.no_grad():
     output = my_model(pre_img_data)
+overlay = visualize_instances_map(pre_img_data,star_label)
+cv2.imwrite('prediction.png', cv2.cvtColor(overlay, cv2.COLOR_RGB2BGR))
+
 
 ```