Spaces:
Running
Running
File size: 1,014 Bytes
cd89698 bae28ee cd89698 46506c1 cd89698 69bd6d1 cd89698 46506c1 69bd6d1 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 |
# Getting Started
## Installation
For installation and downloading the saved weights of the models, please check the [github](https://github.com/anirudhbalaraman/WSAttention-Prostate) repository.
## Data Format
Input bpMRI scans should have all three sequnces, T2W, ADC and DWI as separate files in **NRRD** format. The prostate mask and heatmap are generated by the sample for each sample. To train the models or run testing with reference labels, the data pipeline uses MONAI's decathlon-format JSON:
```json
{
"train": [
{
"image": "path/to/t2.nrrd",
"dwi": "path/to/dwi.nrrd",
"adc": "path/to/adc.nrrd",
"mask": "path/to/prostate_mask.nrrd",
"heatmap": "path/to/heatmap.nrrd",
"label": 0
}
],
"test": [...]
}
```
Sample JSON files are available in the github repository.
Paths are relative to `data_root`. PI-RADS labels are 0-indexed (`0` = PI-RADS 2, `3` = PI-RADS 5). csPCa labels are binary (0 or 1).
|