update

README.md

@@ -56,7 +56,7 @@ cd CelloType
 Then Download the model weights:
 
 ```bash
-cd data
+cd models
 sh download.sh
 cd ..
 ```

docs/source/index.rst

@@ -15,6 +15,7 @@ CelloType is an end-to-end Transformer-based method for automated cell/nucleus s
    :caption: Examples:
 
    notebooks/cell_segmentation
+   inference_large
    notebooks/tissue_annotation
 
 .. toctree::

docs/source/quickstart.rst

@@ -1,22 +1,31 @@
 Quickstart
 ------------------------------
 
-Clone the repository:
+Clone the repository and download the pre-trained models:
+~~~~~~~~~~~~~~~~~~~~~~~~~
 
 .. code-block:: bash
 
     git clone https://github.com/maxpmx/CelloType.git
     cd CelloType
 
-Then Download the model weights:
-
 .. code-block:: bash
 
-    cd data
+    cd models
     sh download.sh
     cd ..
 
-Predict on an example image:
+Prepare the input images
+~~~~~~~~~~~~~~~~~~~~~~~~~
+
+Convert the input images into an RGB format where the blue channel represents the nuclear channel, the green channel corresponds to the membrane channel.
+
+.. image:: ../../data/example/example_tissuenet.png
+    :width: 250px
+    :alt: drawing
+
+Inference the cell segmentation
+~~~~~~~~~~~~~~~~~~~~~~~~~
 
 .. code-block:: python
 
@@ -31,4 +40,4 @@ Predict on an example image:
       device='cuda', 
       config_path='./configs/maskdino_R50_bs16_50ep_4s_dowsample1_2048.yaml')
 
-    mask = model.predict(img) # [H, W]
+    mask = model.predict(img) # [H, W]