Repository navigation
In this release we provide:
- A pre-trained model for prostate MRI scan classification .keras format, (model.zip)
- The same model in onnx format, and (model.onnx)
- A scaling factorcs csv needed for preprocessing the metadata (scaling_factors_df.csv)
The model performs classification of:
- 0: "T1-weighted axial",
- 1: "T2-weighted axial",
- 2: "diffusion-weighted images (DWI)",
- 3: "apparent diffusion coefficient (ADC)}
The CNN is trained using a combination of images and DICOM metadata from parts of the following collections in Imaging Data Commons (IDC):
- QIN-Prostate-Repeatability
- ProstateX
- Prostate3T
- ProstateDiagnosis
- ProstateMRI
- Prostate-fused-MRI-Pathology
- Prostate-MRI-US-Biopsy