Spaces:
Sleeping
Sleeping
Anirudh Balaraman commited on
Fix formatting in project structure section of README
Browse files
README.md
CHANGED
|
@@ -108,26 +108,26 @@ See the [full documentation](https://anirudhbalaraman.github.io/WSAttention-Pros
|
|
| 108 |
|
| 109 |
```
|
| 110 |
WSAttention-Prostate/
|
| 111 |
-
βββ run_pirads.py
|
| 112 |
-
βββ run_cspca.py
|
| 113 |
-
βββ run_inference.py
|
| 114 |
-
βββ preprocess_main.py
|
| 115 |
-
βββ config/
|
| 116 |
βββ src/
|
| 117 |
β βββ model/
|
| 118 |
-
β β βββ MIL.py
|
| 119 |
-
β β βββ csPCa_model.py
|
| 120 |
β βββ data/
|
| 121 |
-
β β βββ data_loader.py
|
| 122 |
β β βββ custom_transforms.py # Custom MONAI transforms
|
| 123 |
β βββ train/
|
| 124 |
-
β β βββ train_pirads.py
|
| 125 |
-
β β βββ train_cspca.py
|
| 126 |
-
β βββ preprocessing/
|
| 127 |
-
β βββ utils.py
|
| 128 |
βββ tests/
|
| 129 |
-
βββ dataset/
|
| 130 |
-
βββ models/
|
| 131 |
```
|
| 132 |
|
| 133 |
## π Acknowledgement
|
|
|
|
| 108 |
|
| 109 |
```
|
| 110 |
WSAttention-Prostate/
|
| 111 |
+
βββ run_pirads.py # PI-RADS training/testing entry point
|
| 112 |
+
βββ run_cspca.py # csPCa training/testing entry point
|
| 113 |
+
βββ run_inference.py # Full inference pipeline
|
| 114 |
+
βββ preprocess_main.py # Preprocessing entry point
|
| 115 |
+
βββ config/ # YAML configuration files
|
| 116 |
βββ src/
|
| 117 |
β βββ model/
|
| 118 |
+
β β βββ MIL.py # MILModel_3D β core MIL architecture, PI-RADS model
|
| 119 |
+
β β βββ csPCa_model.py # csPCa_Model
|
| 120 |
β βββ data/
|
| 121 |
+
β β βββ data_loader.py # MONAI data pipeline
|
| 122 |
β β βββ custom_transforms.py # Custom MONAI transforms
|
| 123 |
β βββ train/
|
| 124 |
+
β β βββ train_pirads.py # PI-RADS training loop
|
| 125 |
+
β β βββ train_cspca.py # csPCa training loop
|
| 126 |
+
β βββ preprocessing/ # Registration, segmentation, histogram matching, heatmaps
|
| 127 |
+
β βββ utils.py # Shared utilities
|
| 128 |
βββ tests/
|
| 129 |
+
βββ dataset/ # Reference images for histogram matching
|
| 130 |
+
βββ models/ # Downloaded checkpoints (not in repo)
|
| 131 |
```
|
| 132 |
|
| 133 |
## π Acknowledgement
|