Merge branch 'add_Recommendation' into 'main'

添加openmmlab测试用例

See merge request dcutoolkit/deeplearing/dlexamples_new!32

openmmlab_test/mmclassification-speed-benchmark/mmcls/utils/__init__.py

+from .collect_env import collect_env
+from .logger import get_root_logger
+
+__all__ = ['collect_env', 'get_root_logger']

openmmlab_test/mmclassification-speed-benchmark/mmcls/utils/collect_env.py

+from mmcv.utils import collect_env as collect_base_env
+from mmcv.utils import get_git_hash
+
+import mmcls
+
+
+def collect_env():
+    """Collect the information of the running environments."""
+    env_info = collect_base_env()
+    env_info['MMClassification'] = mmcls.__version__ + '+' + get_git_hash()[:7]
+    return env_info
+
+
+if __name__ == '__main__':
+    for name, val in collect_env().items():
+        print(f'{name}: {val}')

openmmlab_test/mmclassification-speed-benchmark/mmcls/utils/logger.py

+import logging
+
+from mmcv.utils import get_logger
+
+
+def get_root_logger(log_file=None, log_level=logging.INFO):
+    return get_logger('mmcls', log_file, log_level)

openmmlab_test/mmclassification-speed-benchmark/mmcls/version.py

+# Copyright (c) Open-MMLab. All rights reserved.
+
+__version__ = '0.12.0'
+
+
+def parse_version_info(version_str):
+    """Parse a version string into a tuple.
+
+    Args:
+        version_str (str): The version string.
+    Returns:
+        tuple[int | str]: The version info, e.g., "1.3.0" is parsed into
+            (1, 3, 0), and "2.0.0rc1" is parsed into (2, 0, 0, 'rc1').
+    """
+    version_info = []
+    for x in version_str.split('.'):
+        if x.isdigit():
+            version_info.append(int(x))
+        elif x.find('rc') != -1:
+            patch_version = x.split('rc')
+            version_info.append(int(patch_version[0]))
+            version_info.append(f'rc{patch_version[1]}')
+    return tuple(version_info)
+
+
+version_info = parse_version_info(__version__)
+
+__all__ = ['__version__', 'version_info', 'parse_version_info']

openmmlab_test/mmclassification-speed-benchmark/model_zoo.yml

+Import:
+  - configs/fp16/metafile.yml
+  - configs/mobilenet_v2/metafile.yml
+  - configs/resnet/metafile.yml
+  - configs/resnext/metafile.yml
+  - configs/seresnet/metafile.yml
+  - configs/seresnext/metafile.yml
+  - configs/shufflenet_v1/metafile.yml
+  - configs/shufflenet_v2/metafile.yml
+  - configs/vgg/metafile.yml

openmmlab_test/mmclassification-speed-benchmark/multi_test.sh

+#!/bin/bash
+export HIP_VISIBLE_DEVICES=0,1
+export MIOPEN_FIND_MODE=3
+export HSA_FORCE_FINE_GRAIN_PCIE=1
+my_config=$1
+numactl --cpunodebind=2 --membind=2 python3 -m torch.distributed.launch --nproc_per_node=2 --master_port=29500   tools/train.py $my_config --launcher pytorch
+

openmmlab_test/mmclassification-speed-benchmark/requirements.txt

+-r requirements/optional.txt
+-r requirements/runtime.txt
+-r requirements/tests.txt

openmmlab_test/mmclassification-speed-benchmark/requirements/docs.txt

+recommonmark
+sphinx
+sphinx_markdown_tables
+sphinx_rtd_theme

openmmlab_test/mmclassification-speed-benchmark/requirements/mminstall.txt

+mmcv-full>=1.3.0,<=1.5.0

openmmlab_test/mmclassification-speed-benchmark/requirements/optional.txt

+albumentations>=0.3.2

openmmlab_test/mmclassification-speed-benchmark/requirements/readthedocs.txt

+mmcv>=1.3.0
+torch
+torchvision

openmmlab_test/mmclassification-speed-benchmark/requirements/runtime.txt

+matplotlib
+numpy

openmmlab_test/mmclassification-speed-benchmark/requirements/tests.txt

+codecov
+flake8
+interrogate
+isort==4.3.21
+pytest
+xdoctest >= 0.10.0
+yapf

openmmlab_test/mmclassification-speed-benchmark/setup.cfg

+[bdist_wheel]
+universal=1
+
+[aliases]
+test=pytest
+
+[yapf]
+based_on_style = pep8
+blank_line_before_nested_class_or_def = true
+split_before_expression_after_opening_paren = true
+
+[isort]
+line_length = 79
+multi_line_output = 0
+known_standard_library = pkg_resources,setuptools
+known_first_party = mmcls
+known_third_party = PIL,cv2,matplotlib,mmcv,numpy,onnxruntime,pytest,torch,torchvision,ts
+no_lines_before = STDLIB,LOCALFOLDER
+default_section = THIRDPARTY

openmmlab_test/mmclassification-speed-benchmark/setup.py

+from setuptools import find_packages, setup
+
+
+def readme():
+    with open('README.md', encoding='utf-8') as f:
+        content = f.read()
+    return content
+
+
+def get_version():
+    version_file = 'mmcls/version.py'
+    with open(version_file, 'r', encoding='utf-8') as f:
+        exec(compile(f.read(), version_file, 'exec'))
+    return locals()['__version__']
+
+
+def parse_requirements(fname='requirements.txt', with_version=True):
+    """Parse the package dependencies listed in a requirements file but strips
+    specific versioning information.
+
+    Args:
+        fname (str): path to requirements file
+        with_version (bool, default=False): if True include version specs
+
+    Returns:
+        List[str]: list of requirements items
+
+    CommandLine:
+        python -c "import setup; print(setup.parse_requirements())"
+    """
+    import re
+    import sys
+    from os.path import exists
+    require_fpath = fname
+
+    def parse_line(line):
+        """Parse information from a line in a requirements text file."""
+        if line.startswith('-r '):
+            # Allow specifying requirements in other files
+            target = line.split(' ')[1]
+            for info in parse_require_file(target):
+                yield info
+        else:
+            info = {'line': line}
+            if line.startswith('-e '):
+                info['package'] = line.split('#egg=')[1]
+            else:
+                # Remove versioning from the package
+                pat = '(' + '|'.join(['>=', '==', '>']) + ')'
+                parts = re.split(pat, line, maxsplit=1)
+                parts = [p.strip() for p in parts]
+
+                info['package'] = parts[0]
+                if len(parts) > 1:
+                    op, rest = parts[1:]
+                    if ';' in rest:
+                        # Handle platform specific dependencies
+                        # http://setuptools.readthedocs.io/en/latest/setuptools.html#declaring-platform-specific-dependencies
+                        version, platform_deps = map(str.strip,
+                                                     rest.split(';'))
+                        info['platform_deps'] = platform_deps
+                    else:
+                        version = rest  # NOQA
+                    info['version'] = (op, version)
+            yield info
+
+    def parse_require_file(fpath):
+        with open(fpath, 'r') as f:
+            for line in f.readlines():
+                line = line.strip()
+                if line and not line.startswith('#'):
+                    for info in parse_line(line):
+                        yield info
+
+    def gen_packages_items():
+        if exists(require_fpath):
+            for info in parse_require_file(require_fpath):
+                parts = [info['package']]
+                if with_version and 'version' in info:
+                    parts.extend(info['version'])
+                if not sys.version.startswith('3.4'):
+                    # apparently package_deps are broken in 3.4
+                    platform_deps = info.get('platform_deps')
+                    if platform_deps is not None:
+                        parts.append(';' + platform_deps)
+                item = ''.join(parts)
+                yield item
+
+    packages = list(gen_packages_items())
+    return packages
+
+
+setup(
+    name='mmcls',
+    version=get_version(),
+    description='OpenMMLab Image Classification Toolbox and Benchmark',
+    long_description=readme(),
+    long_description_content_type='text/markdown',
+    author='OpenMMLab',
+    author_email='openmmlab@gmail.com',
+    keywords='computer vision, image classification',
+    url='https://github.com/open-mmlab/mmclassification',
+    packages=find_packages(exclude=('configs', 'tools', 'demo')),
+    include_package_data=True,
+    classifiers=[
+        'Development Status :: 4 - Beta',
+        'License :: OSI Approved :: Apache Software License',
+        'Operating System :: OS Independent',
+        'Programming Language :: Python :: 3',
+        'Programming Language :: Python :: 3.5',
+        'Programming Language :: Python :: 3.6',
+        'Programming Language :: Python :: 3.7',
+        'Programming Language :: Python :: 3.8',
+    ],
+    license='Apache License 2.0',
+    tests_require=parse_requirements('requirements/tests.txt'),
+    install_requires=parse_requirements('requirements/runtime.txt'),
+    zip_safe=False)

openmmlab_test/mmclassification-speed-benchmark/sing_test.sh

+#!/bin/bash
+export HIP_VISIBLE_DEVICES=3
+export MIOPEN_FIND_MODE=3
+my_config=$1
+python3 tools/train.py $my_config
+