mist-medical / MIST

MIST: A simple, scalable, and end-to-end framework for 3D medical imaging segmentation.
Apache License 2.0
29 stars 11 forks source link

Edits for dose prediction (regression): Ensure that segmentation mask still works. #29

Closed doucejosy closed 1 month ago

doucejosy commented 2 months ago

Edited analyze, preprocess, train, evaluate functions. More needed: train/val losses, dose metrics, full preprocessing of input data for dose prediction. I made sure that both segmentation mask and dose prediction still work with these changes.

The assumption is that the input data for now are already preprocessed. I can always integrate the preprocessing steps later as the main task is to see if we can train and get reasonable results with these changes.

I still have many questions regarding the code and we can schedule a meeting for Friday to discuss them.

doucejosy commented 2 months ago

New commit: I just edited the loss and loss_utils code to correctly compute training and loss function. I wasn't indexing the y_true and y_pred correctly.

aecelaya commented 1 month ago

Closing pull request since we moved this to another repo.