mmr/tkDNN
Archived
1
0
Fork 0
Code Issues 46 Pull Requests 12 Releases 4 Activity

Fixed boxes to float, add conf thresh as param

Browse Source 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
This commit is contained in:
Micaela Verucchi
2020-08-05 19:55:10 +02:00
parent a5d2d4792a
commit f778e1aa99
11 changed files with 22 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files
demo/demo/demo.cpp
+4 -1
View File
@@ -40,6 +40,9 @@ int main(int argc, char *argv[]) {
bool show = true;
if(argc > 6)
show = atoi(argv[6]);
float conf_thresh=0.3;
if(argc > 7)
conf_thresh = atof(argv[7]);
if(n_batch < 1 || n_batch > 64)
FatalError("Batch dim not supported");
@@ -69,7 +72,7 @@ int main(int argc, char *argv[]) {
FatalError("Network type not allowed (3rd parameter)\n");
}
detNN->init(net, n_classes, n_batch);
detNN->init(net, n_classes, n_batch, conf_thresh);
gRun = true;