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

My solution for the 7th place / 245 in the Umoja Hack 2022 challenge

Umoja Hack 2022 : Insurance Claim Challenge My solution for the 7th place / 245 in the Umoja Hack 2022 challenge Umoja Hack Africa is a yearly hackath

17 Jun 03, 2022

Official implementation of NeurIPS'2021 paper TransformerFusion

TransformerFusion: Monocular RGB Scene Reconstruction using Transformers Project Page | Paper | Video TransformerFusion: Monocular RGB Scene Reconstru

118 Dec 25, 2022

SEC'21: Sparse Bitmap Compression for Memory-Efficient Training onthe Edge

Training Deep Learning Models on The Edge Training on the Edge enables continuous learning from new data for deployed neural networks on memory-constr

4 Nov 18, 2022

[LREC] MMChat: Multi-Modal Chat Dataset on Social Media

MMChat This repo contains the code and data for the LREC2022 paper MMChat: Multi-Modal Chat Dataset on Social Media. Dataset MMChat is a large-scale d

47 Jan 03, 2023

Code to train models from "Paraphrastic Representations at Scale".

Paraphrastic Representations at Scale Code to train models from "Paraphrastic Representations at Scale". The code is written in Python 3.7 and require

71 Dec 19, 2022

Winning solution of the Indoor Location & Navigation Kaggle competition

This repository contains the code to generate the winning solution of the Kaggle competition on indoor location and navigation organized by Microsoft

62 Dec 28, 2022

Official PyTorch implementation of MX-Font (Multiple Heads are Better than One: Few-shot Font Generation with Multiple Localized Experts)

Introduction Pytorch implementation of Multiple Heads are Better than One: Few-shot Font Generation with Multiple Localized Expert. | paper Song Park1

97 Dec 23, 2022

A unified 3D Transformer Pipeline for visual synthesis

Overview This is the official repo for the paper: NÜWA: Visual Synthesis Pre-training for Neural visUal World creAtion. NÜWA is a unified multimodal p

2.6k Jan 06, 2023

Implementation of Uniformer, a simple attention and 3d convolutional net that achieved SOTA in a number of video classification tasks

Uniformer - Pytorch Implementation of Uniformer, a simple attention and 3d convolutional net that achieved SOTA in a number of video classification ta

90 Nov 24, 2022

Official release of MSHT: Multi-stage Hybrid Transformer for the ROSE Image Analysis of Pancreatic Cancer axriv: http://arxiv.org/abs/2112.13513

MSHT: Multi-stage Hybrid Transformer for the ROSE Image Analysis This is the official page of the MSHT with its experimental script and records. We de

53 Dec 27, 2022

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

My solution for the 7th place / 245 in the Umoja Hack 2022 challenge

Official implementation of NeurIPS'2021 paper TransformerFusion

SEC'21: Sparse Bitmap Compression for Memory-Efficient Training onthe Edge

[LREC] MMChat: Multi-Modal Chat Dataset on Social Media

Code to train models from "Paraphrastic Representations at Scale".

Winning solution of the Indoor Location & Navigation Kaggle competition

Official PyTorch implementation of MX-Font (Multiple Heads are Better than One: Few-shot Font Generation with Multiple Localized Experts)

A unified 3D Transformer Pipeline for visual synthesis

Implementation of Uniformer, a simple attention and 3d convolutional net that achieved SOTA in a number of video classification tasks

Official release of MSHT: Multi-stage Hybrid Transformer for the ROSE Image Analysis of Pancreatic Cancer axriv: http://arxiv.org/abs/2112.13513

領域を指定し、キーを入力することで画像を保存するツールです。クラス分類用のデータセット作成を想定しています。

DeFMO: Deblurring and Shape Recovery of Fast Moving Objects (CVPR 2021)

JDet is Object Detection Framework based on Jittor.

A pytorch-based real-time segmentation model for autonomous driving

My take on a practical implementation of Linformer for Pytorch.

Easy Parallel Library (EPL) is a general and efficient deep learning framework for distributed model training.

The code for paper "Learning Implicit Fields for Generative Shape Modeling".

tinykernel - A minimal Python kernel so you can run Python in your Python

Implementation of the paper Recurrent Glimpse-based Decoder for Detection with Transformer.

The official implementation of Equalization Loss for Long-Tailed Object Recognition (CVPR 2020) based on Detectron2