File size: 163 Bytes
be2715b
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
from .model import SegmentationModel

from .modules import (
    Conv2dReLU,
    Attention,
)

from .heads import (
    SegmentationHead,
    ClassificationHead,
)