Stars
Model interpretability and understanding for PyTorch
Change is Everywhere: Single-Temporal Supervised Object Change Detection in Remote Sensing Imagery (ICCV 2021) https://arxiv.org/abs/2108.07002
The source code and data for "Fast 3D image generation for healthy brain aging using diffeomorphic registration", published in HBM, 2023.
TransMorph: Transformer for Unsupervised Medical Image Registration (PyTorch)
Code for ECA-Net: Efficient Channel Attention for Deep Convolutional Neural Networks
CPAB Transformations: finite-dimensional spaces of simple, fast, and highly-expressive diffeomorphisms derived from parametric, continuously-defined, velocity fields in Numpy, Tensorflow and Pytorch
Implementation of "Data augmentation using learned transforms for one-shot medical image segmentation"
A PyTorch framework for medical image segmentation
Contextual Attention Network: Transformer Meets U-Net
Implementations of recent research prototypes/demonstrations using MONAI.
Implementation of ViViT: A Video Vision Transformer
modify self-attention model for EEG signal as input and image embedding layer as output
fMRIPrep is a robust and easy-to-use pipeline for preprocessing of diverse fMRI data. The transparent workflow dispenses of manual intervention, thereby ensuring the reproducibility of the results.
🤗 Transformers: the model-definition framework for state-of-the-art machine learning models in text, vision, audio, and multimodal models, for both inference and training.
3D residual neural network that predicts age based on brain MRI images (implemented using TensorFlow 2).
STAGIN: Spatio-Temporal Attention Graph Isomorphism Network
Master's thesis about sparse approximation and dictionary learning using Cloud K-SVD for image denoising. Results show that the algorithm is able to learn sparse representations of signal vectors f…
DPT: Deformable Patch-based Transformer for Visual Recognition (ACM MM2021)
The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (V…
PyTorch code for Vision Transformers training with the Self-Supervised learning method DINO
Official code implementation for "Personalized Federated Learning using Hypernetworks" [ICML 2021]
Implementation of OmniNet, Omnidirectional Representations from Transformers, in Pytorch
Serve, optimize and scale PyTorch models in production
This repository includes the official project of TransUNet, presented in our paper: TransUNet: Transformers Make Strong Encoders for Medical Image Segmentation.
TensorFlow code and pre-trained models for BERT