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

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_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

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) 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 attribute) num_train_pids (torchreid.data.datamanager.DataManager attribute)

O

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

open_specified_layers() (in module torchreid.utils.torchtools) OSNet (class in torchreid.models.osnet)

P

parse_data() (torchreid.data.datasets.dataset.Dataset method) PCB (class in torchreid.models.pcb)

prepare_split() (torchreid.data.datasets.image.cuhk01.CUHK01 method) PRID (class in torchreid.data.datasets.image.prid) PRID2011 (class in torchreid.data.datasets.video.prid2011)

R

Random2DTranslation (class in torchreid.data.transforms) RandomErasing (class in torchreid.data.transforms) RandomIdentitySampler (class in torchreid.data.sampler) 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) return_dataloaders() (torchreid.data.datamanager.DataManager method) return_testdataset_by_name() (torchreid.data.datamanager.DataManager method) 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) train() (torchreid.engine.engine.Engine method) (torchreid.engine.image.softmax.ImageSoftmaxEngine method) (torchreid.engine.image.triplet.ImageTripletEngine method) TripletLoss (class in torchreid.losses.hard_mine_triplet_loss)

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)