Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
test_config.py 2.87 KB
Newer Older
Sugon_ldc's avatar
add model TSM  
Sugon_ldc committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74 
# Copyright (c) OpenMMLab. All rights reserved.
import glob
import os
import os.path as osp

import mmcv
import torch.nn as nn

from mmaction.models import build_localizer, build_recognizer


def _get_config_path():
    """Find the predefined recognizer config path."""
    repo_dir = osp.dirname(osp.dirname(osp.dirname(__file__)))
    config_dpath = osp.join(repo_dir, 'configs')
    if not osp.exists(config_dpath):
        raise Exception('Cannot find config path')
    config_fpaths = list(glob.glob(osp.join(config_dpath, '*.py')))
    config_names = [os.path.relpath(p, config_dpath) for p in config_fpaths]
    print(f'Using {len(config_names)} config files')
    config_fpaths = [
        osp.join(config_dpath, config_fpath) for config_fpath in config_fpaths
    ]
    return config_fpaths


def test_config_build_recognizer():
    """Test that all mmaction models defined in the configs can be
    initialized."""
    repo_dir = osp.dirname(osp.dirname(osp.dirname(__file__)))
    config_dpath = osp.join(repo_dir, 'configs/recognition')
    if not osp.exists(config_dpath):
        raise Exception('Cannot find config path')
    config_fpaths = list(glob.glob(osp.join(config_dpath, '*.py')))
    # test all config file in `configs` directory
    for config_fpath in config_fpaths:
        config_mod = mmcv.Config.fromfile(config_fpath)
        print(f'Building recognizer, config_fpath = {config_fpath!r}')

        # Remove pretrained keys to allow for testing in an offline environment
        if 'pretrained' in config_mod.model['backbone']:
            config_mod.model['backbone']['pretrained'] = None

        recognizer = build_recognizer(config_mod.model)
        assert isinstance(recognizer, nn.Module)


def _get_config_path_for_localizer():
    """Find the predefined localizer config path for localizer."""
    repo_dir = osp.dirname(osp.dirname(osp.dirname(__file__)))
    config_dpath = osp.join(repo_dir, 'configs/localization')
    if not osp.exists(config_dpath):
        raise Exception('Cannot find config path')
    config_fpaths = list(glob.glob(osp.join(config_dpath, '*.py')))
    config_names = [os.path.relpath(p, config_dpath) for p in config_fpaths]
    print(f'Using {len(config_names)} config files')
    config_fpaths = [
        osp.join(config_dpath, config_fpath) for config_fpath in config_fpaths
    ]
    return config_fpaths


def test_config_build_localizer():
    """Test that all mmaction models defined in the configs can be
    initialized."""
    config_fpaths = _get_config_path_for_localizer()

    # test all config file in `configs/localization` directory
    for config_fpath in config_fpaths:
        config_mod = mmcv.Config.fromfile(config_fpath)
        print(f'Building localizer, config_fpath = {config_fpath!r}')
        if config_mod.get('model', None):
            localizer = build_localizer(config_mod.model)
            assert isinstance(localizer, nn.Module)