diff --git a/.gitignore b/.gitignore index d62c96d..b56526f 100644 --- a/.gitignore +++ b/.gitignore @@ -13,4 +13,4 @@ build/ *.pk *.table demo/COCO_val2017 -demo/BDD100k_val \ No newline at end of file +demo/BDD100K_val \ No newline at end of file diff --git a/README.md b/README.md index fdfc877..1175465 100644 --- a/README.md +++ b/README.md @@ -2,9 +2,18 @@ tkDNN is a Deep Neural Network library built with cuDNN and tensorRT primitives, specifically thought to work on NVIDIA Jetson Boards. It has been tested on TK1(branch cudnn2), TX1, TX2, AGX Xavier and several discrete GPU. The main goal of this project is to exploit NVIDIA boards as much as possible to obtain the best inference performance. It does not allow training. + +If you use tkDNN in your research, please cite one of the following papers. For use in commercial solutions, write at gattifrancesco@hotmail.it or refer to https://hipert.unimore.it/ . + +``` Accepted paper @ IRC 2020, will soon been published. M. Verucchi, L. Bartoli, F. Bagni, F. Gatti, P. Burgio and M. Bertogna, "Real-Time clustering and LiDAR-camera fusion on embedded platforms for self-driving cars", in proceedings in IEEE Robotic Computing (2020) +Accepted paper @ ETFA 2020, will soon been published. +M. Verucchi, G. Brilli, D. Sapienza, M. Verasani, M. Arena, F. Gatti, A. Capotondi, R. Cavicchioli, M. Bertogna, M. Solieri +"A Systematic Assessment of Embedded Neural Networks for Object Detection", in IEEE International Conference on Emerging Technologies and Factory Automation (2020) +``` + ## Index - [tkDNN](#tkdnn) - [Index](#index) diff --git a/demo/demo/map.cpp b/demo/demo/map.cpp index 60aa7fa..d724db0 100644 --- a/demo/demo/map.cpp +++ b/demo/demo/map.cpp @@ -153,7 +153,7 @@ int main(int argc, char *argv[]) std::ofstream myfile; if(write_dets) - myfile.open ("det/"+f.lFilename.substr(f.lFilename.find("000"))); + myfile.open ("det/"+f.lFilename.substr(f.lFilename.find("labels/") + 7)); // save detections labels for(auto d:detected_bbox){ @@ -169,7 +169,7 @@ int main(int argc, char *argv[]) f.det.push_back(b); if(write_dets) - myfile << d.cl << " "<< d.prob << " "<< d.x << " "<< d.y << " "<< d.w << " "<< d.h <<"\n"; + myfile << d.cl << " "<< d.prob << " "<< b.x << " "<< b.y << " "<< b.w << " "<< b.h <<"\n"; if(show)// draw rectangle for detection cv::rectangle(batch_frames[0], cv::Point(d.x, d.y), cv::Point(d.x + d.w, d.y + d.h), cv::Scalar(0, 0, 255), 2); diff --git a/scripts/install_OpenCV4.sh b/scripts/install_OpenCV4.sh index 8862cdc..f57c87b 100644 --- a/scripts/install_OpenCV4.sh +++ b/scripts/install_OpenCV4.sh @@ -62,5 +62,5 @@ make -j4 sudo make install sudo ldconfig -cd '~/Downloads/opencv4/lib/python3.6/site-packages' +cd ~/Downloads/opencv4/lib/python3.6/site-packages ln -s /usr/local/lib/python3.6/site-packages/cv2.cpython-36m-aarch64-linux-gnu.so cv2.so