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