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

Add Yolov3 (COCO80) and Yolov3-tiny (COCO80), TensorRT for tiny not working

Browse Source 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
This commit is contained in:
Micaela Verucchi
2019-12-04 17:01:40 +00:00
parent 2594f59d0d
commit a2db98670a
9 changed files with 1222 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/Yolo.cpp
+5 -3
View File
@@ -15,14 +15,16 @@ Yolo::Yolo(Network *net, int classes, int num, std::string fname_weights) :
Layer(net) {
this->classes = classes;
this->num = num;
this->num = 3;
// load anchors
if(fname_weights != "") {
int seek = 0;
readBinaryFile(fname_weights, num, &mask_h, &mask_d, seek);
seek += num;
readBinaryFile(fname_weights, 3, &mask_h, &mask_d, seek);
seek += 3;
readBinaryFile(fname_weights, 3*num*2, &bias_h, &bias_d, seek);
for(int i=0; i<3*num*2; i++)
printf("%f\n", bias_h[i]);
}
// init default classes name