Anirudh Balaraman commited on
Commit
cc831d6
·
unverified ·
1 Parent(s): 62bf25f

Fix HTML entities and enhance pipeline documentation

Browse files

Updated the pipeline documentation to improve clarity and fix HTML entities.

Files changed (1) hide show
  1. docs/pipeline.md +5 -3
docs/pipeline.md CHANGED
@@ -4,18 +4,18 @@ The full pipeline has three phases: preprocessing, PI-RADS training (Stage 1), a
4
 
5
  ```mermaid
6
  flowchart TD
7
- subgraph <b>Preprocessing</b>
8
  R[register_and_crop] --> S[get_segmentation_mask]
9
  S --> H[histogram_match]
10
  H --> G[get_heatmap]
11
  end
12
 
13
  subgraph Stage 1
14
- P[<b>PI-RADS Training</b><br/>CrossEntropy + Attention Loss]
15
  end
16
 
17
  subgraph Stage 2
18
- C[<b>csPCa Training</b><br/>Frozen Backbone + BCE Loss]
19
  end
20
 
21
  G --> P
@@ -96,3 +96,5 @@ python run_cspca.py --mode train --config config/config_cspca_train.yaml
96
  | Metrics | AUC, Sensitivity, Specificity |
97
 
98
  The backbone's feature extractor (`net`), transformer, and `myfc` are frozen. The attention module and `SimpleNN` classification head are trained. After training the framework reports mean and 95% confidence intervals for AUC, sensitivity, and specificity by testing across 20 random seeds.
 
 
 
4
 
5
  ```mermaid
6
  flowchart TD
7
+ subgraph Preprocessing
8
  R[register_and_crop] --> S[get_segmentation_mask]
9
  S --> H[histogram_match]
10
  H --> G[get_heatmap]
11
  end
12
 
13
  subgraph Stage 1
14
+ P[PI-RADS Training<br/>CrossEntropy + Attention Loss]
15
  end
16
 
17
  subgraph Stage 2
18
+ C[csPCa Training<br/>Frozen Backbone + BCE Loss]
19
  end
20
 
21
  G --> P
 
96
  | Metrics | AUC, Sensitivity, Specificity |
97
 
98
  The backbone's feature extractor (`net`), transformer, and `myfc` are frozen. The attention module and `SimpleNN` classification head are trained. After training the framework reports mean and 95% confidence intervals for AUC, sensitivity, and specificity by testing across 20 random seeds.
99
+
100
+ Refer to [Getting Started](getting-started.md) for JSON dataset format to run run_pirads.py and run_cspca.py