cwmok / Conditional_LapIRN

Conditional Deformable Image Registration with Convolutional Neural Network
MIT License
79 stars 16 forks source link

Dice loss #7

Closed tracob-github closed 1 year ago

tracob-github commented 1 year ago

Dear author, Thank you for this excellent work. I have seen that you mentioned dice loss in the Presentation, but why is it not reflected in this repo? Looking forward to your reply.

cwmok commented 1 year ago

Hi @tracob-github,

This repository is for the implementation of our paper in MICCAI:

In the original paper, the training is fully unsupervised; therefore, the dice loss is not included in this repository.

tracob-github commented 1 year ago

Thank you for your reply. I found that dice loss was added to Task1 and Task3 in your presentation. Did you add dice loss in the final implementation? 121ea7e2af4be1477b07262356c544e

960336c8fa70ba1bc82ff011390c59e

Looking forward to your reply.

cwmok commented 1 year ago

Hi @tracob-github,

We did use dice loss during the learn2reg challenge in order to maximize the performance. For more details of our approach in the challenge, you may refer to our challenge paper at https://link.springer.com/chapter/10.1007/978-3-030-97281-3_23.

tracob-github commented 1 year ago

Thank you for your patient reply. And is it convenient for you to provide the source code with dice loss? I want to learn about weakly-supervised registration. Looking forward to your reply.

cwmok commented 1 year ago

Hi @tracob-github,

Example of the dice loss https://github.com/voxelmorph/voxelmorph/blob/793b90fbddfd93f31019ecbb8c036454c6ee0720/voxelmorph/torch/losses.py#L79

tracob-github commented 1 year ago

Thank you very much. It helped me a lot.