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

from .modules import (
    Conv2dReLU,
    Attention,
)

from .heads import (
    SegmentationHead,
    ClassificationHead,
)