Register
Login
Resources
Docs Blog Datasets Glossary Case Studies Tutorials & Webinars
Product
Data Engine LLMs Platform Enterprise
Pricing Explore
Connect to our Discord channel
Resources  
Docs Blog Datasets Glossary Case Studies Tutorials & Webinars
Explore Pricing
Product  
Data Engine LLMs Platform Enterprise
Connect to Dagshub Discord channel
Register
Login

Deci-AI
/
super-gradients
connected to https://github.com/Deci-AI/super-gradients.git

Watch 4
Issues 62 Pull Requests 44 Discussions 0

#634 Feature/sg 573 pose estimation

Merged
Ghost merged 1 commits into Deci-AI:master from deci-ai:feature/SG-573-pose-estimation
Conversation 0 Files changed 23 Commits 1
File Compare
super-gradients
/
docs
/
_modules
/
super_gradients
/
common
/
data_interface
/
dataset_data_interface.html
View left file View right file
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172 
  1. <!DOCTYPE html>
    2. 

  2. <html class="writer-html5" lang="en" >
    3. 

  3. <head>
    4. 

  4.   <meta charset="utf-8" />
    5. 

  5.   <meta name="viewport" content="width=device-width, initial-scale=1.0" />
    6. 

  6.   <title>super_gradients.common.data_interface.dataset_data_interface &mdash; SuperGradients 3.0.3 documentation</title>
    7. 

  7.       <link rel="stylesheet" href="../../../../_static/pygments.css" type="text/css" />
    8. 

  8.       <link rel="stylesheet" href="../../../../_static/css/theme.css" type="text/css" />
    9. 

  9.       <link rel="stylesheet" href="../../../../_static/graphviz.css" type="text/css" />
    10. 

  10.       <link rel="stylesheet" href="../../../../_static/custom.css" type="text/css" />
    11. 

  11.   <!--[if lt IE 9]>
    12. 

  12.     <script src="../../../../_static/js/html5shiv.min.js"></script>
    13. 

  13.   <![endif]-->
    14. 

  14.   
    15. 

  15.         <script data-url_root="../../../../" id="documentation_options" src="../../../../_static/documentation_options.js"></script>
    16. 

  16.         <script src="../../../../_static/jquery.js"></script>
    17. 

  17.         <script src="../../../../_static/underscore.js"></script>
    18. 

  18.         <script src="../../../../_static/_sphinx_javascript_frameworks_compat.js"></script>
    19. 

  19.         <script src="../../../../_static/doctools.js"></script>
    20. 

  20.         <script src="../../../../_static/sphinx_highlight.js"></script>
    21. 

  21.     <script src="../../../../_static/js/theme.js"></script>
    22. 

  22.     <link rel="index" title="Index" href="../../../../genindex.html" />
    23. 

  23.     <link rel="search" title="Search" href="../../../../search.html" /> 
    24. 

  24. </head>
    25. 

  25. 

  26. <body class="wy-body-for-nav"> 
    27. 

  27.   <div class="wy-grid-for-nav">
    28. 

  28.     <nav data-toggle="wy-nav-shift" class="wy-nav-side">
    29. 

  29.       <div class="wy-side-scroll">
    30. 

  30.         <div class="wy-side-nav-search" >
    31. 

  31.             <a href="../../../../index.html" class="icon icon-home"> SuperGradients
    32. 

  32.           </a>
    33. 

  33. <div role="search">
    34. 

  34.   <form id="rtd-search-form" class="wy-form" action="../../../../search.html" method="get">
    35. 

  35.     <input type="text" name="q" placeholder="Search docs" />
    36. 

  36.     <input type="hidden" name="check_keywords" value="yes" />
    37. 

  37.     <input type="hidden" name="area" value="default" />
    38. 

  38.   </form>
    39. 

  39. </div>
    40. 

  40.         </div><div class="wy-menu wy-menu-vertical" data-spy="affix" role="navigation" aria-label="Navigation menu">
    41. 

  41.               <p class="caption" role="heading"><span class="caption-text">Welcome To SuperGradients</span></p>
    42. 

  42. <ul>
    43. 

  43. <li class="toctree-l1"><a class="reference internal" href="../../../../welcome.html">Version 3 is out! Notebooks have been updated!</a></li>
    44. 

  44. <li class="toctree-l1"><a class="reference internal" href="../../../../welcome.html#build-with-supergradients">Build with SuperGradients</a></li>
    45. 

  45. <li class="toctree-l1"><a class="reference internal" href="../../../../welcome.html#quick-installation">Quick Installation</a></li>
    46. 

  46. <li class="toctree-l1"><a class="reference internal" href="../../../../welcome.html#what-s-new">What’s New</a></li>
    47. 

  47. <li class="toctree-l1"><a class="reference internal" href="../../../../welcome.html#coming-soon">Coming soon</a></li>
    48. 

  48. <li class="toctree-l1"><a class="reference internal" href="../../../../welcome.html#table-of-content">Table of Content</a></li>
    49. 

  49. <li class="toctree-l1"><a class="reference internal" href="../../../../welcome.html#getting-started">Getting Started</a></li>
    50. 

  50. <li class="toctree-l1"><a class="reference internal" href="../../../../welcome.html#advanced-features">Advanced Features</a></li>
    51. 

  51. <li class="toctree-l1"><a class="reference internal" href="../../../../welcome.html#installation-methods">Installation Methods</a></li>
    52. 

  52. <li class="toctree-l1"><a class="reference internal" href="../../../../welcome.html#implemented-model-architectures">Implemented Model Architectures</a></li>
    53. 

  53. <li class="toctree-l1"><a class="reference internal" href="../../../../welcome.html#documentation">Documentation</a></li>
    54. 

  54. <li class="toctree-l1"><a class="reference internal" href="../../../../welcome.html#contributing">Contributing</a></li>
    55. 

  55. <li class="toctree-l1"><a class="reference internal" href="../../../../welcome.html#citation">Citation</a></li>
    56. 

  56. <li class="toctree-l1"><a class="reference internal" href="../../../../welcome.html#community">Community</a></li>
    57. 

  57. <li class="toctree-l1"><a class="reference internal" href="../../../../welcome.html#license">License</a></li>
    58. 

  58. <li class="toctree-l1"><a class="reference internal" href="../../../../welcome.html#deci-platform">Deci Platform</a></li>
    59. 

  59. </ul>
    60. 

  60. <p class="caption" role="heading"><span class="caption-text">Technical Documentation</span></p>
    61. 

  61. <ul>
    62. 

  62. <li class="toctree-l1"><a class="reference internal" href="../../../../super_gradients.common.html">Common package</a></li>
    63. 

  63. <li class="toctree-l1"><a class="reference internal" href="../../../../super_gradients.training.html">Training package</a></li>
    64. 

  64. </ul>
    65. 

  65. 

  66.         </div>
    67. 

  67.       </div>
    68. 

  68.     </nav>
    69. 

  69. 

  70.     <section data-toggle="wy-nav-shift" class="wy-nav-content-wrap"><nav class="wy-nav-top" aria-label="Mobile navigation menu" >
    71. 

  71.           <i data-toggle="wy-nav-top" class="fa fa-bars"></i>
    72. 

  72.           <a href="../../../../index.html">SuperGradients</a>
    73. 

  73.       </nav>
    74. 

  74. 

  75.       <div class="wy-nav-content">
    76. 

  76.         <div class="rst-content">
    77. 

  77.           <div role="navigation" aria-label="Page navigation">
    78. 

  78.   <ul class="wy-breadcrumbs">
    79. 

  79.       <li><a href="../../../../index.html" class="icon icon-home"></a> &raquo;</li>
    80. 

  80.           <li><a href="../../../index.html">Module code</a> &raquo;</li>
    81. 

  81.       <li>super_gradients.common.data_interface.dataset_data_interface</li>
    82. 

  82.       <li class="wy-breadcrumbs-aside">
    83. 

  83.       </li>
    84. 

  84.   </ul>
    85. 

  85.   <hr/>
    86. 

  86. </div>
    87. 

  87.           <div role="main" class="document" itemscope="itemscope" itemtype="http://schema.org/Article">
    88. 

  88.            <div itemprop="articleBody">
    89. 

  89.              
    90. 

  90.   <h1>Source code for super_gradients.common.data_interface.dataset_data_interface</h1><div class="highlight"><pre>
    91. 

  91. <span></span><span class="kn">import</span> <span class="nn">os</span>
    92. 

  92. <span class="kn">from</span> <span class="nn">super_gradients.common</span> <span class="kn">import</span> <span class="n">S3Connector</span>
    93. 

  93. <span class="kn">from</span> <span class="nn">super_gradients.common</span> <span class="kn">import</span> <span class="n">explicit_params_validation</span>
    94. 

  94. <span class="kn">import</span> <span class="nn">zipfile</span>
    95. 

  95. 

  96. 

  97. <div class="viewcode-block" id="DatasetDataInterface"><a class="viewcode-back" href="../../../../super_gradients.common.html#super_gradients.common.DatasetDataInterface">[docs]</a><span class="k">class</span> <span class="nc">DatasetDataInterface</span><span class="p">:</span>
    98. 

  98. 

  99.     <span class="k">def</span> <span class="fm">__init__</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">env</span><span class="p">:</span> <span class="nb">str</span><span class="p">,</span> <span class="n">data_connection_source</span><span class="p">:</span> <span class="nb">str</span> <span class="o">=</span> <span class="s1">&#39;s3&#39;</span><span class="p">):</span>
    100. 

  100.         <span class="sd">&quot;&quot;&quot;</span>
    101. 

  101. 

  102. <span class="sd">        :param env: str &quot;development&quot;/&quot;production&quot;</span>
    103. 

  103. <span class="sd">        :param data_connection_source: str &quot;s3&quot; for aws bny default</span>
    104. 

  104. <span class="sd">        &quot;&quot;&quot;</span>
    105. 

  105.         <span class="bp">self</span><span class="o">.</span><span class="n">env</span> <span class="o">=</span> <span class="n">env</span>
    106. 

  106.         <span class="bp">self</span><span class="o">.</span><span class="n">s3_connector</span> <span class="o">=</span> <span class="kc">None</span>
    107. 

  107.         <span class="bp">self</span><span class="o">.</span><span class="n">data_connection_source</span> <span class="o">=</span> <span class="n">data_connection_source</span>
    108. 

  108. 

  109.     <span class="nd">@explicit_params_validation</span><span class="p">(</span><span class="n">validation_type</span><span class="o">=</span><span class="s1">&#39;None&#39;</span><span class="p">)</span>
    110. 

  110.     <span class="k">def</span> <span class="nf">load_remote_dataset_file</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">remote_file</span><span class="p">:</span> <span class="nb">str</span><span class="p">,</span> <span class="n">local_dir</span><span class="p">:</span> <span class="nb">str</span><span class="p">,</span> <span class="n">overwrite_local_dataset</span><span class="p">:</span> <span class="nb">bool</span> <span class="o">=</span> <span class="kc">False</span><span class="p">)</span> <span class="o">-&gt;</span> <span class="nb">str</span><span class="p">:</span>
    111. 

  111.         <span class="sd">&quot;&quot;&quot;</span>
    112. 

  112. 

  113. <span class="sd">        :param remote_file: str - the name of s3 file</span>
    114. 

  114. <span class="sd">        :param local_dir: str - the directory to put the dataset in</span>
    115. 

  115. <span class="sd">        :param overwrite_local_dataset: Whether too  delete the dataset dir before downloading</span>
    116. 

  116. <span class="sd">        :return:</span>
    117. 

  117. <span class="sd">        &quot;&quot;&quot;</span>
    118. 

  118. 

  119.         <span class="n">dataset_full_path</span> <span class="o">=</span> <span class="n">local_dir</span>
    120. 

  120.         <span class="n">bucket</span> <span class="o">=</span> <span class="n">remote_file</span><span class="o">.</span><span class="n">split</span><span class="p">(</span><span class="s2">&quot;/&quot;</span><span class="p">)[</span><span class="mi">2</span><span class="p">]</span>
    121. 

  121.         <span class="n">file_path</span> <span class="o">=</span> <span class="s2">&quot;/&quot;</span><span class="o">.</span><span class="n">join</span><span class="p">(</span><span class="n">remote_file</span><span class="o">.</span><span class="n">split</span><span class="p">(</span><span class="s2">&quot;/&quot;</span><span class="p">)[</span><span class="mi">3</span><span class="p">:])</span>
    122. 

  122.         <span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">data_connection_source</span> <span class="o">==</span> <span class="s1">&#39;s3&#39;</span><span class="p">:</span>
    123. 

  123.             <span class="bp">self</span><span class="o">.</span><span class="n">s3_connector</span> <span class="o">=</span> <span class="n">S3Connector</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">env</span><span class="p">,</span> <span class="n">bucket</span><span class="p">)</span>
    124. 

  124. 

  125.             <span class="c1"># DELETE THE LOCAL VERSION ON THE MACHINE</span>
    126. 

  126.             <span class="k">if</span> <span class="n">os</span><span class="o">.</span><span class="n">path</span><span class="o">.</span><span class="n">exists</span><span class="p">(</span><span class="n">dataset_full_path</span><span class="p">):</span>
    127. 

  127.                 <span class="k">if</span> <span class="n">overwrite_local_dataset</span><span class="p">:</span>
    128. 

  128. 

  129.                     <span class="n">filelist</span> <span class="o">=</span> <span class="n">os</span><span class="o">.</span><span class="n">listdir</span><span class="p">(</span><span class="n">local_dir</span><span class="p">)</span>
    130. 

  130.                     <span class="k">for</span> <span class="n">f</span> <span class="ow">in</span> <span class="n">filelist</span><span class="p">:</span>
    131. 

  131.                         <span class="n">os</span><span class="o">.</span><span class="n">remove</span><span class="p">(</span><span class="n">os</span><span class="o">.</span><span class="n">path</span><span class="o">.</span><span class="n">join</span><span class="p">(</span><span class="n">local_dir</span><span class="p">,</span> <span class="n">f</span><span class="p">))</span>
    132. 

  132.                 <span class="k">else</span><span class="p">:</span>
    133. 

  133.                     <span class="ne">Warning</span><span class="p">(</span><span class="s2">&quot;Overwrite local dataset set to False but dataset exists in the dir&quot;</span><span class="p">)</span>
    134. 

  134.             <span class="k">if</span> <span class="ow">not</span> <span class="n">os</span><span class="o">.</span><span class="n">path</span><span class="o">.</span><span class="n">exists</span><span class="p">(</span><span class="n">local_dir</span><span class="p">):</span>
    135. 

  135.                 <span class="n">os</span><span class="o">.</span><span class="n">mkdir</span><span class="p">(</span><span class="n">local_dir</span><span class="p">)</span>
    136. 

  136. 

  137.             <span class="n">local_file</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">s3_connector</span><span class="o">.</span><span class="n">download_file_by_path</span><span class="p">(</span><span class="n">file_path</span><span class="p">,</span> <span class="n">local_dir</span><span class="p">)</span>
    138. 

  138.             <span class="k">with</span> <span class="n">zipfile</span><span class="o">.</span><span class="n">ZipFile</span><span class="p">(</span><span class="n">local_dir</span> <span class="o">+</span> <span class="s2">&quot;/&quot;</span> <span class="o">+</span> <span class="n">local_file</span><span class="p">,</span> <span class="s1">&#39;r&#39;</span><span class="p">)</span> <span class="k">as</span> <span class="n">zip_ref</span><span class="p">:</span>
    139. 

  139.                 <span class="n">zip_ref</span><span class="o">.</span><span class="n">extractall</span><span class="p">(</span><span class="n">local_dir</span> <span class="o">+</span> <span class="s2">&quot;/&quot;</span><span class="p">)</span>
    140. 

  140.             <span class="n">os</span><span class="o">.</span><span class="n">remove</span><span class="p">(</span><span class="n">local_dir</span> <span class="o">+</span> <span class="s2">&quot;/&quot;</span> <span class="o">+</span> <span class="n">local_file</span><span class="p">)</span>
    141. 

  141. 

  142.         <span class="k">return</span> <span class="n">local_dir</span></div>
    143. 

  143. </pre></div>
    144. 

  144. 

  145.            </div>
    146. 

  146.           </div>
    147. 

  147.           <footer>
    148. 

  148. 

  149.   <hr/>
    150. 

  150. 

  151.   <div role="contentinfo">
    152. 

  152.     <p>&#169; Copyright 2021, SuperGradients team.</p>
    153. 

  153.   </div>
    154. 

  154. 

  155.   Built with <a href="https://www.sphinx-doc.org/">Sphinx</a> using a
    156. 

  156.     <a href="https://github.com/readthedocs/sphinx_rtd_theme">theme</a>
    157. 

  157.     provided by <a href="https://readthedocs.org">Read the Docs</a>.
    158. 

  158.    
    159. 

  159. 

  160. </footer>
    161. 

  161.         </div>
    162. 

  162.       </div>
    163. 

  163.     </section>
    164. 

  164.   </div>
    165. 

  165.   <script>
    166. 

  166.       jQuery(function () {
    167. 

  167.           SphinxRtdTheme.Navigation.enable(true);
    168. 

  168.       });
    169. 

  169.   </script> 
    170. 

  170. 

  171. </body>
    172. 

  172. </html>
    173.
Discard
Prev
Next
Tip!

Press p or to see the previous file or, n or to see the next file