NVIDIA DeepStream SDK API Reference

7.1 Release
gstnvinferaudio_custom_parser.h
Go to the documentation of this file.
1 /*
2  * SPDX-FileCopyrightText: Copyright (c) 2020 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
3  * SPDX-License-Identifier: LicenseRef-NvidiaProprietary
4  *
5  * NVIDIA CORPORATION, its affiliates and licensors retain all intellectual
6  * property and proprietary rights in and to this material, related
7  * documentation and any modifications thereto. Any use, reproduction,
8  * disclosure or distribution of this material and related documentation
9  * without an express license agreement from NVIDIA CORPORATION or
10  * its affiliates is strictly prohibited.
11  */
12 
13 #pragma once
14 
15 #include <initializer_list>
16 #include <iostream>
17 #include <vector>
18 
19 constexpr unsigned int NB_COARSE_LABEL_AUDIO = 8;
20 
21 constexpr auto LABELS_AUDIO = {
22  "1_engine",
23  "2_machinery-impact",
24  "3_non-machinery-impact",
25  "4_powered-saw",
26  "5_alert-signal",
27  "6_music",
28  "7_human-voice",
29  "8_dog",
30  "1-1_small-sounding-engine",
31  "1-2_medium-sounding-engine",
32  "1-3_large-sounding-engine",
33  "2-1_rock-drill",
34  "2-2_jackhammer",
35  "2-3_hoe-ram",
36  "2-4_pile-driver",
37  "3-1_non-machinery-impact",
38  "4-1_chainsaw",
39  "4-2_small-medium-rotating-saw",
40  "4-3_large-rotating-saw",
41  "5-1_car-horn",
42  "5-2_car-alarm",
43  "5-3_siren",
44  "5-4_reverse-beeper",
45  "6-1_stationary-music",
46  "6-2_mobile-music",
47  "6-3_ice-cream-truck",
48  "7-1_person-or-small-group-talking",
49  "7-2_person-or-small-group-shouting",
50  "7-3_large-crowd",
51  "7-4_amplified-speech",
52  "8-1_dog-barking-whining"};
53 
54 /* This is a sample audio neural network parsing function. */
55 
56 extern "C"
57 {
58  bool NvDsInferParseCustomAudio(std::vector<NvDsInferLayerInfo> const &outputLayersInfo,
59  NvDsInferNetworkInfo const &networkInfo, float classifierThreshold,
60  std::vector<NvDsInferAttribute> &attrList, std::string &attrString);
61 }
62 
63 std::vector<unsigned int> index_giver_subcategory(const char *label);
index_giver_subcategory
std::vector< unsigned int > index_giver_subcategory(const char *label)
NB_COARSE_LABEL_AUDIO
constexpr unsigned int NB_COARSE_LABEL_AUDIO
Definition: gstnvinferaudio_custom_parser.h:19
NvDsInferNetworkInfo
Holds information about the model network.
Definition: nvdsinfer.h:112
NvDsInferParseCustomAudio
bool NvDsInferParseCustomAudio(std::vector< NvDsInferLayerInfo > const &outputLayersInfo, NvDsInferNetworkInfo const &networkInfo, float classifierThreshold, std::vector< NvDsInferAttribute > &attrList, std::string &attrString)
LABELS_AUDIO
constexpr auto LABELS_AUDIO
Definition: gstnvinferaudio_custom_parser.h:21