cwmok / C2FViT

This is the official Pytorch implementation of "Affine Medical Image Registration with Coarse-to-Fine Vision Transformer" (CVPR 2022), written by Tony C. W. Mok and Albert C. S. Chung.
MIT License
131 stars 3 forks source link

code for your cvpr2024paper #21

Open kkkkkk123-ops opened 1 week ago

kkkkkk123-ops commented 1 week ago

Dear author: Recently I read your new paper on CVPR2024 'Modality-Agnostic Structural Image Representation Learning for Deformable Multi-Modality Medical Image Registration'. It's really a great work and very helpful to me. May I ask when the code of this paper is available? Really looking forward to it! Thanks!

cwmok commented 1 week ago

@kkkkkk123-ops

Thanks for your interest in my latest work. The project is a company project and is undergoing patent registration. Once the patent registration is completed, I will publish the code.

I use the following code to implement the method. For your reference, Contrastive loss: https://github.com/taesungp/contrastive-unpaired-translation/blob/master/models/patchnce.py#L6--L55 Patch sampler: https://github.com/taesungp/contrastive-unpaired-translation/blob/master/models/networks.py#L565-L570 Self-similarity (MIND): https://github.com/multimodallearning/convexAdam/blob/de33303de0ef3c55b6e5f642ed28210c50ca40a9/l2r_2020_convexAdam_CuRIOUS.py#L29-L73

kkkkkk123-ops commented 1 week ago

thank you for your response, it helps me a lot! Best!