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

Add writing results on file for map demo

Browse Source 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
This commit is contained in:
Micaela Verucchi
2020-03-10 14:35:25 +01:00
parent 0773f216b3
commit 75d5e20843
5 changed files with 71 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files
include/evaluation.h
+3 -3
View File
@@ -52,8 +52,8 @@ void readParams(char* config_filename, int& classes, int& map_points,
float& conf_thresh, bool& verbose);
double computeMap(std::vector<Frame> &images,const int classes,const float IoU_thresh, const float conf_thresh=0.3, const int map_points=101, const bool verbose=false);
double computeMapNIoULevels(std::vector<Frame> &images,const int classes,const float i_IoU_thresh=0.5, const float conf_thresh=0.3, const int map_points=101, const float map_step=0.05, const int map_levels=10, const bool verbose=false);
double computeMapNIoULevels(std::vector<Frame> &images,const int classes,const float i_IoU_thresh=0.5, const float conf_thresh=0.3, const int map_points=101, const float map_step=0.05, const int map_levels=10, const bool verbose=false, const bool write_on_file = false, std::string net = "");
void computeTPFPFN(std::vector<Frame> &images,const int classes,const float IoU_thresh=0.5, const float conf_thresh=0.3, bool verbose=false);
void computeTPFPFN(std::vector<Frame> &images,const int classes,const float IoU_thresh=0.5, const float conf_thresh=0.3, bool verbose=false, const bool write_on_file=false, std::string net="");
#endif /*EVALUATION_H*/
#endif /*EVALUATION_H*/
include/tkDNN/Yolo3Detection.h
+2 -2
View File
@@ -37,7 +37,7 @@ class Yolo3Detection {
int classes = 0;
int num = 0;
int n_masks = 0;
float thresh = 0.3;
float thresh = 0.05;
cv::Scalar colors[256];
// this is filled with results
@@ -70,4 +70,4 @@ class Yolo3Detection {
}}
#endif /* YOLODETECTION_H*/
#endif /* YOLODETECTION_H*/