Skip to content

Commit a0e4e9f

Browse files
committed
Modify yolov4_berkeley link for download, clean to merge with master
Signed-off-by: Micaela Verucchi <[email protected]>
1 parent d936e5f commit a0e4e9f

File tree

8 files changed

+8
-2015
lines changed

8 files changed

+8
-2015
lines changed

CMakeLists.txt

-9
Original file line numberDiff line numberDiff line change
@@ -116,15 +116,6 @@ target_link_libraries(test_yolo3_flir tkDNN)
116116
add_executable(test_yolo4 tests/yolo4/yolo4.cpp)
117117
target_link_libraries(test_yolo4 tkDNN)
118118

119-
add_executable(test_yolo4_320 tests/yolo4/yolo4_320.cpp)
120-
target_link_libraries(test_yolo4_320 tkDNN)
121-
122-
add_executable(test_yolo4_512 tests/yolo4/yolo4_512.cpp)
123-
target_link_libraries(test_yolo4_512 tkDNN)
124-
125-
add_executable(test_yolo4_608 tests/yolo4/yolo4_608.cpp)
126-
target_link_libraries(test_yolo4_608 tkDNN)
127-
128119
add_executable(test_yolo4_berkeley tests/yolo4_berkeley/yolo4_berkeley.cpp)
129120
target_link_libraries(test_yolo4_berkeley tkDNN)
130121

demo/demo/map.cpp

+1-1
Original file line numberDiff line numberDiff line change
@@ -210,7 +210,7 @@ int main(int argc, char *argv[])
210210
}
211211

212212
if(write_coco_json){
213-
coco_json.seekp (coco_json.tellp()-2);
213+
coco_json.seekp (coco_json.tellp() - std::streampos(2));
214214
coco_json << "\n]\n";
215215
coco_json.close();
216216
}

include/tkDNN/DetectionNN.h

+4-4
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@
1414

1515
#include "tkdnn.h"
1616

17-
#define OPENCV_CUDACONTRIB //if OPENCV has been compiled with CUDA and contrib.
17+
// #define OPENCV_CUDACONTRIB //if OPENCV has been compiled with CUDA and contrib.
1818

1919
#ifdef OPENCV_CUDACONTRIB
2020
#include <opencv2/cudawarping.hpp>
@@ -104,7 +104,7 @@ class DetectionNN {
104104
FatalError("A batch size greater than nBatches cannot be used");
105105

106106
originalSize.clear();
107-
if(VERBOSE) printCenteredTitle(" TENSORRT detection ", '=', 30);
107+
if(TKDNN_VERBOSE) printCenteredTitle(" TENSORRT detection ", '=', 30);
108108
{
109109
TIMER_START
110110
for(int bi=0; bi<cur_batches;++bi){
@@ -121,11 +121,11 @@ class DetectionNN {
121121
tk::dnn::dataDim_t dim = netRT->input_dim;
122122
dim.n = cur_batches;
123123
{
124-
if(VERBOSE) dim.print();
124+
if(TKDNN_VERBOSE) dim.print();
125125
TIMER_START
126126
netRT->infer(dim, input_d);
127127
TIMER_STOP
128-
if(VERBOSE) dim.print();
128+
if(TKDNN_VERBOSE) dim.print();
129129
stats.push_back(t_ns);
130130
if(save_times) *times<<t_ns<<";";
131131
}

include/tkDNN/utils.h

+2-2
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,7 @@
3636
#define COL_PURPLEB "\033[1;35m"
3737
#define COL_CYANB "\033[1;36m"
3838

39-
#define VERBOSE 0
39+
#define TKDNN_VERBOSE 0
4040

4141
// Simple Timer
4242
#define TIMER_START timespec start, end; \
@@ -47,7 +47,7 @@
4747
(double)(end.tv_nsec - start.tv_nsec))/1.0e6; \
4848
if(show) std::cout<<col<<"Time:"<<std::setw(16)<<t_ns<<" ms\n"<<COL_END;
4949

50-
#define TIMER_STOP TIMER_STOP_C(COL_CYANB, VERBOSE)
50+
#define TIMER_STOP TIMER_STOP_C(COL_CYANB, TKDNN_VERBOSE)
5151

5252
/********************************************************
5353
* Prints the error message, and exits

0 commit comments

Comments
 (0)