(cfg)
| 86 | |
| 87 | |
| 88 | def load_seg_module(cfg): |
| 89 | # Network Builders |
| 90 | net_encoder = ModelBuilder.build_encoder( |
| 91 | arch=cfg.MODEL.arch_encoder, |
| 92 | fc_dim=cfg.MODEL.fc_dim, |
| 93 | weights=cfg.MODEL.weights_encoder) |
| 94 | net_decoder = ModelBuilder.build_decoder( |
| 95 | arch=cfg.MODEL.arch_decoder, |
| 96 | fc_dim=cfg.MODEL.fc_dim, |
| 97 | num_class=cfg.DATASET.num_class, |
| 98 | weights=cfg.MODEL.weights_decoder, |
| 99 | use_softmax=True) |
| 100 | segmentation_module = SegmentationModule(net_encoder, net_decoder, nn.NLLLoss(ignore_index=-1)) |
| 101 | return segmentation_module.cuda().eval() |
| 102 | |
| 103 | |
| 104 | def load_test_data(cfg): |
no test coverage detected