Skip to content

torchmil

torchmil.nn.gnns

Initializing search

Franblueee/torchmil

Introduction
Examples
API reference

torchmil

Franblueee/torchmil

Introduction
Introduction
Examples
Examples
API reference
API reference
- torchmil.data
  torchmil.data
  - Collate
  - Representation
- torchmil.datasets
  torchmil.datasets
- torchmil.nn
  torchmil.nn
  - Attention
    
    Attention
    
    Attention Pool
    
    Sm Attention Pool
    
    Prob Smooth Attention Pool
    
    Multihead Self-Attention
    
    iRPE Multihead Self-Attention
    
    Nyström Attention
    
    Multihead Cross-Attention
  - Graph Neural Networks (GNNs)
    
    Graph Neural Networks (GNNs)
    
    DeepGCNLayer
    
    GCNConv
    
    Dense MinCut pooling
  - Transformers
    
    Transformers
    
    Transformer base class
    
    Conventional Transformer
    
    Sm Transformer
    
    Nyström Transformer
    
    iRPE Transformer
    
    Tokens-to-Token (T2T)
  - Sm operator
  - Mean Pool
  - Max Pool
- torchmil.models
  torchmil.models
  - General MIL model
  - ABMIL
  - CAMIL
  - CLAM
  - DeepGraphSurv
  - DFTDMIL
  - DSMIL
  - GTP
  - IIBMIL
  - PatchGCN
  - ProbSmoothABMIL
  - SmABMIL
  - SmTransformerABMIL
  - TransformerABMIL
  - TransformerProbSmoothABMIL
  - TransMIL
  - SETMIL
  - IIBMIL
- torchmil.visualize
  torchmil.visualize
  - Visualizing CT scans
  - Visualizing WSIs
- torchmil.utils
  torchmil.utils

torchmil.nn.gnns

Deep Graph Convolutional Network (DeepGCN) layer
Graph Convolutional Network (GCN) convolution
Dense MinCut pooling

Made with Material for MkDocs