File tree Expand file tree Collapse file tree 1 file changed +2
-3
lines changed
Expand file tree Collapse file tree 1 file changed +2
-3
lines changed Original file line number Diff line number Diff line change @@ -31,10 +31,10 @@ def get_model(config: dict, pretrained=False) -> torch.nn.Module:
3131
3232 if config ['model' ] == 'UNet' :
3333 model = models .unet .UNet (config [config ['model' ]]['num_classes' ])
34- elif config ['model' ] == 'Proposed' :
35- model = models .proposed .Proposed (config [config ['model' ]]['num_classes' ])
3634 elif config ['model' ] == 'Backbone' :
3735 model = models .backbone .Backbone (config [config ['model' ]]['num_classes' ])
36+ elif config ['model' ] == 'Proposed' :
37+ model = models .proposed .Proposed (config [config ['model' ]]['num_classes' ])
3838 else :
3939 raise NameError ('Wrong model_name.' )
4040
@@ -46,7 +46,6 @@ def get_model(config: dict, pretrained=False) -> torch.nn.Module:
4646 return model
4747
4848
49- # Cityscapes 데이터셋
5049class Cityscapes :
5150 def __init__ (self , config : dict ):
5251 self .config = config
You can’t perform that action at this time.
0 commit comments