Update README.md

15 cell types (16 output classes including background) edited

README.md

@@ -14,10 +14,10 @@ tags:
 
 # HNE2Cell — H&E Whole-Slide Image Cell Detection & Classification
 
-HNE2Cell detects and classifies **16 cell types** from H&E-stained whole-slide images (WSI).
+HNE2Cell detects and classifies **15 cell types (16 output classes including background)** from H&E-stained whole-slide images (WSI).
 It takes 256×256 px patches as input and outputs per-cell contours, centroids, and type labels.
 
-## Cell Types (16 classes)
+## Output Classes (background + 15 cell types)
 
 | ID | Cell Type | Color |
 |----|-----------|-------|
@@ -261,7 +261,7 @@ python patchify.py \
 python inference.py \
     --input_dir /path/to/patch_folders \
     --output_dir /path/to/results \
-    --model_path ./HNE2cell_all_patch73_jit.pt \
+    --model_path ./HNE2cell_pub_patch73_jit.pt \
     --magnification 40 \
     --batch_size 32
 ```