文件名称：TransReID:TransReID
文件大小：802KB
文件格式：ZIP
更新时间：2024-04-21 11:19:34
Python
TransReID：基于变压器的对象重新标识 的官方存储库实现了对象重新ID的最新性能，包括人员重新ID和车辆重新ID。 管道 基于变压器的强基准的合理化研究 要求 安装 pip install -r requirements.txt (we use /torch 1.6.0 /torchvision 0.7.0 /timm 0.3.2 /cuda 10.1 / 16G or 32G V100 for training and evaluation. Note that we use torch.cuda.amp to accelerate speed of training which requires pytorch > =1.6) 准备数据集 mkdir data 下载人员数据集 ， ， ， 以及车辆数据集 ， ，然后将其解压缩并在以下目录下重命名 data ├── mar
【文件预览】：
TransReID-main
----processor()
--------processor.py(7KB)
--------__init__.py(45B)
----loss()
--------center_loss.py(2KB)
--------softmax_loss.py(2KB)
--------triplet_loss.py(5KB)
--------__init__.py(61B)
--------arcface.py(3KB)
--------metric_learning.py(7KB)
--------make_loss.py(4KB)
----train.py(3KB)
----utils()
--------meter.py(455B)
--------reranking.py(4KB)
--------__init__.py(0B)
--------metrics.py(5KB)
--------iotools.py(775B)
--------logger.py(815B)
----configs()
--------MSMT17()
--------transformer_base.yml(1KB)
--------OCC_Duke()
--------Market()
--------DukeMTMC()
----test.py(2KB)
----model()
--------__init__.py(34B)
--------make_model.py(18KB)
--------backbones()
----requirements.txt(41B)
----datasets()
--------occ_duke.py(4KB)
--------msmt17.py(3KB)
--------dukemtmcreid.py(4KB)
--------market1501.py(3KB)
--------preprocessing.py(2KB)
--------__init__.py(44B)
--------make_dataloader.py(4KB)
--------sampler.py(2KB)
--------sampler_ddp.py(7KB)
--------bases.py(3KB)
----config()
--------defaults.py(6KB)
--------__init__.py(150B)
----LICENSE(1KB)
----README.md(9KB)
----solver()
--------make_optimizer.py(1KB)
--------__init__.py(86B)
--------scheduler.py(5KB)
--------cosine_lr.py(4KB)
--------scheduler_factory.py(1KB)
--------lr_scheduler.py(2KB)
----.gitignore(4KB)
----figs()
--------framework.png(373KB)
--------ablation.png(149KB)
--------sota.png(209KB)