guatavita / Prediction_Model_Class

My base class for letting the server predict from multiple models

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Prediction_Model_Class

Class by https://github.com/brianmanderson to run multiple models with Tensorflow (>2) on a workstation or DGX server.

Models available (PRIVATE)

Model Name Author Version Description
Create_Liver_BMA BMAnderson _BMA_Program_4 Liver
Create_Liver_Disease_Ablation_BMA BMAnderson _BMA_Program_0 Liver_Disease_Ablation
Create_Liver_Lobe_Segments_BMA BMAnderson _BMAProgram3 Liver_Segment_1, Liver_Segment_2, Liver_Segment_3, Liver_Segment_4, Liver_Segment_5-8
Create_Lung_BMA BMAnderson _BMA_Program_2 Ground Glass, Lung
Create_LACC_MorfeusLab / AIP_LACC_AI_contours BRigaud _MorfeusLab_v4 Bladder, Rectum, Sigmoid, Vagina, Parametrium, Femur_Head_R, Femur_Head_L, Kidney_R, Kidney_L, SpinalCord, BowelSpace, Femoral Heads, Upper_Vagina_2.0cm, CTVp
AIP_LACC_3D_AI_contours BRigaud _MorfeusLab_v5 Bladder, Rectum, Sigmoid, Vagina, Parametrium, Femur_Head_R, Femur_Head_L, Kidney_R, Kidney_L, SpinalCord, BowelSpace, Femoral Heads, Upper_Vagina_2.0cm, CTVp
AIP_LACC_AI_CTVN_contours BRigaud _MorfeusLab_v2 CTVn, CTV_PAN, Nodal_CTV
AIP_LACC_AI_Duodenum_contours BRigaud _MorfeusLab_v2 Duodenum
Create_Pancreas_DLv3_MorfeusLab_v0 BRigaud _MorfeusLab_v0 Pancreas
Create_Cyst_HybridDLv3_MorfeusLab_v0 BRigaud _HybridDLv3_v0 (need pancreas) Cyst
Create_Disease_Ablation_MorfeusLab_v0 BRigaud _MorfeusLab_v0 Disease, Ablation
Create_PSMA_MorfeusLab BRigaud _MorfeusLab_v3 (need femoral heads) Bladder, Rectum, Iliac Veins, Iliac Arteries
Create_PSMA_3D_MorfeusLab BRigaud _MorfeusLab_v4 (need femoral heads) Bladder, Rectum, Iliac Veins, Iliac Arteries
Create_FemHeads_MorfeusLab BRigaud _MorfeusLab_v0 Femoral Heads
Create_Liver_3D_MorfeusLab BRigaud _MorfeusLab_v0 Liver
Create_Lung_GTV_MorfeusLab_v0 BRigaud _MorfeusLab_v0 GTV (need Lungs contour or bounding box)

Dependencies

This repository is expecting a folder with model per localization. Models are not distributed in that repository. This repository is using a "networks" private repository.

docker pull tensorflow/tensorflow:2.4.1-gpu
pip install -r requirements.txt

How to reference this work

This repository was used in the following published manuscripts:

https://doi.org/10.1016/j.prro.2021.02.003 @article{anderson2021simple, title={Simple Python Module for Conversions Between DICOM Images and Radiation Therapy Structures, Masks, and Prediction Arrays}, author={Anderson, Brian M and Wahid, Kareem A and Brock, Kristy K}, journal={Practical radiation oncology}, volume={11}, number={3}, pages={226--229}, year={2021}, publisher={Elsevier} }

https://doi.org/10.1016/j.adro.2020.04.023 @article{anderson2021automated, title={Automated contouring of contrast and noncontrast computed tomography liver images with fully convolutional networks}, author={Anderson, Brian M and Lin, Ethan Y and Cardenas, Carlos E and Gress, Dustin A and Erwin, William D and Odisio, Bruno C and Koay, Eugene J and Brock, Kristy K}, journal={Advances in radiation oncology}, volume={6}, number={1}, pages={100464}, year={2021}, publisher={Elsevier} }

https://doi.org/10.1016/j.ijrobp.2020.10.038 @article{rigaud2021automatic, title={Automatic segmentation using deep learning to enable online dose optimization during adaptive radiation therapy of cervical cancer}, author={Rigaud, Bastien and Anderson, Brian M and Zhiqian, H Yu and Gobeli, Maxime and Cazoulat, Guillaume and S{"o}derberg, Jonas and Samuelsson, Elin and Lidberg, David and Ward, Christopher and Taku, Nicolette and others}, journal={International Journal of Radiation Oncology* Biology* Physics}, volume={109}, number={4}, pages={1096--1110}, year={2021}, publisher={Elsevier} }

About

My base class for letting the server predict from multiple models


Languages

Language:Python 100.0%