Skip to content

Commit cead6f4

Browse files
authored
fix bug in demo_skeleton.py (#2214)
1 parent a4cdf17 commit cead6f4

File tree

2 files changed

+2
-4
lines changed

2 files changed

+2
-4
lines changed

demo/demo_skeleton.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -170,9 +170,6 @@ def main():
170170

171171
config = mmengine.Config.fromfile(args.config)
172172
config.merge_from_dict(args.cfg_options)
173-
if 'data_preprocessor' in config.model:
174-
config.model.data_preprocessor['mean'] = (w // 2, h // 2, .5)
175-
config.model.data_preprocessor['std'] = (w, h, 1.)
176173

177174
model = init_recognizer(config, args.checkpoint, args.device)
178175
result = inference_recognizer(model, fake_anno)

mmaction/apis/inference.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -37,9 +37,10 @@ def init_recognizer(config: Union[str, Path, mmengine.Config],
3737
raise TypeError('config must be a filename or Config object, '
3838
f'but got {type(config)}')
3939

40-
config.model.backbone.pretrained = None
4140
init_default_scope(config.get('default_scope', 'mmaction'))
4241

42+
if config.model.backbone.get('pretrained', None):
43+
config.model.backbone.pretrained = None
4344
model = MODELS.build(config.model)
4445

4546
if checkpoint is not None:

0 commit comments

Comments
 (0)