2019-03-25 01:22:43 +08:00
<!DOCTYPE html>
<!-- [if IE 8]><html class="no - js lt - ie9" lang="en" > <![endif] -->
<!-- [if gt IE 8]><! --> < html class = "no-js" lang = "en" > <!-- <![endif] -->
< head >
< meta charset = "utf-8" >
< meta name = "viewport" content = "width=device-width, initial-scale=1.0" >
2019-10-23 23:12:46 +08:00
< title > torchreid.data.datasets.video.mars — torchreid 1.0.5 documentation< / title >
2019-03-25 01:22:43 +08:00
< script type = "text/javascript" src = "../../../../../_static/js/modernizr.min.js" > < / script >
< script type = "text/javascript" id = "documentation_options" data-url_root = "../../../../../" src = "../../../../../_static/documentation_options.js" > < / script >
< script type = "text/javascript" src = "../../../../../_static/jquery.js" > < / script >
< script type = "text/javascript" src = "../../../../../_static/underscore.js" > < / script >
< script type = "text/javascript" src = "../../../../../_static/doctools.js" > < / script >
< script type = "text/javascript" src = "../../../../../_static/language_data.js" > < / script >
< script async = "async" type = "text/javascript" src = "https://cdnjs.cloudflare.com/ajax/libs/mathjax/2.7.5/latest.js?config=TeX-AMS-MML_HTMLorMML" > < / script >
< script type = "text/javascript" src = "../../../../../_static/js/theme.js" > < / script >
< link rel = "stylesheet" href = "../../../../../_static/css/theme.css" type = "text/css" / >
< link rel = "stylesheet" href = "../../../../../_static/pygments.css" type = "text/css" / >
< link rel = "index" title = "Index" href = "../../../../../genindex.html" / >
< link rel = "search" title = "Search" href = "../../../../../search.html" / >
< / head >
< body class = "wy-body-for-nav" >
< div class = "wy-grid-for-nav" >
< nav data-toggle = "wy-nav-shift" class = "wy-nav-side" >
< div class = "wy-side-scroll" >
< div class = "wy-side-nav-search" >
< a href = "../../../../../index.html" class = "icon icon-home" > torchreid
< / a >
< div class = "version" >
2019-10-23 23:12:46 +08:00
1.0.5
2019-03-25 01:22:43 +08:00
< / div >
< div role = "search" >
< form id = "rtd-search-form" class = "wy-form" action = "../../../../../search.html" method = "get" >
< input type = "text" name = "q" placeholder = "Search docs" / >
< input type = "hidden" name = "check_keywords" value = "yes" / >
< input type = "hidden" name = "area" value = "default" / >
< / form >
< / div >
< / div >
< div class = "wy-menu wy-menu-vertical" data-spy = "affix" role = "navigation" aria-label = "main navigation" >
< ul >
< li class = "toctree-l1" > < a class = "reference internal" href = "../../../../../user_guide.html" > How-to< / a > < / li >
< li class = "toctree-l1" > < a class = "reference internal" href = "../../../../../datasets.html" > Datasets< / a > < / li >
< li class = "toctree-l1" > < a class = "reference internal" href = "../../../../../evaluation.html" > Evaluation< / a > < / li >
< / ul >
< p class = "caption" > < span class = "caption-text" > Package Reference< / span > < / p >
< ul >
< li class = "toctree-l1" > < a class = "reference internal" href = "../../../../../pkg/data.html" > torchreid.data< / a > < / li >
< li class = "toctree-l1" > < a class = "reference internal" href = "../../../../../pkg/engine.html" > torchreid.engine< / a > < / li >
< li class = "toctree-l1" > < a class = "reference internal" href = "../../../../../pkg/losses.html" > torchreid.losses< / a > < / li >
< li class = "toctree-l1" > < a class = "reference internal" href = "../../../../../pkg/metrics.html" > torchreid.metrics< / a > < / li >
< li class = "toctree-l1" > < a class = "reference internal" href = "../../../../../pkg/models.html" > torchreid.models< / a > < / li >
< li class = "toctree-l1" > < a class = "reference internal" href = "../../../../../pkg/optim.html" > torchreid.optim< / a > < / li >
< li class = "toctree-l1" > < a class = "reference internal" href = "../../../../../pkg/utils.html" > torchreid.utils< / a > < / li >
< / ul >
< p class = "caption" > < span class = "caption-text" > Resources< / span > < / p >
< ul >
< li class = "toctree-l1" > < a class = "reference internal" href = "../../../../../AWESOME_REID.html" > Awesome-ReID< / a > < / li >
< li class = "toctree-l1" > < a class = "reference internal" href = "../../../../../MODEL_ZOO.html" > Model Zoo< / a > < / li >
< / ul >
< / div >
< / div >
< / nav >
< section data-toggle = "wy-nav-shift" class = "wy-nav-content-wrap" >
< nav class = "wy-nav-top" aria-label = "top navigation" >
< i data-toggle = "wy-nav-top" class = "fa fa-bars" > < / i >
< a href = "../../../../../index.html" > torchreid< / a >
< / nav >
< div class = "wy-nav-content" >
< div class = "rst-content" >
< div role = "navigation" aria-label = "breadcrumbs navigation" >
< ul class = "wy-breadcrumbs" >
< li > < a href = "../../../../../index.html" > Docs< / a > » < / li >
< li > < a href = "../../../../index.html" > Module code< / a > » < / li >
< li > torchreid.data.datasets.video.mars< / li >
< li class = "wy-breadcrumbs-aside" >
< / li >
< / ul >
< hr / >
< / div >
< div role = "main" class = "document" itemscope = "itemscope" itemtype = "http://schema.org/Article" >
< div itemprop = "articleBody" >
< h1 > Source code for torchreid.data.datasets.video.mars< / h1 > < div class = "highlight" > < pre >
< span > < / span > < span class = "kn" > from< / span > < span class = "nn" > __future__< / span > < span class = "k" > import< / span > < span class = "n" > absolute_import< / span >
< span class = "kn" > from< / span > < span class = "nn" > __future__< / span > < span class = "k" > import< / span > < span class = "n" > print_function< / span >
< span class = "kn" > from< / span > < span class = "nn" > __future__< / span > < span class = "k" > import< / span > < span class = "n" > division< / span >
< span class = "kn" > import< / span > < span class = "nn" > os.path< / span > < span class = "k" > as< / span > < span class = "nn" > osp< / span >
< span class = "kn" > from< / span > < span class = "nn" > scipy.io< / span > < span class = "k" > import< / span > < span class = "n" > loadmat< / span >
< span class = "kn" > import< / span > < span class = "nn" > warnings< / span >
< span class = "kn" > from< / span > < span class = "nn" > torchreid.data.datasets< / span > < span class = "k" > import< / span > < span class = "n" > VideoDataset< / span >
< div class = "viewcode-block" id = "Mars" > < a class = "viewcode-back" href = "../../../../../pkg/data.html#torchreid.data.datasets.video.mars.Mars" > [docs]< / a > < span class = "k" > class< / span > < span class = "nc" > Mars< / span > < span class = "p" > (< / span > < span class = "n" > VideoDataset< / span > < span class = "p" > ):< / span >
< span class = "sd" > " " " MARS.< / span >
< span class = "sd" > Reference:< / span >
< span class = "sd" > Zheng et al. MARS: A Video Benchmark for Large-Scale Person Re-identification. ECCV 2016.< / span >
< span class = "sd" > URL: `< http://www.liangzheng.com.cn/Project/project_mars.html> `_< / span >
< span class = "sd" > < / span >
< span class = "sd" > Dataset statistics:< / span >
< span class = "sd" > - identities: 1261.< / span >
< span class = "sd" > - tracklets: 8298 (train) + 1980 (query) + 9330 (gallery).< / span >
< span class = "sd" > - cameras: 6.< / span >
< span class = "sd" > " " " < / span >
< span class = "n" > dataset_dir< / span > < span class = "o" > =< / span > < span class = "s1" > ' mars' < / span >
< span class = "n" > dataset_url< / span > < span class = "o" > =< / span > < span class = "kc" > None< / span >
< span class = "k" > def< / span > < span class = "nf" > __init__< / span > < span class = "p" > (< / span > < span class = "bp" > self< / span > < span class = "p" > ,< / span > < span class = "n" > root< / span > < span class = "o" > =< / span > < span class = "s1" > ' ' < / span > < span class = "p" > ,< / span > < span class = "o" > **< / span > < span class = "n" > kwargs< / span > < span class = "p" > ):< / span >
< span class = "bp" > self< / span > < span class = "o" > .< / span > < span class = "n" > root< / span > < span class = "o" > =< / span > < span class = "n" > osp< / span > < span class = "o" > .< / span > < span class = "n" > abspath< / span > < span class = "p" > (< / span > < span class = "n" > osp< / span > < span class = "o" > .< / span > < span class = "n" > expanduser< / span > < span class = "p" > (< / span > < span class = "n" > root< / span > < span class = "p" > ))< / span >
< span class = "bp" > self< / span > < span class = "o" > .< / span > < span class = "n" > dataset_dir< / span > < span class = "o" > =< / span > < span class = "n" > osp< / span > < span class = "o" > .< / span > < span class = "n" > join< / span > < span class = "p" > (< / span > < span class = "bp" > self< / span > < span class = "o" > .< / span > < span class = "n" > root< / span > < span class = "p" > ,< / span > < span class = "bp" > self< / span > < span class = "o" > .< / span > < span class = "n" > dataset_dir< / span > < span class = "p" > )< / span >
< span class = "bp" > self< / span > < span class = "o" > .< / span > < span class = "n" > download_dataset< / span > < span class = "p" > (< / span > < span class = "bp" > self< / span > < span class = "o" > .< / span > < span class = "n" > dataset_dir< / span > < span class = "p" > ,< / span > < span class = "bp" > self< / span > < span class = "o" > .< / span > < span class = "n" > dataset_url< / span > < span class = "p" > )< / span >
< span class = "bp" > self< / span > < span class = "o" > .< / span > < span class = "n" > train_name_path< / span > < span class = "o" > =< / span > < span class = "n" > osp< / span > < span class = "o" > .< / span > < span class = "n" > join< / span > < span class = "p" > (< / span > < span class = "bp" > self< / span > < span class = "o" > .< / span > < span class = "n" > dataset_dir< / span > < span class = "p" > ,< / span > < span class = "s1" > ' info/train_name.txt' < / span > < span class = "p" > )< / span >
< span class = "bp" > self< / span > < span class = "o" > .< / span > < span class = "n" > test_name_path< / span > < span class = "o" > =< / span > < span class = "n" > osp< / span > < span class = "o" > .< / span > < span class = "n" > join< / span > < span class = "p" > (< / span > < span class = "bp" > self< / span > < span class = "o" > .< / span > < span class = "n" > dataset_dir< / span > < span class = "p" > ,< / span > < span class = "s1" > ' info/test_name.txt' < / span > < span class = "p" > )< / span >
< span class = "bp" > self< / span > < span class = "o" > .< / span > < span class = "n" > track_train_info_path< / span > < span class = "o" > =< / span > < span class = "n" > osp< / span > < span class = "o" > .< / span > < span class = "n" > join< / span > < span class = "p" > (< / span > < span class = "bp" > self< / span > < span class = "o" > .< / span > < span class = "n" > dataset_dir< / span > < span class = "p" > ,< / span > < span class = "s1" > ' info/tracks_train_info.mat' < / span > < span class = "p" > )< / span >
< span class = "bp" > self< / span > < span class = "o" > .< / span > < span class = "n" > track_test_info_path< / span > < span class = "o" > =< / span > < span class = "n" > osp< / span > < span class = "o" > .< / span > < span class = "n" > join< / span > < span class = "p" > (< / span > < span class = "bp" > self< / span > < span class = "o" > .< / span > < span class = "n" > dataset_dir< / span > < span class = "p" > ,< / span > < span class = "s1" > ' info/tracks_test_info.mat' < / span > < span class = "p" > )< / span >
< span class = "bp" > self< / span > < span class = "o" > .< / span > < span class = "n" > query_IDX_path< / span > < span class = "o" > =< / span > < span class = "n" > osp< / span > < span class = "o" > .< / span > < span class = "n" > join< / span > < span class = "p" > (< / span > < span class = "bp" > self< / span > < span class = "o" > .< / span > < span class = "n" > dataset_dir< / span > < span class = "p" > ,< / span > < span class = "s1" > ' info/query_IDX.mat' < / span > < span class = "p" > )< / span >
< span class = "n" > required_files< / span > < span class = "o" > =< / span > < span class = "p" > [< / span >
< span class = "bp" > self< / span > < span class = "o" > .< / span > < span class = "n" > dataset_dir< / span > < span class = "p" > ,< / span >
< span class = "bp" > self< / span > < span class = "o" > .< / span > < span class = "n" > train_name_path< / span > < span class = "p" > ,< / span >
< span class = "bp" > self< / span > < span class = "o" > .< / span > < span class = "n" > test_name_path< / span > < span class = "p" > ,< / span >
< span class = "bp" > self< / span > < span class = "o" > .< / span > < span class = "n" > track_train_info_path< / span > < span class = "p" > ,< / span >
< span class = "bp" > self< / span > < span class = "o" > .< / span > < span class = "n" > track_test_info_path< / span > < span class = "p" > ,< / span >
< span class = "bp" > self< / span > < span class = "o" > .< / span > < span class = "n" > query_IDX_path< / span >
< span class = "p" > ]< / span >
< span class = "bp" > self< / span > < span class = "o" > .< / span > < span class = "n" > check_before_run< / span > < span class = "p" > (< / span > < span class = "n" > required_files< / span > < span class = "p" > )< / span >
< span class = "n" > train_names< / span > < span class = "o" > =< / span > < span class = "bp" > self< / span > < span class = "o" > .< / span > < span class = "n" > get_names< / span > < span class = "p" > (< / span > < span class = "bp" > self< / span > < span class = "o" > .< / span > < span class = "n" > train_name_path< / span > < span class = "p" > )< / span >
< span class = "n" > test_names< / span > < span class = "o" > =< / span > < span class = "bp" > self< / span > < span class = "o" > .< / span > < span class = "n" > get_names< / span > < span class = "p" > (< / span > < span class = "bp" > self< / span > < span class = "o" > .< / span > < span class = "n" > test_name_path< / span > < span class = "p" > )< / span >
< span class = "n" > track_train< / span > < span class = "o" > =< / span > < span class = "n" > loadmat< / span > < span class = "p" > (< / span > < span class = "bp" > self< / span > < span class = "o" > .< / span > < span class = "n" > track_train_info_path< / span > < span class = "p" > )[< / span > < span class = "s1" > ' track_train_info' < / span > < span class = "p" > ]< / span > < span class = "c1" > # numpy.ndarray (8298, 4)< / span >
< span class = "n" > track_test< / span > < span class = "o" > =< / span > < span class = "n" > loadmat< / span > < span class = "p" > (< / span > < span class = "bp" > self< / span > < span class = "o" > .< / span > < span class = "n" > track_test_info_path< / span > < span class = "p" > )[< / span > < span class = "s1" > ' track_test_info' < / span > < span class = "p" > ]< / span > < span class = "c1" > # numpy.ndarray (12180, 4)< / span >
< span class = "n" > query_IDX< / span > < span class = "o" > =< / span > < span class = "n" > loadmat< / span > < span class = "p" > (< / span > < span class = "bp" > self< / span > < span class = "o" > .< / span > < span class = "n" > query_IDX_path< / span > < span class = "p" > )[< / span > < span class = "s1" > ' query_IDX' < / span > < span class = "p" > ]< / span > < span class = "o" > .< / span > < span class = "n" > squeeze< / span > < span class = "p" > ()< / span > < span class = "c1" > # numpy.ndarray (1980,)< / span >
< span class = "n" > query_IDX< / span > < span class = "o" > -=< / span > < span class = "mi" > 1< / span > < span class = "c1" > # index from 0< / span >
< span class = "n" > track_query< / span > < span class = "o" > =< / span > < span class = "n" > track_test< / span > < span class = "p" > [< / span > < span class = "n" > query_IDX< / span > < span class = "p" > ,:]< / span >
< span class = "n" > gallery_IDX< / span > < span class = "o" > =< / span > < span class = "p" > [< / span > < span class = "n" > i< / span > < span class = "k" > for< / span > < span class = "n" > i< / span > < span class = "ow" > in< / span > < span class = "nb" > range< / span > < span class = "p" > (< / span > < span class = "n" > track_test< / span > < span class = "o" > .< / span > < span class = "n" > shape< / span > < span class = "p" > [< / span > < span class = "mi" > 0< / span > < span class = "p" > ])< / span > < span class = "k" > if< / span > < span class = "n" > i< / span > < span class = "ow" > not< / span > < span class = "ow" > in< / span > < span class = "n" > query_IDX< / span > < span class = "p" > ]< / span >
< span class = "n" > track_gallery< / span > < span class = "o" > =< / span > < span class = "n" > track_test< / span > < span class = "p" > [< / span > < span class = "n" > gallery_IDX< / span > < span class = "p" > ,:]< / span >
< span class = "n" > train< / span > < span class = "o" > =< / span > < span class = "bp" > self< / span > < span class = "o" > .< / span > < span class = "n" > process_data< / span > < span class = "p" > (< / span > < span class = "n" > train_names< / span > < span class = "p" > ,< / span > < span class = "n" > track_train< / span > < span class = "p" > ,< / span > < span class = "n" > home_dir< / span > < span class = "o" > =< / span > < span class = "s1" > ' bbox_train' < / span > < span class = "p" > ,< / span > < span class = "n" > relabel< / span > < span class = "o" > =< / span > < span class = "kc" > True< / span > < span class = "p" > )< / span >
< span class = "n" > query< / span > < span class = "o" > =< / span > < span class = "bp" > self< / span > < span class = "o" > .< / span > < span class = "n" > process_data< / span > < span class = "p" > (< / span > < span class = "n" > test_names< / span > < span class = "p" > ,< / span > < span class = "n" > track_query< / span > < span class = "p" > ,< / span > < span class = "n" > home_dir< / span > < span class = "o" > =< / span > < span class = "s1" > ' bbox_test' < / span > < span class = "p" > ,< / span > < span class = "n" > relabel< / span > < span class = "o" > =< / span > < span class = "kc" > False< / span > < span class = "p" > )< / span >
< span class = "n" > gallery< / span > < span class = "o" > =< / span > < span class = "bp" > self< / span > < span class = "o" > .< / span > < span class = "n" > process_data< / span > < span class = "p" > (< / span > < span class = "n" > test_names< / span > < span class = "p" > ,< / span > < span class = "n" > track_gallery< / span > < span class = "p" > ,< / span > < span class = "n" > home_dir< / span > < span class = "o" > =< / span > < span class = "s1" > ' bbox_test' < / span > < span class = "p" > ,< / span > < span class = "n" > relabel< / span > < span class = "o" > =< / span > < span class = "kc" > False< / span > < span class = "p" > )< / span >
< span class = "nb" > super< / span > < span class = "p" > (< / span > < span class = "n" > Mars< / span > < span class = "p" > ,< / span > < span class = "bp" > self< / span > < span class = "p" > )< / span > < span class = "o" > .< / span > < span class = "fm" > __init__< / span > < span class = "p" > (< / span > < span class = "n" > train< / span > < span class = "p" > ,< / span > < span class = "n" > query< / span > < span class = "p" > ,< / span > < span class = "n" > gallery< / span > < span class = "p" > ,< / span > < span class = "o" > **< / span > < span class = "n" > kwargs< / span > < span class = "p" > )< / span >
< span class = "k" > def< / span > < span class = "nf" > get_names< / span > < span class = "p" > (< / span > < span class = "bp" > self< / span > < span class = "p" > ,< / span > < span class = "n" > fpath< / span > < span class = "p" > ):< / span >
< span class = "n" > names< / span > < span class = "o" > =< / span > < span class = "p" > []< / span >
< span class = "k" > with< / span > < span class = "nb" > open< / span > < span class = "p" > (< / span > < span class = "n" > fpath< / span > < span class = "p" > ,< / span > < span class = "s1" > ' r' < / span > < span class = "p" > )< / span > < span class = "k" > as< / span > < span class = "n" > f< / span > < span class = "p" > :< / span >
< span class = "k" > for< / span > < span class = "n" > line< / span > < span class = "ow" > in< / span > < span class = "n" > f< / span > < span class = "p" > :< / span >
< span class = "n" > new_line< / span > < span class = "o" > =< / span > < span class = "n" > line< / span > < span class = "o" > .< / span > < span class = "n" > rstrip< / span > < span class = "p" > ()< / span >
< span class = "n" > names< / span > < span class = "o" > .< / span > < span class = "n" > append< / span > < span class = "p" > (< / span > < span class = "n" > new_line< / span > < span class = "p" > )< / span >
< span class = "k" > return< / span > < span class = "n" > names< / span >
< span class = "k" > def< / span > < span class = "nf" > process_data< / span > < span class = "p" > (< / span > < span class = "bp" > self< / span > < span class = "p" > ,< / span > < span class = "n" > names< / span > < span class = "p" > ,< / span > < span class = "n" > meta_data< / span > < span class = "p" > ,< / span > < span class = "n" > home_dir< / span > < span class = "o" > =< / span > < span class = "kc" > None< / span > < span class = "p" > ,< / span > < span class = "n" > relabel< / span > < span class = "o" > =< / span > < span class = "kc" > False< / span > < span class = "p" > ,< / span > < span class = "n" > min_seq_len< / span > < span class = "o" > =< / span > < span class = "mi" > 0< / span > < span class = "p" > ):< / span >
< span class = "k" > assert< / span > < span class = "n" > home_dir< / span > < span class = "ow" > in< / span > < span class = "p" > [< / span > < span class = "s1" > ' bbox_train' < / span > < span class = "p" > ,< / span > < span class = "s1" > ' bbox_test' < / span > < span class = "p" > ]< / span >
< span class = "n" > num_tracklets< / span > < span class = "o" > =< / span > < span class = "n" > meta_data< / span > < span class = "o" > .< / span > < span class = "n" > shape< / span > < span class = "p" > [< / span > < span class = "mi" > 0< / span > < span class = "p" > ]< / span >
< span class = "n" > pid_list< / span > < span class = "o" > =< / span > < span class = "nb" > list< / span > < span class = "p" > (< / span > < span class = "nb" > set< / span > < span class = "p" > (< / span > < span class = "n" > meta_data< / span > < span class = "p" > [:,< / span > < span class = "mi" > 2< / span > < span class = "p" > ]< / span > < span class = "o" > .< / span > < span class = "n" > tolist< / span > < span class = "p" > ()))< / span >
2019-10-24 00:17:08 +08:00
< span class = "k" > if< / span > < span class = "n" > relabel< / span > < span class = "p" > :< / span >
< span class = "n" > pid2label< / span > < span class = "o" > =< / span > < span class = "p" > {< / span > < span class = "n" > pid< / span > < span class = "p" > :< / span > < span class = "n" > label< / span > < span class = "k" > for< / span > < span class = "n" > label< / span > < span class = "p" > ,< / span > < span class = "n" > pid< / span > < span class = "ow" > in< / span > < span class = "nb" > enumerate< / span > < span class = "p" > (< / span > < span class = "n" > pid_list< / span > < span class = "p" > )}< / span >
2019-03-25 01:22:43 +08:00
< span class = "n" > tracklets< / span > < span class = "o" > =< / span > < span class = "p" > []< / span >
< span class = "k" > for< / span > < span class = "n" > tracklet_idx< / span > < span class = "ow" > in< / span > < span class = "nb" > range< / span > < span class = "p" > (< / span > < span class = "n" > num_tracklets< / span > < span class = "p" > ):< / span >
< span class = "n" > data< / span > < span class = "o" > =< / span > < span class = "n" > meta_data< / span > < span class = "p" > [< / span > < span class = "n" > tracklet_idx< / span > < span class = "p" > ,< / span > < span class = "o" > ...< / span > < span class = "p" > ]< / span >
< span class = "n" > start_index< / span > < span class = "p" > ,< / span > < span class = "n" > end_index< / span > < span class = "p" > ,< / span > < span class = "n" > pid< / span > < span class = "p" > ,< / span > < span class = "n" > camid< / span > < span class = "o" > =< / span > < span class = "n" > data< / span >
< span class = "k" > if< / span > < span class = "n" > pid< / span > < span class = "o" > ==< / span > < span class = "o" > -< / span > < span class = "mi" > 1< / span > < span class = "p" > :< / span >
< span class = "k" > continue< / span > < span class = "c1" > # junk images are just ignored< / span >
< span class = "k" > assert< / span > < span class = "mi" > 1< / span > < span class = "o" > < =< / span > < span class = "n" > camid< / span > < span class = "o" > < =< / span > < span class = "mi" > 6< / span >
< span class = "k" > if< / span > < span class = "n" > relabel< / span > < span class = "p" > :< / span > < span class = "n" > pid< / span > < span class = "o" > =< / span > < span class = "n" > pid2label< / span > < span class = "p" > [< / span > < span class = "n" > pid< / span > < span class = "p" > ]< / span >
< span class = "n" > camid< / span > < span class = "o" > -=< / span > < span class = "mi" > 1< / span > < span class = "c1" > # index starts from 0< / span >
< span class = "n" > img_names< / span > < span class = "o" > =< / span > < span class = "n" > names< / span > < span class = "p" > [< / span > < span class = "n" > start_index< / span > < span class = "o" > -< / span > < span class = "mi" > 1< / span > < span class = "p" > :< / span > < span class = "n" > end_index< / span > < span class = "p" > ]< / span >
< span class = "c1" > # make sure image names correspond to the same person< / span >
< span class = "n" > pnames< / span > < span class = "o" > =< / span > < span class = "p" > [< / span > < span class = "n" > img_name< / span > < span class = "p" > [:< / span > < span class = "mi" > 4< / span > < span class = "p" > ]< / span > < span class = "k" > for< / span > < span class = "n" > img_name< / span > < span class = "ow" > in< / span > < span class = "n" > img_names< / span > < span class = "p" > ]< / span >
< span class = "k" > assert< / span > < span class = "nb" > len< / span > < span class = "p" > (< / span > < span class = "nb" > set< / span > < span class = "p" > (< / span > < span class = "n" > pnames< / span > < span class = "p" > ))< / span > < span class = "o" > ==< / span > < span class = "mi" > 1< / span > < span class = "p" > ,< / span > < span class = "s1" > ' Error: a single tracklet contains different person images' < / span >
< span class = "c1" > # make sure all images are captured under the same camera< / span >
< span class = "n" > camnames< / span > < span class = "o" > =< / span > < span class = "p" > [< / span > < span class = "n" > img_name< / span > < span class = "p" > [< / span > < span class = "mi" > 5< / span > < span class = "p" > ]< / span > < span class = "k" > for< / span > < span class = "n" > img_name< / span > < span class = "ow" > in< / span > < span class = "n" > img_names< / span > < span class = "p" > ]< / span >
< span class = "k" > assert< / span > < span class = "nb" > len< / span > < span class = "p" > (< / span > < span class = "nb" > set< / span > < span class = "p" > (< / span > < span class = "n" > camnames< / span > < span class = "p" > ))< / span > < span class = "o" > ==< / span > < span class = "mi" > 1< / span > < span class = "p" > ,< / span > < span class = "s1" > ' Error: images are captured under different cameras!' < / span >
< span class = "c1" > # append image names with directory information< / span >
< span class = "n" > img_paths< / span > < span class = "o" > =< / span > < span class = "p" > [< / span > < span class = "n" > osp< / span > < span class = "o" > .< / span > < span class = "n" > join< / span > < span class = "p" > (< / span > < span class = "bp" > self< / span > < span class = "o" > .< / span > < span class = "n" > dataset_dir< / span > < span class = "p" > ,< / span > < span class = "n" > home_dir< / span > < span class = "p" > ,< / span > < span class = "n" > img_name< / span > < span class = "p" > [:< / span > < span class = "mi" > 4< / span > < span class = "p" > ],< / span > < span class = "n" > img_name< / span > < span class = "p" > )< / span > < span class = "k" > for< / span > < span class = "n" > img_name< / span > < span class = "ow" > in< / span > < span class = "n" > img_names< / span > < span class = "p" > ]< / span >
< span class = "k" > if< / span > < span class = "nb" > len< / span > < span class = "p" > (< / span > < span class = "n" > img_paths< / span > < span class = "p" > )< / span > < span class = "o" > > =< / span > < span class = "n" > min_seq_len< / span > < span class = "p" > :< / span >
< span class = "n" > img_paths< / span > < span class = "o" > =< / span > < span class = "nb" > tuple< / span > < span class = "p" > (< / span > < span class = "n" > img_paths< / span > < span class = "p" > )< / span >
< span class = "n" > tracklets< / span > < span class = "o" > .< / span > < span class = "n" > append< / span > < span class = "p" > ((< / span > < span class = "n" > img_paths< / span > < span class = "p" > ,< / span > < span class = "n" > pid< / span > < span class = "p" > ,< / span > < span class = "n" > camid< / span > < span class = "p" > ))< / span >
< span class = "k" > return< / span > < span class = "n" > tracklets< / span >
< div class = "viewcode-block" id = "Mars.combine_all" > < a class = "viewcode-back" href = "../../../../../pkg/data.html#torchreid.data.datasets.video.mars.Mars.combine_all" > [docs]< / a > < span class = "k" > def< / span > < span class = "nf" > combine_all< / span > < span class = "p" > (< / span > < span class = "bp" > self< / span > < span class = "p" > ):< / span >
< span class = "n" > warnings< / span > < span class = "o" > .< / span > < span class = "n" > warn< / span > < span class = "p" > (< / span > < span class = "s1" > ' Some query IDs do not appear in gallery. Therefore, combineall ' < / span >
< span class = "s1" > ' does not make any difference to Mars' < / span > < span class = "p" > )< / span > < / div > < / div >
< / pre > < / div >
< / div >
< / div >
< footer >
< hr / >
< div role = "contentinfo" >
< p >
© Copyright 2019, Kaiyang Zhou
< / p >
< / div >
Built with < a href = "http://sphinx-doc.org/" > Sphinx< / a > using a < a href = "https://github.com/rtfd/sphinx_rtd_theme" > theme< / a > provided by < a href = "https://readthedocs.org" > Read the Docs< / a > .
< / footer >
< / div >
< / div >
< / section >
< / div >
< script type = "text/javascript" >
jQuery(function () {
SphinxRtdTheme.Navigation.enable(true);
});
< / script >
< / body >
< / html >
