├── .github └── workflows │ └── publish.yml ├── .gitignore ├── .pre-commit-config.yaml ├── .readthedocs.yaml ├── LICENSE ├── README.md ├── docs ├── Makefile ├── requirements.txt └── source │ ├── api.rst │ ├── conf.py │ ├── examples.ipynb │ ├── index.rst │ └── readme_link.rst ├── requirements-dev.txt ├── requirements.txt ├── setup.py ├── tests ├── __init__.py ├── test_extractor.py └── test_naming.py └── torchextractor ├── __init__.py ├── extractor.py ├── naming.py └── version.py /.github/workflows/publish.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/antoinebrl/torchextractor/HEAD/.github/workflows/publish.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/antoinebrl/torchextractor/HEAD/.gitignore -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/antoinebrl/torchextractor/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /.readthedocs.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/antoinebrl/torchextractor/HEAD/.readthedocs.yaml -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/antoinebrl/torchextractor/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/antoinebrl/torchextractor/HEAD/README.md -------------------------------------------------------------------------------- /docs/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/antoinebrl/torchextractor/HEAD/docs/Makefile -------------------------------------------------------------------------------- /docs/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/antoinebrl/torchextractor/HEAD/docs/requirements.txt -------------------------------------------------------------------------------- /docs/source/api.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/antoinebrl/torchextractor/HEAD/docs/source/api.rst -------------------------------------------------------------------------------- /docs/source/conf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/antoinebrl/torchextractor/HEAD/docs/source/conf.py -------------------------------------------------------------------------------- /docs/source/examples.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/antoinebrl/torchextractor/HEAD/docs/source/examples.ipynb -------------------------------------------------------------------------------- /docs/source/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/antoinebrl/torchextractor/HEAD/docs/source/index.rst -------------------------------------------------------------------------------- /docs/source/readme_link.rst: -------------------------------------------------------------------------------- 1 | .. mdinclude:: ../../README.md 2 | -------------------------------------------------------------------------------- /requirements-dev.txt: -------------------------------------------------------------------------------- 1 | pre-commit==2.10.1 2 | -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | torch>=1.4.0 2 | numpy 3 | -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/antoinebrl/torchextractor/HEAD/setup.py -------------------------------------------------------------------------------- /tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/test_extractor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/antoinebrl/torchextractor/HEAD/tests/test_extractor.py -------------------------------------------------------------------------------- /tests/test_naming.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/antoinebrl/torchextractor/HEAD/tests/test_naming.py -------------------------------------------------------------------------------- /torchextractor/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/antoinebrl/torchextractor/HEAD/torchextractor/__init__.py -------------------------------------------------------------------------------- /torchextractor/extractor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/antoinebrl/torchextractor/HEAD/torchextractor/extractor.py -------------------------------------------------------------------------------- /torchextractor/naming.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/antoinebrl/torchextractor/HEAD/torchextractor/naming.py -------------------------------------------------------------------------------- /torchextractor/version.py: -------------------------------------------------------------------------------- 1 | __version__ = "0.3.0" 2 | --------------------------------------------------------------------------------