Args: opt: training options
(self, opt)
| 16 | Main class for segmentation model creation |
| 17 | """ |
| 18 | def __init__(self, opt): |
| 19 | """ |
| 20 | Args: |
| 21 | opt: training options |
| 22 | """ |
| 23 | self.model_type = opt["model_type"] |
| 24 | self.image_height = opt["datasets"]["train"]["img_size_h"] |
| 25 | self.image_width = opt["datasets"]["train"]["img_size_w"] |
| 26 | self.num_classes = opt["train"]["num_classes"] + 1 |
| 27 | |
| 28 | def build(self): |
| 29 | """ |
nothing calls this directly
no outgoing calls
no test coverage detected