├── LICENSE ├── README.md ├── config ├── __init__.py └── defaults.py ├── configs ├── MARS │ ├── pit-test.yml │ └── pit.yml └── iLIDS-VID │ ├── pit-test.yml │ └── pit.yml ├── datasets ├── __init__.py ├── bases.py ├── ilids.py ├── make_dataloader.py ├── mars.py ├── misc.py ├── preprocessing.py ├── sampler.py ├── sampler_ddp.py └── temporal_transforms.py ├── framework.jpg ├── loss ├── __init__.py ├── arcface.py ├── center_loss.py ├── make_loss.py ├── metric_learning.py ├── softmax_loss.py └── triplet_loss.py ├── model ├── __init__.py ├── backbones │ ├── __init__.py │ ├── __pycache__ │ │ ├── __init__.cpython-36.pyc │ │ ├── resnet.cpython-36.pyc │ │ └── vit_pytorch.cpython-36.pyc │ ├── layers.py │ ├── resnet.py │ └── vit_pytorch.py └── make_model.py ├── processor ├── __init__.py └── processor.py ├── solver ├── __init__.py ├── cosine_lr.py ├── lr_scheduler.py ├── make_optimizer.py ├── scheduler.py └── scheduler_factory.py ├── thop ├── __init__.py ├── fx_profile.py ├── onnx_profile.py ├── profile.py ├── rnn_hooks.py ├── utils.py └── vision │ ├── __init__.py │ ├── basic_hooks.py │ ├── counter.py │ ├── efficientnet.py │ └── onnx_counter.py ├── train.py ├── utils ├── __init__.py ├── iotools.py ├── logger.py ├── meter.py ├── metrics.py ├── reranking.py └── saver.py └── vis.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deropty/PiT/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deropty/PiT/HEAD/README.md -------------------------------------------------------------------------------- /config/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deropty/PiT/HEAD/config/__init__.py -------------------------------------------------------------------------------- /config/defaults.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deropty/PiT/HEAD/config/defaults.py -------------------------------------------------------------------------------- /configs/MARS/pit-test.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deropty/PiT/HEAD/configs/MARS/pit-test.yml -------------------------------------------------------------------------------- /configs/MARS/pit.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deropty/PiT/HEAD/configs/MARS/pit.yml -------------------------------------------------------------------------------- /configs/iLIDS-VID/pit-test.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deropty/PiT/HEAD/configs/iLIDS-VID/pit-test.yml -------------------------------------------------------------------------------- /configs/iLIDS-VID/pit.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deropty/PiT/HEAD/configs/iLIDS-VID/pit.yml -------------------------------------------------------------------------------- /datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deropty/PiT/HEAD/datasets/__init__.py -------------------------------------------------------------------------------- /datasets/bases.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deropty/PiT/HEAD/datasets/bases.py -------------------------------------------------------------------------------- /datasets/ilids.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deropty/PiT/HEAD/datasets/ilids.py -------------------------------------------------------------------------------- /datasets/make_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deropty/PiT/HEAD/datasets/make_dataloader.py -------------------------------------------------------------------------------- /datasets/mars.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deropty/PiT/HEAD/datasets/mars.py -------------------------------------------------------------------------------- /datasets/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deropty/PiT/HEAD/datasets/misc.py -------------------------------------------------------------------------------- /datasets/preprocessing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deropty/PiT/HEAD/datasets/preprocessing.py -------------------------------------------------------------------------------- /datasets/sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deropty/PiT/HEAD/datasets/sampler.py -------------------------------------------------------------------------------- /datasets/sampler_ddp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deropty/PiT/HEAD/datasets/sampler_ddp.py -------------------------------------------------------------------------------- /datasets/temporal_transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deropty/PiT/HEAD/datasets/temporal_transforms.py -------------------------------------------------------------------------------- /framework.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deropty/PiT/HEAD/framework.jpg -------------------------------------------------------------------------------- /loss/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deropty/PiT/HEAD/loss/__init__.py -------------------------------------------------------------------------------- /loss/arcface.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deropty/PiT/HEAD/loss/arcface.py -------------------------------------------------------------------------------- /loss/center_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deropty/PiT/HEAD/loss/center_loss.py -------------------------------------------------------------------------------- /loss/make_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deropty/PiT/HEAD/loss/make_loss.py -------------------------------------------------------------------------------- /loss/metric_learning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deropty/PiT/HEAD/loss/metric_learning.py -------------------------------------------------------------------------------- /loss/softmax_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deropty/PiT/HEAD/loss/softmax_loss.py -------------------------------------------------------------------------------- /loss/triplet_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deropty/PiT/HEAD/loss/triplet_loss.py -------------------------------------------------------------------------------- /model/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deropty/PiT/HEAD/model/__init__.py -------------------------------------------------------------------------------- /model/backbones/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /model/backbones/__pycache__/__init__.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deropty/PiT/HEAD/model/backbones/__pycache__/__init__.cpython-36.pyc -------------------------------------------------------------------------------- /model/backbones/__pycache__/resnet.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deropty/PiT/HEAD/model/backbones/__pycache__/resnet.cpython-36.pyc -------------------------------------------------------------------------------- /model/backbones/__pycache__/vit_pytorch.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deropty/PiT/HEAD/model/backbones/__pycache__/vit_pytorch.cpython-36.pyc -------------------------------------------------------------------------------- /model/backbones/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deropty/PiT/HEAD/model/backbones/layers.py -------------------------------------------------------------------------------- /model/backbones/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deropty/PiT/HEAD/model/backbones/resnet.py -------------------------------------------------------------------------------- /model/backbones/vit_pytorch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deropty/PiT/HEAD/model/backbones/vit_pytorch.py -------------------------------------------------------------------------------- /model/make_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deropty/PiT/HEAD/model/make_model.py -------------------------------------------------------------------------------- /processor/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deropty/PiT/HEAD/processor/__init__.py -------------------------------------------------------------------------------- /processor/processor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deropty/PiT/HEAD/processor/processor.py -------------------------------------------------------------------------------- /solver/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deropty/PiT/HEAD/solver/__init__.py -------------------------------------------------------------------------------- /solver/cosine_lr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deropty/PiT/HEAD/solver/cosine_lr.py -------------------------------------------------------------------------------- /solver/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deropty/PiT/HEAD/solver/lr_scheduler.py -------------------------------------------------------------------------------- /solver/make_optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deropty/PiT/HEAD/solver/make_optimizer.py -------------------------------------------------------------------------------- /solver/scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deropty/PiT/HEAD/solver/scheduler.py -------------------------------------------------------------------------------- /solver/scheduler_factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deropty/PiT/HEAD/solver/scheduler_factory.py -------------------------------------------------------------------------------- /thop/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deropty/PiT/HEAD/thop/__init__.py -------------------------------------------------------------------------------- /thop/fx_profile.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deropty/PiT/HEAD/thop/fx_profile.py -------------------------------------------------------------------------------- /thop/onnx_profile.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deropty/PiT/HEAD/thop/onnx_profile.py -------------------------------------------------------------------------------- /thop/profile.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deropty/PiT/HEAD/thop/profile.py -------------------------------------------------------------------------------- /thop/rnn_hooks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deropty/PiT/HEAD/thop/rnn_hooks.py -------------------------------------------------------------------------------- /thop/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deropty/PiT/HEAD/thop/utils.py -------------------------------------------------------------------------------- /thop/vision/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /thop/vision/basic_hooks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deropty/PiT/HEAD/thop/vision/basic_hooks.py -------------------------------------------------------------------------------- /thop/vision/counter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deropty/PiT/HEAD/thop/vision/counter.py -------------------------------------------------------------------------------- /thop/vision/efficientnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deropty/PiT/HEAD/thop/vision/efficientnet.py -------------------------------------------------------------------------------- /thop/vision/onnx_counter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deropty/PiT/HEAD/thop/vision/onnx_counter.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deropty/PiT/HEAD/train.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/iotools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deropty/PiT/HEAD/utils/iotools.py -------------------------------------------------------------------------------- /utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deropty/PiT/HEAD/utils/logger.py -------------------------------------------------------------------------------- /utils/meter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deropty/PiT/HEAD/utils/meter.py -------------------------------------------------------------------------------- /utils/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deropty/PiT/HEAD/utils/metrics.py -------------------------------------------------------------------------------- /utils/reranking.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deropty/PiT/HEAD/utils/reranking.py -------------------------------------------------------------------------------- /utils/saver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deropty/PiT/HEAD/utils/saver.py -------------------------------------------------------------------------------- /vis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deropty/PiT/HEAD/vis.py --------------------------------------------------------------------------------