├── .flake8 ├── .github └── workflows │ ├── CI.yml │ └── Upload.yml ├── .gitignore ├── .style.yapf ├── LICENSE ├── README.md ├── setup.py ├── tests ├── README.md ├── __init__.py ├── test_autoaugment.py ├── test_data.py ├── test_lr_scheduler.py ├── test_metric.py ├── test_nn.py ├── test_obj.py ├── test_op.py ├── test_summary.py └── test_transform.py └── torchtoolbox ├── __init__.py ├── data ├── __init__.py ├── dataprefetcher.py ├── datasets.py ├── dynamic_data_provider.py ├── lmdb_dataset.py ├── sampler.py └── utils.py ├── io ├── __init__.py └── structured_data_io.py ├── metric ├── __init__.py ├── feature_verification.py ├── map.py └── metric.py ├── nn ├── __init__.py ├── activation.py ├── conv.py ├── functional.py ├── init.py ├── loss.py ├── metric_loss.py ├── modules.py ├── norm.py ├── operators.py ├── parallel │ ├── EncodingDataParallel.py │ └── __init__.py ├── sequential.py └── transformer.py ├── objects ├── __init__.py └── bbox.py ├── optimizer ├── __init__.py ├── drop_block_scheduler.py ├── lookahead.py ├── lr_scheduler.py └── sgd_gc.py ├── tools ├── __init__.py ├── config_parser.py ├── convert_lmdb.py ├── distribute.py ├── dotdict.py ├── mixup.py ├── registry.py ├── reset_model_setting.py ├── summary.py ├── tensor_transfer.py └── utils.py └── transform ├── __init__.py ├── autoaugment.py ├── dynamic_transform.py ├── functional.py ├── hybrid.py └── transforms.py /.flake8: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PistonY/torch-toolbox/HEAD/.flake8 -------------------------------------------------------------------------------- /.github/workflows/CI.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PistonY/torch-toolbox/HEAD/.github/workflows/CI.yml -------------------------------------------------------------------------------- /.github/workflows/Upload.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PistonY/torch-toolbox/HEAD/.github/workflows/Upload.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PistonY/torch-toolbox/HEAD/.gitignore -------------------------------------------------------------------------------- /.style.yapf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PistonY/torch-toolbox/HEAD/.style.yapf -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PistonY/torch-toolbox/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PistonY/torch-toolbox/HEAD/README.md -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PistonY/torch-toolbox/HEAD/setup.py -------------------------------------------------------------------------------- /tests/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PistonY/torch-toolbox/HEAD/tests/README.md -------------------------------------------------------------------------------- /tests/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PistonY/torch-toolbox/HEAD/tests/__init__.py -------------------------------------------------------------------------------- /tests/test_autoaugment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PistonY/torch-toolbox/HEAD/tests/test_autoaugment.py -------------------------------------------------------------------------------- /tests/test_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PistonY/torch-toolbox/HEAD/tests/test_data.py -------------------------------------------------------------------------------- /tests/test_lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PistonY/torch-toolbox/HEAD/tests/test_lr_scheduler.py -------------------------------------------------------------------------------- /tests/test_metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PistonY/torch-toolbox/HEAD/tests/test_metric.py -------------------------------------------------------------------------------- /tests/test_nn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PistonY/torch-toolbox/HEAD/tests/test_nn.py -------------------------------------------------------------------------------- /tests/test_obj.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PistonY/torch-toolbox/HEAD/tests/test_obj.py -------------------------------------------------------------------------------- /tests/test_op.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PistonY/torch-toolbox/HEAD/tests/test_op.py -------------------------------------------------------------------------------- /tests/test_summary.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PistonY/torch-toolbox/HEAD/tests/test_summary.py -------------------------------------------------------------------------------- /tests/test_transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PistonY/torch-toolbox/HEAD/tests/test_transform.py -------------------------------------------------------------------------------- /torchtoolbox/__init__.py: -------------------------------------------------------------------------------- 1 | # -*- coding: utf-8 -*- 2 | # Author: pistonyang@gmail.com 3 | 4 | VERSION = '0.1.8.3' 5 | -------------------------------------------------------------------------------- /torchtoolbox/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PistonY/torch-toolbox/HEAD/torchtoolbox/data/__init__.py -------------------------------------------------------------------------------- /torchtoolbox/data/dataprefetcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PistonY/torch-toolbox/HEAD/torchtoolbox/data/dataprefetcher.py -------------------------------------------------------------------------------- /torchtoolbox/data/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PistonY/torch-toolbox/HEAD/torchtoolbox/data/datasets.py -------------------------------------------------------------------------------- /torchtoolbox/data/dynamic_data_provider.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PistonY/torch-toolbox/HEAD/torchtoolbox/data/dynamic_data_provider.py -------------------------------------------------------------------------------- /torchtoolbox/data/lmdb_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PistonY/torch-toolbox/HEAD/torchtoolbox/data/lmdb_dataset.py -------------------------------------------------------------------------------- /torchtoolbox/data/sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PistonY/torch-toolbox/HEAD/torchtoolbox/data/sampler.py -------------------------------------------------------------------------------- /torchtoolbox/data/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PistonY/torch-toolbox/HEAD/torchtoolbox/data/utils.py -------------------------------------------------------------------------------- /torchtoolbox/io/__init__.py: -------------------------------------------------------------------------------- 1 | from .structured_data_io import * 2 | -------------------------------------------------------------------------------- /torchtoolbox/io/structured_data_io.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PistonY/torch-toolbox/HEAD/torchtoolbox/io/structured_data_io.py -------------------------------------------------------------------------------- /torchtoolbox/metric/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PistonY/torch-toolbox/HEAD/torchtoolbox/metric/__init__.py -------------------------------------------------------------------------------- /torchtoolbox/metric/feature_verification.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PistonY/torch-toolbox/HEAD/torchtoolbox/metric/feature_verification.py -------------------------------------------------------------------------------- /torchtoolbox/metric/map.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PistonY/torch-toolbox/HEAD/torchtoolbox/metric/map.py -------------------------------------------------------------------------------- /torchtoolbox/metric/metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PistonY/torch-toolbox/HEAD/torchtoolbox/metric/metric.py -------------------------------------------------------------------------------- /torchtoolbox/nn/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PistonY/torch-toolbox/HEAD/torchtoolbox/nn/__init__.py -------------------------------------------------------------------------------- /torchtoolbox/nn/activation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PistonY/torch-toolbox/HEAD/torchtoolbox/nn/activation.py -------------------------------------------------------------------------------- /torchtoolbox/nn/conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PistonY/torch-toolbox/HEAD/torchtoolbox/nn/conv.py -------------------------------------------------------------------------------- /torchtoolbox/nn/functional.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PistonY/torch-toolbox/HEAD/torchtoolbox/nn/functional.py -------------------------------------------------------------------------------- /torchtoolbox/nn/init.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PistonY/torch-toolbox/HEAD/torchtoolbox/nn/init.py -------------------------------------------------------------------------------- /torchtoolbox/nn/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PistonY/torch-toolbox/HEAD/torchtoolbox/nn/loss.py -------------------------------------------------------------------------------- /torchtoolbox/nn/metric_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PistonY/torch-toolbox/HEAD/torchtoolbox/nn/metric_loss.py -------------------------------------------------------------------------------- /torchtoolbox/nn/modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PistonY/torch-toolbox/HEAD/torchtoolbox/nn/modules.py -------------------------------------------------------------------------------- /torchtoolbox/nn/norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PistonY/torch-toolbox/HEAD/torchtoolbox/nn/norm.py -------------------------------------------------------------------------------- /torchtoolbox/nn/operators.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PistonY/torch-toolbox/HEAD/torchtoolbox/nn/operators.py -------------------------------------------------------------------------------- /torchtoolbox/nn/parallel/EncodingDataParallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PistonY/torch-toolbox/HEAD/torchtoolbox/nn/parallel/EncodingDataParallel.py -------------------------------------------------------------------------------- /torchtoolbox/nn/parallel/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PistonY/torch-toolbox/HEAD/torchtoolbox/nn/parallel/__init__.py -------------------------------------------------------------------------------- /torchtoolbox/nn/sequential.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PistonY/torch-toolbox/HEAD/torchtoolbox/nn/sequential.py -------------------------------------------------------------------------------- /torchtoolbox/nn/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PistonY/torch-toolbox/HEAD/torchtoolbox/nn/transformer.py -------------------------------------------------------------------------------- /torchtoolbox/objects/__init__.py: -------------------------------------------------------------------------------- 1 | from .bbox import * 2 | -------------------------------------------------------------------------------- /torchtoolbox/objects/bbox.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PistonY/torch-toolbox/HEAD/torchtoolbox/objects/bbox.py -------------------------------------------------------------------------------- /torchtoolbox/optimizer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PistonY/torch-toolbox/HEAD/torchtoolbox/optimizer/__init__.py -------------------------------------------------------------------------------- /torchtoolbox/optimizer/drop_block_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PistonY/torch-toolbox/HEAD/torchtoolbox/optimizer/drop_block_scheduler.py -------------------------------------------------------------------------------- /torchtoolbox/optimizer/lookahead.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PistonY/torch-toolbox/HEAD/torchtoolbox/optimizer/lookahead.py -------------------------------------------------------------------------------- /torchtoolbox/optimizer/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PistonY/torch-toolbox/HEAD/torchtoolbox/optimizer/lr_scheduler.py -------------------------------------------------------------------------------- /torchtoolbox/optimizer/sgd_gc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PistonY/torch-toolbox/HEAD/torchtoolbox/optimizer/sgd_gc.py -------------------------------------------------------------------------------- /torchtoolbox/tools/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PistonY/torch-toolbox/HEAD/torchtoolbox/tools/__init__.py -------------------------------------------------------------------------------- /torchtoolbox/tools/config_parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PistonY/torch-toolbox/HEAD/torchtoolbox/tools/config_parser.py -------------------------------------------------------------------------------- /torchtoolbox/tools/convert_lmdb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PistonY/torch-toolbox/HEAD/torchtoolbox/tools/convert_lmdb.py -------------------------------------------------------------------------------- /torchtoolbox/tools/distribute.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PistonY/torch-toolbox/HEAD/torchtoolbox/tools/distribute.py -------------------------------------------------------------------------------- /torchtoolbox/tools/dotdict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PistonY/torch-toolbox/HEAD/torchtoolbox/tools/dotdict.py -------------------------------------------------------------------------------- /torchtoolbox/tools/mixup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PistonY/torch-toolbox/HEAD/torchtoolbox/tools/mixup.py -------------------------------------------------------------------------------- /torchtoolbox/tools/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PistonY/torch-toolbox/HEAD/torchtoolbox/tools/registry.py -------------------------------------------------------------------------------- /torchtoolbox/tools/reset_model_setting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PistonY/torch-toolbox/HEAD/torchtoolbox/tools/reset_model_setting.py -------------------------------------------------------------------------------- /torchtoolbox/tools/summary.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PistonY/torch-toolbox/HEAD/torchtoolbox/tools/summary.py -------------------------------------------------------------------------------- /torchtoolbox/tools/tensor_transfer.py: -------------------------------------------------------------------------------- 1 | """just let it empty. 2 | """ 3 | -------------------------------------------------------------------------------- /torchtoolbox/tools/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PistonY/torch-toolbox/HEAD/torchtoolbox/tools/utils.py -------------------------------------------------------------------------------- /torchtoolbox/transform/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PistonY/torch-toolbox/HEAD/torchtoolbox/transform/__init__.py -------------------------------------------------------------------------------- /torchtoolbox/transform/autoaugment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PistonY/torch-toolbox/HEAD/torchtoolbox/transform/autoaugment.py -------------------------------------------------------------------------------- /torchtoolbox/transform/dynamic_transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PistonY/torch-toolbox/HEAD/torchtoolbox/transform/dynamic_transform.py -------------------------------------------------------------------------------- /torchtoolbox/transform/functional.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PistonY/torch-toolbox/HEAD/torchtoolbox/transform/functional.py -------------------------------------------------------------------------------- /torchtoolbox/transform/hybrid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PistonY/torch-toolbox/HEAD/torchtoolbox/transform/hybrid.py -------------------------------------------------------------------------------- /torchtoolbox/transform/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PistonY/torch-toolbox/HEAD/torchtoolbox/transform/transforms.py --------------------------------------------------------------------------------