def __init__(self, model_id, project_dir): super(DeepLabV3, self).__init__() self.num_classes = 5 self.model_id = model_id self.project_dir = project_dir self.resnet = ResNet18_OS8() # NOTE! specify the type of ResNet here self.aspp = ASPP(num_classes=self.num_classes) # NOTE! if you use ResNet50-152, set self.aspp = ASPP_Bottleneck(num_classes=self.num_classes) instead
def __init__(self, model_id, project_dir): super(DeepLabV3, self).__init__() # extend from parent class's constructor self.num_classes = 20 # cityscapes class constains 20 kinds + one background class self.model_id = model_id # model_id = "eval_seq" self.project_dir = project_dir # project_dir = "/root/deeplabv3" self.create_model_dirs() self.resnet = ResNet18_OS8() # NOTE! specify the type of ResNet here self.aspp = ASPP(num_classes=self.num_classes) # NOTE! if you use ResNet50-152, set self.aspp = ASPP_Bottleneck(num_classes=self.num_classes) instead