From 9a6dcfa68f29c35ad1540195f10094cb78a95a90 Mon Sep 17 00:00:00 2001 From: gaotingquan Date: Fri, 25 Feb 2022 06:26:41 +0000 Subject: [PATCH] fix: rm some repvgg models not supported remove RepVGG_B3, RepVGG_B2g2 and RepVGG_B3g2 that are not supported by PaddleClas. --- ppcls/arch/backbone/model_zoo/repvgg.py | 39 ------------------------- 1 file changed, 39 deletions(-) diff --git a/ppcls/arch/backbone/model_zoo/repvgg.py b/ppcls/arch/backbone/model_zoo/repvgg.py index 4786016b..1218be7f 100644 --- a/ppcls/arch/backbone/model_zoo/repvgg.py +++ b/ppcls/arch/backbone/model_zoo/repvgg.py @@ -33,18 +33,12 @@ MODEL_URLS = { "https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/RepVGG_B1_pretrained.pdparams", "RepVGG_B2": "https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/RepVGG_B2_pretrained.pdparams", - "RepVGG_B3": - "https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/RepVGG_B3_pretrained.pdparams", "RepVGG_B1g2": "https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/RepVGG_B1g2_pretrained.pdparams", "RepVGG_B1g4": "https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/RepVGG_B1g4_pretrained.pdparams", - "RepVGG_B2g2": - "https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/RepVGG_B2g2_pretrained.pdparams", "RepVGG_B2g4": "https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/RepVGG_B2g4_pretrained.pdparams", - "RepVGG_B3g2": - "https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/RepVGG_B3g2_pretrained.pdparams", "RepVGG_B3g4": "https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/RepVGG_B3g4_pretrained.pdparams", } @@ -366,17 +360,6 @@ def RepVGG_B2(pretrained=False, use_ssld=False, **kwargs): return model -def RepVGG_B2g2(pretrained=False, use_ssld=False, **kwargs): - model = RepVGG( - num_blocks=[4, 6, 16, 1], - width_multiplier=[2.5, 2.5, 2.5, 5], - override_groups_map=g2_map, - **kwargs) - _load_pretrained( - pretrained, model, MODEL_URLS["RepVGG_B2g2"], use_ssld=use_ssld) - return model - - def RepVGG_B2g4(pretrained=False, use_ssld=False, **kwargs): model = RepVGG( num_blocks=[4, 6, 16, 1], @@ -388,28 +371,6 @@ def RepVGG_B2g4(pretrained=False, use_ssld=False, **kwargs): return model -def RepVGG_B3(pretrained=False, use_ssld=False, **kwargs): - model = RepVGG( - num_blocks=[4, 6, 16, 1], - width_multiplier=[3, 3, 3, 5], - override_groups_map=None, - **kwargs) - _load_pretrained( - pretrained, model, MODEL_URLS["RepVGG_B3"], use_ssld=use_ssld) - return model - - -def RepVGG_B3g2(pretrained=False, use_ssld=False, **kwargs): - model = RepVGG( - num_blocks=[4, 6, 16, 1], - width_multiplier=[3, 3, 3, 5], - override_groups_map=g2_map, - **kwargs) - _load_pretrained( - pretrained, model, MODEL_URLS["RepVGG_B3g2"], use_ssld=use_ssld) - return model - - def RepVGG_B3g4(pretrained=False, use_ssld=False, **kwargs): model = RepVGG( num_blocks=[4, 6, 16, 1], -- GitLab