a | b/models/base_model.py | ||
---|---|---|---|
1 | # Manuel A. Morales (moralesq@mit.edu) |
||
2 | # Harvard-MIT Department of Health Sciences & Technology |
||
3 | # Athinoula A. Martinos Center for Biomedical Imaging |
||
4 | |||
5 | from abc import ABC, abstractmethod |
||
6 | |||
7 | class BaseModel(ABC): |
||
8 | |||
9 | def __init__(self, opt): |
||
10 | self.opt = opt |
||
11 | |||
12 | @abstractmethod |
||
13 | def get_netS(self): |
||
14 | """Generate and save segmentations.""" |
||
15 | pass |
||
16 | |||
17 | @abstractmethod |
||
18 | def get_netME(self): |
||
19 | """Generate and save motion estimates.""" |
||
20 | pass |
||
21 | |||
22 |