Index

A

accuracy() (in module torchreid.metrics.accuracy)

AverageMeter (class in torchreid.utils.avgmeter)

B

build_lr_scheduler() (in module torchreid.optim.lr_scheduler) build_model() (in module torchreid.models.__init__)

build_optimizer() (in module torchreid.optim.optimizer) build_train_sampler() (in module torchreid.data.sampler) build_transforms() (in module torchreid.data.transforms)

C

check_before_run() (torchreid.data.datasets.dataset.Dataset method) check_isfile() (in module torchreid.utils.tools) collect_env_info() (in module torchreid.utils.tools) ColorAugmentation (class in torchreid.data.transforms) combine_all() (torchreid.data.datasets.dataset.Dataset method) (torchreid.data.datasets.video.mars.Mars method)

compute_distance_matrix() (in module torchreid.metrics.distance) compute_model_complexity() (in module torchreid.utils.model_complexity) cosine_distance() (in module torchreid.metrics.distance) count_num_param() (in module torchreid.utils.torchtools) CrossEntropyLoss (class in torchreid.losses.cross_entropy_loss) CUHK01 (class in torchreid.data.datasets.image.cuhk01) CUHK03 (class in torchreid.data.datasets.image.cuhk03)

D

DataManager (class in torchreid.data.datamanager) Dataset (class in torchreid.data.datasets.dataset) DenseNet (class in torchreid.models.densenet)

download_dataset() (torchreid.data.datasets.dataset.Dataset method) download_url() (in module torchreid.utils.tools) DukeMTMCreID (class in torchreid.data.datasets.image.dukemtmcreid) DukeMTMCVidReID (class in torchreid.data.datasets.video.dukemtmcvidreid)

E

Engine (class in torchreid.engine.engine)

euclidean_squared_distance() (in module torchreid.metrics.distance) evaluate_rank() (in module torchreid.metrics.rank)

F

fetch_test_loaders() (torchreid.data.datamanager.DataManager method)

forward() (torchreid.losses.cross_entropy_loss.CrossEntropyLoss method) (torchreid.losses.hard_mine_triplet_loss.TripletLoss method)

G

get_num_cams() (torchreid.data.datasets.dataset.Dataset method) get_num_datasets() (torchreid.data.datasets.dataset.Dataset method)

get_num_pids() (torchreid.data.datasets.dataset.Dataset method) GRID (class in torchreid.data.datasets.image.grid)

H

HACNN (class in torchreid.models.hacnn)

I

iLIDS (class in torchreid.data.datasets.image.ilids) iLIDSVID (class in torchreid.data.datasets.video.ilidsvid) ImageDataManager (class in torchreid.data.datamanager) ImageDataset (class in torchreid.data.datasets.dataset) ImageSoftmaxEngine (class in torchreid.engine.image.softmax)

ImageTripletEngine (class in torchreid.engine.image.triplet) InceptionResNetV2 (class in torchreid.models.inceptionresnetv2) InceptionV4 (class in torchreid.models.inceptionv4) init_image_dataset() (in module torchreid.data.datasets.__init__) init_video_dataset() (in module torchreid.data.datasets.__init__)

L

listdir_nohidden() (in module torchreid.utils.tools) load_checkpoint() (in module torchreid.utils.torchtools)

load_pretrained_weights() (in module torchreid.utils.torchtools) Logger (class in torchreid.utils.loggers)

M

Market1501 (class in torchreid.data.datasets.image.market1501) Mars (class in torchreid.data.datasets.video.mars) MetricMeter (class in torchreid.utils.avgmeter) mkdir_if_missing() (in module torchreid.utils.tools)

MLFN (class in torchreid.models.mlfn) MobileNetV2 (class in torchreid.models.mobilenetv2) MSMT17 (class in torchreid.data.datasets.image.msmt17) MuDeep (class in torchreid.models.mudeep)

N

NASNetAMobile (class in torchreid.models.nasnet)

num_train_cams() (torchreid.data.datamanager.DataManager property) num_train_pids() (torchreid.data.datamanager.DataManager property)

O

open_all_layers() (in module torchreid.utils.torchtools) open_specified_layers() (in module torchreid.utils.torchtools)

OSNet (class in torchreid.models.osnet) (class in torchreid.models.osnet_ain)

P

PCB (class in torchreid.models.pcb) prepare_split() (torchreid.data.datasets.image.cuhk01.CUHK01 method)

preprocess_pil_img() (torchreid.data.datamanager.DataManager method) PRID (class in torchreid.data.datasets.image.prid) PRID2011 (class in torchreid.data.datasets.video.prid2011)

R

Random2DTranslation (class in torchreid.data.transforms) RandomDatasetSampler (class in torchreid.data.sampler) RandomDomainSampler (class in torchreid.data.sampler) RandomErasing (class in torchreid.data.transforms) RandomIdentitySampler (class in torchreid.data.sampler) RandomPatch (class in torchreid.data.transforms) RankLogger (class in torchreid.utils.loggers)

read_image() (in module torchreid.utils.tools) read_json() (in module torchreid.utils.tools) register_image_dataset() (in module torchreid.data.datasets.__init__) register_video_dataset() (in module torchreid.data.datasets.__init__) ResNet (class in torchreid.models.resnet) ResNetMid (class in torchreid.models.resnetmid) resume_from_checkpoint() (in module torchreid.utils.torchtools) run() (torchreid.engine.engine.Engine method)

S

save_checkpoint() (in module torchreid.utils.torchtools) SENet (class in torchreid.models.senet) SenseReID (class in torchreid.data.datasets.image.sensereid) show_avai_models() (in module torchreid.models.__init__) show_summary() (torchreid.data.datasets.dataset.Dataset method) (torchreid.data.datasets.dataset.ImageDataset method) (torchreid.data.datasets.dataset.VideoDataset method) (torchreid.utils.loggers.RankLogger method)

ShuffleNet (class in torchreid.models.shufflenet) ShuffleNetV2 (class in torchreid.models.shufflenetv2) SqueezeNet (class in torchreid.models.squeezenet)

T

test() (torchreid.engine.engine.Engine method) torchreid.data.datamanager (module) torchreid.data.datasets.__init__ (module) torchreid.data.datasets.dataset (module) torchreid.data.datasets.image.cuhk01 (module) torchreid.data.datasets.image.cuhk03 (module) torchreid.data.datasets.image.dukemtmcreid (module) torchreid.data.datasets.image.grid (module) torchreid.data.datasets.image.ilids (module) torchreid.data.datasets.image.market1501 (module) torchreid.data.datasets.image.msmt17 (module) torchreid.data.datasets.image.prid (module) torchreid.data.datasets.image.sensereid (module) torchreid.data.datasets.image.viper (module) torchreid.data.datasets.video.dukemtmcvidreid (module) torchreid.data.datasets.video.ilidsvid (module) torchreid.data.datasets.video.mars (module) torchreid.data.datasets.video.prid2011 (module)

torchreid.data.sampler (module) torchreid.data.transforms (module) torchreid.losses.cross_entropy_loss (module) torchreid.losses.hard_mine_triplet_loss (module) torchreid.metrics.accuracy (module) torchreid.metrics.distance (module) torchreid.metrics.rank (module) torchreid.models.__init__ (module) torchreid.optim.lr_scheduler (module) torchreid.optim.optimizer (module) torchreid.utils.avgmeter (module) torchreid.utils.loggers (module) torchreid.utils.model_complexity (module) torchreid.utils.reidtools (module) torchreid.utils.tools (module) torchreid.utils.torchtools (module) TripletLoss (class in torchreid.losses.hard_mine_triplet_loss) two_stepped_transfer_learning() (torchreid.engine.engine.Engine method)

V

VideoDataManager (class in torchreid.data.datamanager) VideoDataset (class in torchreid.data.datasets.dataset) VideoSoftmaxEngine (class in torchreid.engine.video.softmax)

VideoTripletEngine (class in torchreid.engine.video.triplet) VIPeR (class in torchreid.data.datasets.image.viper) visualize_ranked_results() (in module torchreid.utils.reidtools)

W

write() (torchreid.utils.loggers.RankLogger method)

write_json() (in module torchreid.utils.tools)

X

Xception (class in torchreid.models.xception)