docs/html/modelDownloader_8h_source.html
| | Jetson Inference
DNN Vision Library |
modelDownloader.h
Go to the documentation of this file.
1 /*
2 * Copyright (c) 2022, NVIDIA CORPORATION. All rights reserved.
3 *
4 * Permission is hereby granted, free of charge, to any person obtaining a
5 * copy of this software and associated documentation files (the "Software"),
6 * to deal in the Software without restriction, including without limitation
7 * the rights to use, copy, modify, merge, publish, distribute, sublicense,
8 * and/or sell copies of the Software, and to permit persons to whom the
9 * Software is furnished to do so, subject to the following conditions:
10 *
11 * The above copyright notice and this permission notice shall be included in
12 * all copies or substantial portions of the Software.
13 *
14 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
15 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
16 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
17 * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
18 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
19 * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
20 * DEALINGS IN THE SOFTWARE.
21 */
22
23 #ifndef __MODEL_DOWNLOADER_H__
24 #define __MODEL_DOWNLOADER_H__
25
26
27 #include "json.hpp"
28
29
34 bool DownloadModel( nlohmann::json& model, uint32_t retries=2 );
35
36
41 bool DownloadModel( const char* type, const char* name, nlohmann::json& model, uint32_t retries=2 );
42
43
48 bool FindModel( const char* type, const char* name );
49
50
55 bool FindModel( const char* type, const char* name, nlohmann::json& model );
56
57
62 bool FindModel( const char* type, const char* name, nlohmann::json& models, nlohmann::json& model );
63
64
69 bool LoadModelManifest( const char* path="networks/models.json" );
70
71
76 bool LoadModelManifest( nlohmann::json& models, const char* path="networks/models.json" );
77
78
83 #define JSON_STR(x) (x.is_string() ? x.get<std::string>() : "")
84
85
90 #define JSON_STR_DEFAULT(x, default) (x.is_string() ? x.get<std::string>() : default)
91
92
93 #endif
bool LoadModelManifest(const char *path="networks/models.json")
Load the model manifest from a JSON file.
bool FindModel(const char *type, const char *name)
Check if a model can be found in the model manifest.
bool DownloadModel(nlohmann::json &model, uint32_t retries=2)
Download a pre-trained model given its JSON config.