def __init__(self, phase, net_flag):
"""
"""
super(LaneNetFrondEnd, self).__init__()
self._frontend_net_map = {'vgg': vgg16_based_fcn.VGG16FCN(phase=phase)}
self._net = self._frontend_net_map[net_flag]
def __init__(self, phase, net_flag, cfg):
"""
"""
super(LaneNetFrondEnd, self).__init__()
self._cfg = cfg
self._frontend_net_map = {
'vgg': vgg16_based_fcn.VGG16FCN(phase=phase, cfg=self._cfg),
'bisenetv2': bisenet_v2.BiseNetV2(phase=phase, cfg=self._cfg),
}
self._net = self._frontend_net_map[net_flag]