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

fix formatting

Browse Source
This commit is contained in:
Moreno Razzoli
2021-06-21 09:20:19 +02:00
parent 2782f22155
commit f5d6f18c82
1 changed files with 11 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files
demo/demo/demo.cpp
+11 -11
View File
@@ -104,7 +104,7 @@ int main(int argc, char *argv[]) {
std::vector<cv::Mat> batch_dnn_input;
if(save_detections)
outfile.open("detections.csv", std::ofstream::out);
outfile.open("detections.csv", std::ofstream::out);
int frame_num = -1;
@@ -129,17 +129,17 @@ int main(int argc, char *argv[]) {
detNN->update(batch_dnn_input, n_batch);
detNN->draw(batch_frame);
if(save_detections){
frame_num++;
tk::dnn::box b;
for(int bi=0; bi<batch_frame.size(); ++bi){
for(int i=0; i < detNN->batchDetected[bi].size(); i++) {
b = detNN->batchDetected[bi][i];
std::string det_class = detNN->classesNames[b.cl];
outfile<<frame_num<<";"<<det_class<<";"<<b.x<<";"<<b.y<<";"<<b.x+b.w<<";"<<b.y+b.h<<"\n";
if(save_detections){
frame_num++;
tk::dnn::box b;
for(int bi=0; bi<batch_frame.size(); ++bi){
for(int i=0; i < detNN->batchDetected[bi].size(); i++) {
b = detNN->batchDetected[bi][i];
std::string det_class = detNN->classesNames[b.cl];
outfile<<frame_num<<";"<<det_class<<";"<<b.x<<";"<<b.y<<";"<<b.x+b.w<<";"<<b.y+b.h<<"\n";
}
}
}
}
}
if(show){
for(int bi=0; bi< n_batch; ++bi){