self-supervised-landmarks

Repository for self-supervised landmark discovery

Requirements

pytorch
pynrrd (for 3d images)

Usage

The use of this models is via config files, an example config file for Shepp-logan phantom dataset is given in ./configs/phantom_data.json

To train the model

python train.py --model="2d or 3d" --config_file="path to config file"

The network could be 2d or 3d and the second argument is the config file path. all the other parameters including the save and data director is inside teh config file

For inference

python test.py --model="2d or 3d" --config_file="path to config file" --redu_remove --use_best --num_out=num ts to be retained

redu_remove is a boolean argument that determines if redundant points are removed or not use best is also a boolean argument that determines if the best checkpoint is used or the final checkpoint. num_out is an integer that determines the number of particles to be retained after redundancy removal

Reference

If you are utilizing this code please cite one of the following

Leveraging unsupervised image registration for discovery of landmark shape descriptor

@article{bhalodia2021leveraging,
title = {Leveraging unsupervised image registration for discovery of landmark shape descriptor},
journal = {Medical Image Analysis},
volume = {73},
pages = {102157},
year = {2021},
issn = {1361-8415},
doi = {https://doi.org/10.1016/j.media.2021.102157},
url = {https://www.sciencedirect.com/science/article/pii/S1361841521002036},
author = {Riddhish Bhalodia and Shireen Elhabian and Ladislav Kavan and Ross Whitaker}
}

Self-supervised discovery of anatomical shape landmarks

@inproceedings{bhalodia2020self,
  title={Self-supervised discovery of anatomical shape landmarks},
  author={Bhalodia, Riddhish and Kavan, Ladislav and Whitaker, Ross T},
  booktitle={International Conference on Medical Image Computing and Computer-Assisted Intervention},
  pages={627--638},
  year={2020},
  organization={Springer}
}

Repository for self-supervised landmark discovery

Related tags

Overview

self-supervised-landmarks

Requirements

Usage

Reference

Owner

Riddhish Bhalodia

Code repo for realtime multi-person pose estimation in CVPR'17 (Oral)

Scikit-learn compatible estimation of general graphical models

Multi-task Learning of Order-Consistent Causal Graphs (NeuRIPs 2021)

The-Secret-Sharing-Schemes - This interactive script demonstrates the Secret Sharing Schemes algorithm

A lightweight Python-based 3D network multi-agent simulator. Uses a cell-based congestion model. Calculates risk, loudness and battery capacities of the agents. Suitable for 3D network optimization tasks.

Equivariant Imaging: Learning Beyond the Range Space

An introduction to satellite image analysis using Python + OpenCV and JavaScript + Google Earth Engine

High performance distributed framework for training deep learning recommendation models based on PyTorch.

Exposure Time Calculator (ETC) and radial velocity precision estimator for the Near InfraRed Planet Searcher (NIRPS) spectrograph

Instance-level Image Retrieval using Reranking Transformers

Source code for CIKM 2021 paper for Relation-aware Heterogeneous Graph for User Profiling

TensorFlow2 Classification Model Zoo playing with TensorFlow2 on the CIFAR-10 dataset.

Implementation of Enformer, Deepmind's attention network for predicting gene expression, in Pytorch

Deep Learning and Logical Reasoning from Data and Knowledge

This repository contains an overview of important follow-up works based on the original Vision Transformer (ViT) by Google.

Reinforcement Learning via Supervised Learning

Dynamic Bottleneck for Robust Self-Supervised Exploration

UniFormer - official implementation of UniFormer

DeepSTD: Mining Spatio-temporal Disturbances of Multiple Context Factors for Citywide Traffic Flow Prediction

This repository contains all source code, pre-trained models related to the paper "An Empirical Study on GANs with Margin Cosine Loss and Relativistic Discriminator"