A tensorflow2 implementation of some basic CNNs(MobileNetV1/V2/V3, EfficientNet, ResNeXt, InceptionV4, InceptionResNetV1/V2, SENet, SqueezeNet, DenseNet, ShuffleNetV2, ResNet).

SE_ResNet && SE_ResNeXt with pretrained weights on ImageNet (SENet In TensorFlow)

Models supported: ResNet, ResNetV2, SE-ResNet, ResNeXt, SE-ResNeXt [layers: 18, 34, 50, 101, 152] (1D and 2D versions with DEMO for Classification and Regression).

Hierarchical probabilistic 3D U-Net, with attention mechanisms (—𝘈𝘵𝘵𝘦𝘯𝘵𝘪𝘰𝘯 𝘜-𝘕𝘦𝘵, 𝘚𝘌𝘙𝘦𝘴𝘕𝘦𝘵) and a nested decoder structure with deep supervision (—𝘜𝘕𝘦𝘵++). Built in TensorFlow 2.5. Configured for voxel-level clinically significant prostate cancer detection in multi-channel 3D bpMRI scans.

Drastically Reducing the Number of Trainable Parameters in Deep CNNs by Inter-layer Kernel-sharing

Pytorch implementation of vision models.

Official implementation of our paper "Transparency in Sleep Staging: Deep Learning Method for EEG Sleep Stage Classification with Model Interpretability"

Fully supervised, healthy/malignant prostate detection in multi-parametric MRI (T2W, DWI, ADC), using a modified 2D RetinaNet model for medical object detection, built upon a shallow SEResNet backbone.