2019-03-25 01:22:43 +08:00
<!DOCTYPE html>
2020-08-12 19:12:06 +08:00
< html class = "writer-html5" lang = "en" >
2019-03-25 01:22:43 +08:00
< head >
< meta charset = "utf-8" >
< meta name = "viewport" content = "width=device-width, initial-scale=1.0" >
2021-04-27 15:32:14 +08:00
< title > torchreid.utils.reidtools — torchreid 1.4.0 documentation< / title >
2019-03-25 01:22:43 +08:00
2020-08-12 19:12:06 +08:00
< link rel = "stylesheet" href = "../../../_static/css/theme.css" type = "text/css" / >
< link rel = "stylesheet" href = "../../../_static/pygments.css" type = "text/css" / >
2019-03-25 01:22:43 +08:00
2020-08-12 19:12:06 +08:00
<!-- [if lt IE 9]>
< script src = "../../../_static/js/html5shiv.min.js" > < / script >
<![endif]-->
2019-03-25 01:22:43 +08:00
< 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 = "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" >
2020-08-12 19:12:06 +08:00
< a href = "../../../index.html" class = "icon icon-home" alt = "Documentation Home" > torchreid
2019-03-25 01:22:43 +08:00
< / a >
< div class = "version" >
2021-04-27 15:32:14 +08:00
1.4.0
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 >
2020-08-12 19:12:06 +08:00
2019-03-25 01:22:43 +08:00
< 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 >
2020-08-12 19:12:06 +08:00
2019-03-25 01:22:43 +08:00
< / 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" >
2020-08-12 19:12:06 +08:00
< li > < a href = "../../../index.html" class = "icon icon-home" > < / a > » < / li >
2019-03-25 01:22:43 +08:00
< li > < a href = "../../index.html" > Module code< / a > » < / li >
< li > torchreid.utils.reidtools< / 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.utils.reidtools< / h1 > < div class = "highlight" > < pre >
2020-08-12 19:12:06 +08:00
< span > < / span > < span class = "kn" > from< / span > < span class = "nn" > __future__< / span > < span class = "kn" > import< / span > < span class = "n" > print_function< / span > < span class = "p" > ,< / span > < span class = "n" > absolute_import< / span >
2019-03-25 01:22:43 +08:00
< span class = "kn" > import< / span > < span class = "nn" > numpy< / span > < span class = "k" > as< / span > < span class = "nn" > np< / span >
< span class = "kn" > import< / span > < span class = "nn" > shutil< / span >
2019-12-01 11:34:42 +08:00
< span class = "kn" > import< / span > < span class = "nn" > os.path< / span > < span class = "k" > as< / span > < span class = "nn" > osp< / span >
2019-08-04 06:16:36 +08:00
< span class = "kn" > import< / span > < span class = "nn" > cv2< / span >
2019-03-25 01:22:43 +08:00
2020-08-12 19:12:06 +08:00
< span class = "kn" > from< / span > < span class = "nn" > .tools< / span > < span class = "kn" > import< / span > < span class = "n" > mkdir_if_missing< / span >
2019-03-25 01:22:43 +08:00
2019-12-01 11:34:42 +08:00
< span class = "n" > __all__< / span > < span class = "o" > =< / span > < span class = "p" > [< / span > < span class = "s1" > ' visualize_ranked_results' < / span > < span class = "p" > ]< / span >
2019-03-25 01:22:43 +08:00
2019-08-04 06:16:36 +08:00
< span class = "n" > GRID_SPACING< / span > < span class = "o" > =< / span > < span class = "mi" > 10< / span >
< span class = "n" > QUERY_EXTRA_SPACING< / span > < span class = "o" > =< / span > < span class = "mi" > 90< / span >
< span class = "n" > BW< / span > < span class = "o" > =< / span > < span class = "mi" > 5< / span > < span class = "c1" > # border width< / span >
< span class = "n" > GREEN< / span > < span class = "o" > =< / span > < span class = "p" > (< / span > < span class = "mi" > 0< / span > < span class = "p" > ,< / span > < span class = "mi" > 255< / span > < span class = "p" > ,< / span > < span class = "mi" > 0< / span > < span class = "p" > )< / span >
< span class = "n" > RED< / span > < span class = "o" > =< / span > < span class = "p" > (< / span > < span class = "mi" > 0< / span > < span class = "p" > ,< / span > < span class = "mi" > 0< / span > < span class = "p" > ,< / span > < span class = "mi" > 255< / span > < span class = "p" > )< / span >
2019-12-01 11:34:42 +08:00
< div class = "viewcode-block" id = "visualize_ranked_results" > < a class = "viewcode-back" href = "../../../pkg/utils.html#torchreid.utils.reidtools.visualize_ranked_results" > [docs]< / a > < span class = "k" > def< / span > < span class = "nf" > visualize_ranked_results< / span > < span class = "p" > (< / span >
< span class = "n" > distmat< / span > < span class = "p" > ,< / span > < span class = "n" > dataset< / span > < span class = "p" > ,< / span > < span class = "n" > data_type< / span > < span class = "p" > ,< / span > < span class = "n" > width< / span > < span class = "o" > =< / span > < span class = "mi" > 128< / span > < span class = "p" > ,< / span > < span class = "n" > height< / span > < span class = "o" > =< / span > < span class = "mi" > 256< / span > < span class = "p" > ,< / span > < span class = "n" > save_dir< / span > < span class = "o" > =< / span > < span class = "s1" > ' ' < / span > < span class = "p" > ,< / span > < span class = "n" > topk< / span > < span class = "o" > =< / span > < span class = "mi" > 10< / span >
< span class = "p" > ):< / span >
2019-03-25 01:22:43 +08:00
< span class = "sd" > " " " Visualizes ranked results.< / span >
< span class = "sd" > Supports both image-reid and video-reid.< / span >
2019-08-04 06:16:36 +08:00
< span class = "sd" > For image-reid, ranks will be plotted in a single figure. For video-reid, ranks will be< / span >
< span class = "sd" > saved in folders each containing a tracklet.< / span >
2019-03-25 01:22:43 +08:00
< span class = "sd" > Args:< / span >
< span class = "sd" > distmat (numpy.ndarray): distance matrix of shape (num_query, num_gallery).< / span >
< span class = "sd" > dataset (tuple): a 2-tuple containing (query, gallery), each of which contains< / span >
2020-08-19 05:18:22 +08:00
< span class = "sd" > tuples of (img_path(s), pid, camid, dsetid).< / span >
2019-08-04 06:16:36 +08:00
< span class = "sd" > data_type (str): " image" or " video" .< / span >
< span class = "sd" > width (int, optional): resized image width. Default is 128.< / span >
< span class = "sd" > height (int, optional): resized image height. Default is 256.< / span >
2019-03-25 01:22:43 +08:00
< span class = "sd" > save_dir (str): directory to save output images.< / span >
< span class = "sd" > topk (int, optional): denoting top-k images in the rank list to be visualized.< / span >
2019-08-04 06:16:36 +08:00
< span class = "sd" > Default is 10.< / span >
2019-03-25 01:22:43 +08:00
< span class = "sd" > " " " < / span >
< span class = "n" > num_q< / span > < span class = "p" > ,< / span > < span class = "n" > num_g< / span > < span class = "o" > =< / span > < span class = "n" > distmat< / span > < span class = "o" > .< / span > < span class = "n" > shape< / span >
2019-08-04 06:16:36 +08:00
< span class = "n" > mkdir_if_missing< / span > < span class = "p" > (< / span > < span class = "n" > save_dir< / span > < span class = "p" > )< / span >
2019-03-25 01:22:43 +08:00
< span class = "nb" > print< / span > < span class = "p" > (< / span > < span class = "s1" > ' # query: < / span > < span class = "si" > {}< / span > < span class = "se" > \n< / span > < span class = "s1" > # gallery < / span > < span class = "si" > {}< / span > < span class = "s1" > ' < / span > < span class = "o" > .< / span > < span class = "n" > format< / span > < span class = "p" > (< / span > < span class = "n" > num_q< / span > < span class = "p" > ,< / span > < span class = "n" > num_g< / span > < span class = "p" > ))< / span >
2019-08-04 06:16:36 +08:00
< span class = "nb" > print< / span > < span class = "p" > (< / span > < span class = "s1" > ' Visualizing top-< / span > < span class = "si" > {}< / span > < span class = "s1" > ranks ...' < / span > < span class = "o" > .< / span > < span class = "n" > format< / span > < span class = "p" > (< / span > < span class = "n" > topk< / span > < span class = "p" > ))< / span >
2019-12-01 11:34:42 +08:00
2019-03-25 01:22:43 +08:00
< span class = "n" > query< / span > < span class = "p" > ,< / span > < span class = "n" > gallery< / span > < span class = "o" > =< / span > < span class = "n" > dataset< / span >
< span class = "k" > assert< / span > < span class = "n" > num_q< / span > < span class = "o" > ==< / span > < span class = "nb" > len< / span > < span class = "p" > (< / span > < span class = "n" > query< / span > < span class = "p" > )< / span >
< span class = "k" > assert< / span > < span class = "n" > num_g< / span > < span class = "o" > ==< / span > < span class = "nb" > len< / span > < span class = "p" > (< / span > < span class = "n" > gallery< / span > < span class = "p" > )< / span >
2019-12-01 11:34:42 +08:00
2019-03-25 01:22:43 +08:00
< span class = "n" > indices< / span > < span class = "o" > =< / span > < span class = "n" > np< / span > < span class = "o" > .< / span > < span class = "n" > argsort< / span > < span class = "p" > (< / span > < span class = "n" > distmat< / span > < span class = "p" > ,< / span > < span class = "n" > axis< / span > < span class = "o" > =< / span > < span class = "mi" > 1< / span > < span class = "p" > )< / span >
2019-08-04 06:16:36 +08:00
< span class = "k" > def< / span > < span class = "nf" > _cp_img_to< / span > < span class = "p" > (< / span > < span class = "n" > src< / span > < span class = "p" > ,< / span > < span class = "n" > dst< / span > < span class = "p" > ,< / span > < span class = "n" > rank< / span > < span class = "p" > ,< / span > < span class = "n" > prefix< / span > < span class = "p" > ,< / span > < span class = "n" > matched< / span > < span class = "o" > =< / span > < span class = "kc" > False< / span > < span class = "p" > ):< / span >
2019-03-25 01:22:43 +08:00
< span class = "sd" > " " " < / span >
< span class = "sd" > Args:< / span >
< span class = "sd" > src: image path or tuple (for vidreid)< / span >
< span class = "sd" > dst: target directory< / span >
< span class = "sd" > rank: int, denoting ranked position, starting from 1< / span >
< span class = "sd" > prefix: string< / span >
2019-08-04 06:16:36 +08:00
< span class = "sd" > matched: bool< / span >
2019-03-25 01:22:43 +08:00
< span class = "sd" > " " " < / span >
2019-08-24 05:37:55 +08:00
< span class = "k" > if< / span > < span class = "nb" > isinstance< / span > < span class = "p" > (< / span > < span class = "n" > src< / span > < span class = "p" > ,< / span > < span class = "p" > (< / span > < span class = "nb" > tuple< / span > < span class = "p" > ,< / span > < span class = "nb" > list< / span > < span class = "p" > )):< / span >
2019-08-04 06:16:36 +08:00
< span class = "k" > if< / span > < span class = "n" > prefix< / span > < span class = "o" > ==< / span > < span class = "s1" > ' gallery' < / span > < span class = "p" > :< / span >
< span class = "n" > suffix< / span > < span class = "o" > =< / span > < span class = "s1" > ' TRUE' < / span > < span class = "k" > if< / span > < span class = "n" > matched< / span > < span class = "k" > else< / span > < span class = "s1" > ' FALSE' < / span >
2019-12-01 11:34:42 +08:00
< span class = "n" > dst< / 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 = "n" > dst< / span > < span class = "p" > ,< / span > < span class = "n" > prefix< / span > < span class = "o" > +< / span > < span class = "s1" > ' _top' < / span > < span class = "o" > +< / span > < span class = "nb" > str< / span > < span class = "p" > (< / span > < span class = "n" > rank< / span > < span class = "p" > )< / span > < span class = "o" > .< / span > < span class = "n" > zfill< / span > < span class = "p" > (< / span > < span class = "mi" > 3< / span > < span class = "p" > )< / span >
< span class = "p" > )< / span > < span class = "o" > +< / span > < span class = "s1" > ' _' < / span > < span class = "o" > +< / span > < span class = "n" > suffix< / span >
2019-08-04 06:16:36 +08:00
< span class = "k" > else< / span > < span class = "p" > :< / span >
< span class = "n" > dst< / 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 = "n" > dst< / span > < span class = "p" > ,< / span > < span class = "n" > prefix< / span > < span class = "o" > +< / span > < span class = "s1" > ' _top' < / span > < span class = "o" > +< / span > < span class = "nb" > str< / span > < span class = "p" > (< / span > < span class = "n" > rank< / span > < span class = "p" > )< / span > < span class = "o" > .< / span > < span class = "n" > zfill< / span > < span class = "p" > (< / span > < span class = "mi" > 3< / span > < span class = "p" > ))< / span >
2019-03-25 01:22:43 +08:00
< span class = "n" > mkdir_if_missing< / span > < span class = "p" > (< / span > < span class = "n" > dst< / span > < span class = "p" > )< / span >
< span class = "k" > for< / span > < span class = "n" > img_path< / span > < span class = "ow" > in< / span > < span class = "n" > src< / span > < span class = "p" > :< / span >
< span class = "n" > shutil< / span > < span class = "o" > .< / span > < span class = "n" > copy< / span > < span class = "p" > (< / span > < span class = "n" > img_path< / span > < span class = "p" > ,< / span > < span class = "n" > dst< / span > < span class = "p" > )< / span >
< span class = "k" > else< / span > < span class = "p" > :< / span >
2019-12-01 11:34:42 +08:00
< span class = "n" > dst< / 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 = "n" > dst< / span > < span class = "p" > ,< / span > < span class = "n" > prefix< / span > < span class = "o" > +< / span > < span class = "s1" > ' _top' < / span > < span class = "o" > +< / span > < span class = "nb" > str< / span > < span class = "p" > (< / span > < span class = "n" > rank< / span > < span class = "p" > )< / span > < span class = "o" > .< / span > < span class = "n" > zfill< / span > < span class = "p" > (< / span > < span class = "mi" > 3< / span > < span class = "p" > )< / span > < span class = "o" > +< / span > < span class = "s1" > ' _name_' < / span > < span class = "o" > +< / span >
< span class = "n" > osp< / span > < span class = "o" > .< / span > < span class = "n" > basename< / span > < span class = "p" > (< / span > < span class = "n" > src< / span > < span class = "p" > )< / span >
< span class = "p" > )< / span >
2019-03-25 01:22:43 +08:00
< span class = "n" > shutil< / span > < span class = "o" > .< / span > < span class = "n" > copy< / span > < span class = "p" > (< / span > < span class = "n" > src< / span > < span class = "p" > ,< / span > < span class = "n" > dst< / span > < span class = "p" > )< / span >
< span class = "k" > for< / span > < span class = "n" > q_idx< / span > < span class = "ow" > in< / span > < span class = "nb" > range< / span > < span class = "p" > (< / span > < span class = "n" > num_q< / span > < span class = "p" > ):< / span >
2020-08-19 05:18:22 +08:00
< span class = "n" > qimg_path< / span > < span class = "p" > ,< / span > < span class = "n" > qpid< / span > < span class = "p" > ,< / span > < span class = "n" > qcamid< / span > < span class = "o" > =< / span > < span class = "n" > query< / span > < span class = "p" > [< / span > < span class = "n" > q_idx< / span > < span class = "p" > ][:< / span > < span class = "mi" > 3< / span > < span class = "p" > ]< / span >
2019-12-01 11:34:42 +08:00
< span class = "n" > qimg_path_name< / span > < span class = "o" > =< / span > < span class = "n" > qimg_path< / span > < span class = "p" > [< / span > < span class = "mi" > 0< / span > < span class = "p" > ]< / span > < span class = "k" > if< / span > < span class = "nb" > isinstance< / span > < span class = "p" > (< / span >
< span class = "n" > qimg_path< / span > < span class = "p" > ,< / span > < span class = "p" > (< / span > < span class = "nb" > tuple< / span > < span class = "p" > ,< / span > < span class = "nb" > list< / span > < span class = "p" > )< / span >
< span class = "p" > )< / span > < span class = "k" > else< / span > < span class = "n" > qimg_path< / span >
2019-08-04 06:16:36 +08:00
< span class = "k" > if< / span > < span class = "n" > data_type< / span > < span class = "o" > ==< / span > < span class = "s1" > ' image' < / span > < span class = "p" > :< / span >
< span class = "n" > qimg< / span > < span class = "o" > =< / span > < span class = "n" > cv2< / span > < span class = "o" > .< / span > < span class = "n" > imread< / span > < span class = "p" > (< / span > < span class = "n" > qimg_path< / span > < span class = "p" > )< / span >
< span class = "n" > qimg< / span > < span class = "o" > =< / span > < span class = "n" > cv2< / span > < span class = "o" > .< / span > < span class = "n" > resize< / span > < span class = "p" > (< / span > < span class = "n" > qimg< / span > < span class = "p" > ,< / span > < span class = "p" > (< / span > < span class = "n" > width< / span > < span class = "p" > ,< / span > < span class = "n" > height< / span > < span class = "p" > ))< / span >
2019-12-01 11:34:42 +08:00
< span class = "n" > qimg< / span > < span class = "o" > =< / span > < span class = "n" > cv2< / span > < span class = "o" > .< / span > < span class = "n" > copyMakeBorder< / span > < span class = "p" > (< / span >
< span class = "n" > qimg< / span > < span class = "p" > ,< / span > < span class = "n" > BW< / span > < span class = "p" > ,< / span > < span class = "n" > BW< / span > < span class = "p" > ,< / span > < span class = "n" > BW< / span > < span class = "p" > ,< / span > < span class = "n" > BW< / span > < span class = "p" > ,< / span > < span class = "n" > cv2< / span > < span class = "o" > .< / span > < span class = "n" > BORDER_CONSTANT< / span > < span class = "p" > ,< / span > < span class = "n" > value< / span > < span class = "o" > =< / span > < span class = "p" > (< / span > < span class = "mi" > 0< / span > < span class = "p" > ,< / span > < span class = "mi" > 0< / span > < span class = "p" > ,< / span > < span class = "mi" > 0< / span > < span class = "p" > )< / span >
< span class = "p" > )< / span >
2019-08-24 05:06:05 +08:00
< span class = "c1" > # resize twice to ensure that the border width is consistent across images< / span >
< span class = "n" > qimg< / span > < span class = "o" > =< / span > < span class = "n" > cv2< / span > < span class = "o" > .< / span > < span class = "n" > resize< / span > < span class = "p" > (< / span > < span class = "n" > qimg< / span > < span class = "p" > ,< / span > < span class = "p" > (< / span > < span class = "n" > width< / span > < span class = "p" > ,< / span > < span class = "n" > height< / span > < span class = "p" > ))< / span >
< span class = "n" > num_cols< / span > < span class = "o" > =< / span > < span class = "n" > topk< / span > < span class = "o" > +< / span > < span class = "mi" > 1< / span >
2019-12-01 11:34:42 +08:00
< span class = "n" > grid_img< / span > < span class = "o" > =< / span > < span class = "mi" > 255< / span > < span class = "o" > *< / span > < span class = "n" > np< / span > < span class = "o" > .< / span > < span class = "n" > ones< / span > < span class = "p" > (< / span >
< span class = "p" > (< / span >
< span class = "n" > height< / span > < span class = "p" > ,< / span >
< span class = "n" > num_cols< / span > < span class = "o" > *< / span > < span class = "n" > width< / span > < span class = "o" > +< / span > < span class = "n" > topk< / span > < span class = "o" > *< / span > < span class = "n" > GRID_SPACING< / span > < span class = "o" > +< / span > < span class = "n" > QUERY_EXTRA_SPACING< / span > < span class = "p" > ,< / span > < span class = "mi" > 3< / span >
< span class = "p" > ),< / span >
< span class = "n" > dtype< / span > < span class = "o" > =< / span > < span class = "n" > np< / span > < span class = "o" > .< / span > < span class = "n" > uint8< / span >
< span class = "p" > )< / span >
2019-08-04 06:16:36 +08:00
< span class = "n" > grid_img< / span > < span class = "p" > [:,< / span > < span class = "p" > :< / span > < span class = "n" > width< / span > < span class = "p" > ,< / span > < span class = "p" > :]< / span > < span class = "o" > =< / span > < span class = "n" > qimg< / span >
2019-03-25 01:22:43 +08:00
< span class = "k" > else< / span > < span class = "p" > :< / span >
2019-12-01 11:34:42 +08:00
< span class = "n" > qdir< / 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 = "n" > save_dir< / span > < span class = "p" > ,< / span > < span class = "n" > osp< / span > < span class = "o" > .< / span > < span class = "n" > basename< / span > < span class = "p" > (< / span > < span class = "n" > osp< / span > < span class = "o" > .< / span > < span class = "n" > splitext< / span > < span class = "p" > (< / span > < span class = "n" > qimg_path_name< / span > < span class = "p" > )[< / span > < span class = "mi" > 0< / span > < span class = "p" > ])< / span >
< span class = "p" > )< / span >
2019-08-04 06:16:36 +08:00
< span class = "n" > mkdir_if_missing< / span > < span class = "p" > (< / span > < span class = "n" > qdir< / span > < span class = "p" > )< / span >
< span class = "n" > _cp_img_to< / span > < span class = "p" > (< / span > < span class = "n" > qimg_path< / span > < span class = "p" > ,< / span > < span class = "n" > qdir< / span > < span class = "p" > ,< / span > < span class = "n" > rank< / span > < span class = "o" > =< / span > < span class = "mi" > 0< / span > < span class = "p" > ,< / span > < span class = "n" > prefix< / span > < span class = "o" > =< / span > < span class = "s1" > ' query' < / span > < span class = "p" > )< / span >
2019-03-25 01:22:43 +08:00
< span class = "n" > rank_idx< / span > < span class = "o" > =< / span > < span class = "mi" > 1< / span >
2019-12-01 11:34:42 +08:00
< span class = "k" > for< / span > < span class = "n" > g_idx< / span > < span class = "ow" > in< / span > < span class = "n" > indices< / span > < span class = "p" > [< / span > < span class = "n" > q_idx< / span > < span class = "p" > ,< / span > < span class = "p" > :]:< / span >
2020-08-19 05:18:22 +08:00
< span class = "n" > gimg_path< / span > < span class = "p" > ,< / span > < span class = "n" > gpid< / span > < span class = "p" > ,< / span > < span class = "n" > gcamid< / span > < span class = "o" > =< / span > < span class = "n" > gallery< / span > < span class = "p" > [< / span > < span class = "n" > g_idx< / span > < span class = "p" > ][:< / span > < span class = "mi" > 3< / span > < span class = "p" > ]< / span >
2019-03-25 01:22:43 +08:00
< span class = "n" > invalid< / span > < span class = "o" > =< / span > < span class = "p" > (< / span > < span class = "n" > qpid< / span > < span class = "o" > ==< / span > < span class = "n" > gpid< / span > < span class = "p" > )< / span > < span class = "o" > & < / span > < span class = "p" > (< / span > < span class = "n" > qcamid< / span > < span class = "o" > ==< / span > < span class = "n" > gcamid< / span > < span class = "p" > )< / span >
2019-12-01 11:34:42 +08:00
2019-03-25 01:22:43 +08:00
< span class = "k" > if< / span > < span class = "ow" > not< / span > < span class = "n" > invalid< / span > < span class = "p" > :< / span >
2019-12-01 11:34:42 +08:00
< span class = "n" > matched< / span > < span class = "o" > =< / span > < span class = "n" > gpid< / span > < span class = "o" > ==< / span > < span class = "n" > qpid< / span >
2019-08-04 06:16:36 +08:00
< span class = "k" > if< / span > < span class = "n" > data_type< / span > < span class = "o" > ==< / span > < span class = "s1" > ' image' < / span > < span class = "p" > :< / span >
< span class = "n" > border_color< / span > < span class = "o" > =< / span > < span class = "n" > GREEN< / span > < span class = "k" > if< / span > < span class = "n" > matched< / span > < span class = "k" > else< / span > < span class = "n" > RED< / span >
< span class = "n" > gimg< / span > < span class = "o" > =< / span > < span class = "n" > cv2< / span > < span class = "o" > .< / span > < span class = "n" > imread< / span > < span class = "p" > (< / span > < span class = "n" > gimg_path< / span > < span class = "p" > )< / span >
< span class = "n" > gimg< / span > < span class = "o" > =< / span > < span class = "n" > cv2< / span > < span class = "o" > .< / span > < span class = "n" > resize< / span > < span class = "p" > (< / span > < span class = "n" > gimg< / span > < span class = "p" > ,< / span > < span class = "p" > (< / span > < span class = "n" > width< / span > < span class = "p" > ,< / span > < span class = "n" > height< / span > < span class = "p" > ))< / span >
2019-12-01 11:34:42 +08:00
< span class = "n" > gimg< / span > < span class = "o" > =< / span > < span class = "n" > cv2< / span > < span class = "o" > .< / span > < span class = "n" > copyMakeBorder< / span > < span class = "p" > (< / span >
< span class = "n" > gimg< / span > < span class = "p" > ,< / span >
< span class = "n" > BW< / span > < span class = "p" > ,< / span >
< span class = "n" > BW< / span > < span class = "p" > ,< / span >
< span class = "n" > BW< / span > < span class = "p" > ,< / span >
< span class = "n" > BW< / span > < span class = "p" > ,< / span >
< span class = "n" > cv2< / span > < span class = "o" > .< / span > < span class = "n" > BORDER_CONSTANT< / span > < span class = "p" > ,< / span >
< span class = "n" > value< / span > < span class = "o" > =< / span > < span class = "n" > border_color< / span >
< span class = "p" > )< / span >
2019-08-04 06:16:36 +08:00
< span class = "n" > gimg< / span > < span class = "o" > =< / span > < span class = "n" > cv2< / span > < span class = "o" > .< / span > < span class = "n" > resize< / span > < span class = "p" > (< / span > < span class = "n" > gimg< / span > < span class = "p" > ,< / span > < span class = "p" > (< / span > < span class = "n" > width< / span > < span class = "p" > ,< / span > < span class = "n" > height< / span > < span class = "p" > ))< / span >
< span class = "n" > start< / span > < span class = "o" > =< / span > < span class = "n" > rank_idx< / span > < span class = "o" > *< / span > < span class = "n" > width< / span > < span class = "o" > +< / span > < span class = "n" > rank_idx< / span > < span class = "o" > *< / span > < span class = "n" > GRID_SPACING< / span > < span class = "o" > +< / span > < span class = "n" > QUERY_EXTRA_SPACING< / span >
2019-12-01 11:34:42 +08:00
< span class = "n" > end< / span > < span class = "o" > =< / span > < span class = "p" > (< / span >
< span class = "n" > rank_idx< / span > < span class = "o" > +< / span > < span class = "mi" > 1< / span >
< span class = "p" > )< / span > < span class = "o" > *< / span > < span class = "n" > width< / span > < span class = "o" > +< / span > < span class = "n" > rank_idx< / span > < span class = "o" > *< / span > < span class = "n" > GRID_SPACING< / span > < span class = "o" > +< / span > < span class = "n" > QUERY_EXTRA_SPACING< / span >
< span class = "n" > grid_img< / span > < span class = "p" > [:,< / span > < span class = "n" > start< / span > < span class = "p" > :< / span > < span class = "n" > end< / span > < span class = "p" > ,< / span > < span class = "p" > :]< / span > < span class = "o" > =< / span > < span class = "n" > gimg< / span >
2019-08-04 06:16:36 +08:00
< span class = "k" > else< / span > < span class = "p" > :< / span >
2019-12-01 11:34:42 +08:00
< span class = "n" > _cp_img_to< / span > < span class = "p" > (< / span >
< span class = "n" > gimg_path< / span > < span class = "p" > ,< / span >
< span class = "n" > qdir< / span > < span class = "p" > ,< / span >
< span class = "n" > rank< / span > < span class = "o" > =< / span > < span class = "n" > rank_idx< / span > < span class = "p" > ,< / span >
< span class = "n" > prefix< / span > < span class = "o" > =< / span > < span class = "s1" > ' gallery' < / span > < span class = "p" > ,< / span >
< span class = "n" > matched< / span > < span class = "o" > =< / span > < span class = "n" > matched< / span >
< span class = "p" > )< / span >
2019-03-25 01:22:43 +08:00
< span class = "n" > rank_idx< / span > < span class = "o" > +=< / span > < span class = "mi" > 1< / span >
< span class = "k" > if< / span > < span class = "n" > rank_idx< / span > < span class = "o" > > < / span > < span class = "n" > topk< / span > < span class = "p" > :< / span >
< span class = "k" > break< / span >
2019-08-24 05:06:05 +08:00
< span class = "k" > if< / span > < span class = "n" > data_type< / span > < span class = "o" > ==< / span > < span class = "s1" > ' image' < / span > < span class = "p" > :< / span >
< span class = "n" > imname< / span > < span class = "o" > =< / span > < span class = "n" > osp< / span > < span class = "o" > .< / span > < span class = "n" > basename< / span > < span class = "p" > (< / span > < span class = "n" > osp< / span > < span class = "o" > .< / span > < span class = "n" > splitext< / span > < span class = "p" > (< / span > < span class = "n" > qimg_path_name< / span > < span class = "p" > )[< / span > < span class = "mi" > 0< / span > < span class = "p" > ])< / span >
2019-12-01 11:34:42 +08:00
< span class = "n" > cv2< / span > < span class = "o" > .< / span > < span class = "n" > imwrite< / 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 = "n" > save_dir< / span > < span class = "p" > ,< / span > < span class = "n" > imname< / span > < span class = "o" > +< / span > < span class = "s1" > ' .jpg' < / span > < span class = "p" > ),< / span > < span class = "n" > grid_img< / span > < span class = "p" > )< / span >
2019-08-04 06:16:36 +08:00
< span class = "k" > if< / span > < span class = "p" > (< / span > < span class = "n" > q_idx< / span > < span class = "o" > +< / span > < span class = "mi" > 1< / span > < span class = "p" > )< / span > < span class = "o" > %< / span > < span class = "mi" > 100< / span > < span class = "o" > ==< / span > < span class = "mi" > 0< / span > < span class = "p" > :< / span >
2019-12-01 11:34:42 +08:00
< span class = "nb" > print< / span > < span class = "p" > (< / span > < span class = "s1" > ' - done < / span > < span class = "si" > {}< / span > < span class = "s1" > /< / span > < span class = "si" > {}< / span > < span class = "s1" > ' < / span > < span class = "o" > .< / span > < span class = "n" > format< / span > < span class = "p" > (< / span > < span class = "n" > q_idx< / span > < span class = "o" > +< / span > < span class = "mi" > 1< / span > < span class = "p" > ,< / span > < span class = "n" > num_q< / span > < span class = "p" > ))< / span >
2019-08-04 06:16:36 +08:00
< span class = "nb" > print< / span > < span class = "p" > (< / span > < span class = "s1" > ' Done. Images have been saved to " < / span > < span class = "si" > {}< / span > < span class = "s1" > " ...' < / span > < span class = "o" > .< / span > < span class = "n" > format< / span > < span class = "p" > (< / span > < span class = "n" > save_dir< / span > < span class = "p" > ))< / span > < / div >
2019-03-25 01:22:43 +08:00
< / pre > < / div >
< / div >
< / div >
< footer >
< hr / >
< div role = "contentinfo" >
< p >
2020-08-12 19:12:06 +08:00
2019-03-25 01:22:43 +08:00
© Copyright 2019, Kaiyang Zhou
< / p >
< / div >
2020-08-12 19:12:06 +08:00
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 > .
2019-03-25 01:22:43 +08:00
< / footer >
< / div >
< / div >
< / section >
< / div >
< script type = "text/javascript" >
jQuery(function () {
SphinxRtdTheme.Navigation.enable(true);
});
< / script >
< / body >
< / html >
