A multi-scale unsupervised learning for deformable image registration

Last update: Apr 13, 2022

Related tags

Deep Learning MSRegNet

Overview

A multi-scale unsupervised learning for deformable image registration

Shuwei Shao, Zhongcai Pei, Weihai Chen, Wentao Zhu, Xingming Wu and Baochang Zhang – IJCARS 2021

[Link to paper]

✏️ 📄 Citation

If you find our work useful or interesting, please cite our paper:

@article{shao2021multi,
  title={A multi-scale unsupervised learning for deformable image registration},
  author={Shao, Shuwei and Pei, Zhongcai and Chen, Weihai and Zhu, Wentao and Wu, Xingming and Zhang, Baochang},
  journal={International Journal of Computer Assisted Radiology and Surgery},
  pages={1--10},
  year={2021},
  publisher={Springer}
}

📈 Results

To train a model, run:

CUDA_VISIBLE_DEVICES=<your_desired_GPU> \
python train_s2s_2d.py \
     <your_data_path> \
    --model vm2  \
    --batch_size 2  \
    --lambda 0.1 \
    --data_loss mse \
    --epochs 1500 \
    --steps_per_epoch 100 \
    --model_dir <your_dir_to_save_model>

Acknowledgement

Our code is based on the implementation of VoxelMorph.

Owner

ShuweiShao

GitHub Repository

An Unpaired Sketch-to-Photo Translation Model

Unpaired-Sketch-to-Photo-Translation We have released our code at https://github.com/rt219/Unsupervised-Sketch-to-Photo-Synthesis This project is the

38 Oct 28, 2022

3D HourGlass Networks for Human Pose Estimation Through Videos

3D-HourGlass-Network 3D CNN Based Hourglass Network for Human Pose Estimation (3D Human Pose) from videos. This was my summer'18 research project. Dis

51 Jan 02, 2023

Official implementation of NeurIPS 2021 paper "Contextual Similarity Aggregation with Self-attention for Visual Re-ranking"

CSA: Contextual Similarity Aggregation with Self-attention for Visual Re-ranking PyTorch training code for CSA (Contextual Similarity Aggregation). We

19 Oct 21, 2022

All of the figures and notebooks for my deep learning book, for free!

"Deep Learning - A Visual Approach" by Andrew Glassner This is the official repo for my book from No Starch Press. Ordering the book My book is called

227 Jan 04, 2023

PyTorch implementation of the YOLO (You Only Look Once) v2

PyTorch implementation of the YOLO (You Only Look Once) v2 The YOLOv2 is one of the most popular one-stage object detector. This project adopts PyTorc

433 Nov 24, 2022

PyTorch reimplementation of the Smooth ReLU activation function proposed in the paper "Real World Large Scale Recommendation Systems Reproducibility and Smooth Activations" [arXiv 2022].

Smooth ReLU in PyTorch Unofficial PyTorch reimplementation of the Smooth ReLU (SmeLU) activation function proposed in the paper Real World Large Scale

10 Jan 02, 2023

A multi-scale unsupervised learning for deformable image registration

Related tags

Overview

A multi-scale unsupervised learning for deformable image registration

✏️ 📄 Citation

📈 Results

Acknowledgement

Owner

ShuweiShao

An Unpaired Sketch-to-Photo Translation Model

3D HourGlass Networks for Human Pose Estimation Through Videos

Official implementation of NeurIPS 2021 paper "Contextual Similarity Aggregation with Self-attention for Visual Re-ranking"

All of the figures and notebooks for my deep learning book, for free!

PyTorch implementation of the YOLO (You Only Look Once) v2

PyTorch reimplementation of the Smooth ReLU activation function proposed in the paper "Real World Large Scale Recommendation Systems Reproducibility and Smooth Activations" [arXiv 2022].

Semantic Edge Detection with Diverse Deep Supervision

CMUA-Watermark: A Cross-Model Universal Adversarial Watermark for Combating Deepfakes (AAAI2022)

Implementation of ICCV 2021 oral paper -- A Novel Self-Supervised Learning for Gaussian Mixture Model

NVIDIA Deep Learning Examples for Tensor Cores

The codebase for our paper "Generative Occupancy Fields for 3D Surface-Aware Image Synthesis" (NeurIPS 2021)

Co-mining: Self-Supervised Learning for Sparsely Annotated Object Detection, AAAI 2021.

GNPy: Optical Route Planning and DWDM Network Optimization

A Low Complexity Speech Enhancement Framework for Full-Band Audio (48kHz) based on Deep Filtering.

DziriBERT: a Pre-trained Language Model for the Algerian Dialect

An OpenAI-Gym Package for Training and Testing Reinforcement Learning algorithms with OpenSim Models

Equivariant GNN for the prediction of atomic multipoles up to quadrupoles.

A library of extension and helper modules for Python's data analysis and machine learning libraries.

A Re-implementation of the paper "A Deep Learning Framework for Character Motion Synthesis and Editing"

MODALS: Modality-agnostic Automated Data Augmentation in the Latent Space