mmr/tkDNN
Archived
1
0
Fork 0
Code Issues 46 Pull Requests 12 Releases 4 Activity
619 Commits 11 Branches 5 Tags
a37c14d71dfca0833b60c211e51eae1010a983b1
Commit Graph

233 Commits

Author SHA1 Message Date
AdamCuellar a3d607a210 Add yolov4-csp-swish 2022-04-06 12:44:23 -04:00
Francesco Gatti df31375b67 Merge branch 'master' into tensorrt8 2022-03-30 16:41:45 +02:00
Francesco Gatti 7c0620e391 test_all_test script save results in separate files 2022-03-30 15:47:41 +02:00
perseusdg 480b5a9c5a added monodepth2_1024 2022-02-22 14:50:39 +00:00
Harshvardhan Chandirasekar 3e86671c50 Moved batchnorm and test_monodepth2_new_format layer to dev 
Signed-off-by: perseusdg <harshvardhan.chandira@gmail.com>
 2022-01-24 21:41:59 +05:30
Micaela Verucchi decd73d298 Fix saving result video in demoDepth, add automatic download for monodepth2 weights 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2022-01-19 10:22:06 -08:00
Micaela Verucchi 907df27e07 Fix monodepth2, add demoDepth: 
- Fix monodepth2 network, now works with both cuDNN and tensorRT
- Substitute cuDNN ELU with tkDNN one
- add DepthNN class
- add demoDepth demo, now only works with monodepth2 net

Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
               Francesco Gatti <gattifrancesco@hotmail.it>
 2022-01-18 21:34:16 -08:00
perseusdg 19e41a8b99 monodepth2 new format (test) conv + independent batch norm 2022-01-11 12:38:36 +05:30
perseusdg cabebc95a2 added individual layer names in monodepth2 2022-01-09 13:45:44 +05:30
perseusdg 7298dcfb2f Added monodepth2.cpp 2022-01-06 16:30:11 +00:00
Micaela Verucchi 04de9908a6 Add yolo4-csp for crowdhuman dataset, add shelfnet for coco-stuff dataset, fix minor in demo.cpp 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2021-11-29 17:33:57 +01:00
perseusdg ee5000ccca bug fixes for dla networks and ported optimization from different pull request 2021-11-13 02:00:49 +05:30
perseusdg c0e2097397 - CMakeLists.txt opencv cuda contrib autodetect 
- Updated Docker to cuda-11.3+cudnn-8.2.1+TensorRT-8.0.34,Ubuntu to 20.04 and OpenCV to 4.5.4
- Updated OpenCV4 to 4.5.4 in install_OpenCV4.sh
- Updated README.md
 2021-11-10 12:30:57 +05:30
perseusdg c5e66c6bf6 TRT8 works with almost every nerual network now!!!!(including demo3d) 2021-10-28 23:35:37 +05:30
Micaela Verucchi eca10ac0a8 Update weights 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2021-09-19 00:16:31 +02:00
perseusdg 03473743c4 tkDNN works with trt8!!,need to test int8 and mobilenet,dla_cnet (fps seems to be a bit low 350 on trt8 compared to 396 on trt7) 2021-09-07 22:00:28 +05:30
perseusdg 504dee5016 test_yolo4 netRT->destroy() 2021-09-06 15:35:57 +05:30
perseusdg de83ae5d25 update tensorrt8 branch 2021-08-30 19:04:26 +05:30
perseusdg 2ffe07057e Mnist works at the moment with trt8,others like yolo4tiny and mobilenet generate the engine files but crash after throwing nvifer1::CudaRuntimeError and when demo is being run ,it doesnt deserialize properly and crashes 2021-08-29 03:18:58 +05:30
perseusdg 3d8b1ac494 Yolo3Detection.cpp doesn't build yet,issues with dependecies of preprocessing and postprocessing on IPluginFactory 2021-08-21 22:29:36 +05:30
Davide Sapienza 0292e21c33 Merge remote-tracking branch 'origin/master' into cnet 2021-07-22 17:03:23 +02:00
Davide Sapienza be2d361ac6 Update 3d detection 
This commit updates the CenternetDetection3d class and fixes
some bugs.

It also removes the resnet101 network for CenterNet 3D.
This network doesn't exist.

Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2021-07-21 15:51:13 +02:00
Micaela Verucchi f4e71a5a28 Add yolo4_berkeley_f1 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2021-07-21 12:56:00 +02:00
Micaela Verucchi 89b1bb7bee Add missing files 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2021-07-20 18:39:25 +02:00
Micaela Verucchi 318dfe4b55 merge with master 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2021-07-20 18:38:46 +02:00
Micaela Verucchi 84ff978ccb Merge with master, all tests passed 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2021-07-20 12:48:41 +02:00
Micaela Verucchi b3c44a86a8 Fix shelfnet berkeley, update readme, minors 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2021-07-20 10:57:53 +02:00
Davide Sapienza 34c1c3d577 Update cnet branch. 
This commit splits the demo3D in two demo: one for the 3D object
detection and one for the tracking.

It renames the files related to CenterTrack.

It adds a new parameter to select the tracker mode (2D or 3D).

Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2021-05-11 16:17:23 +02:00
Davide Sapienza 28fab9c3e1 Merge remote-tracking branch 'origin/master' into cnet 2021-04-27 14:43:39 +02:00
Francesco Gatti 6611a91201 merge perseusdg windows10 support 2021-04-24 18:55:26 +02:00
hchandirasekar f12ec3c935 added yolo4x and yolo4-csp from the github repo and download file corrections 2021-03-26 12:18:46 +05:30
Micaela Verucchi 4b3731928c Add yolo4_320_coco2 (pedestrian and stop sign) 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2021-02-25 09:57:10 +01:00
Micaela Verucchi adac8576b0 Add support to Scaled-YOLO4, update Yolov4x-mish (tested) 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2021-01-22 17:57:42 +01:00
Davide Sapienza 9f10c3f6e2 Add CenterTrack based on dla34. 
The implemented network refers to the nuScenes_3Dtracking.pth model.

Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2020-12-07 17:59:27 +01:00
Davide Sapienza 7f65ee0b2a Merge remote-tracking branch 'origin/master' into cnet 
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2020-12-07 17:45:44 +01:00
Micaela Verucchi a52e18b6e6 Add shelfnet_mapillary, README_seg, resize of input 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-11-24 12:37:33 +01:00
Micaela Verucchi 702791e41a Add support for yolov4x-mish. 
Changes:
- add parameters nms_kind, nms_thresh, new_coords to yolo layer and darknet parser
- added diou nms, new method to compute the BBs
- created test for yolov4x-mish called yolo4x

Tested, all tests work.
Problem to solve: little loss in mAP of yolo4x

Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-11-23 11:29:32 +01:00
Micaela Verucchi 86478f9384 Add yolo4_mmr test 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-10-23 11:40:55 +02:00
tk a68f45cb4e Merge with master, add yolov4tiny_512 
Signed-off-by: tk <micaelaverucchi@gmail.com>
 2020-07-06 17:57:50 +02:00
Micaela Verucchi b4c8c2bbad Add stats 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-07-03 18:45:20 +02:00
Micaela Verucchi 7c2155decf Add weights download link for csresnext50-panet-spp_berkeley ( fix #63 ) 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-07-01 11:56:46 +02:00
Micaela Verucchi a5cc4e3eda Add berkeley test, add weights for shelfnets 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-07-01 10:53:24 +02:00
Micaela Verucchi fe2e4eae92 yolo4tiny works on tensorRT 🐬 🐬 🐬 🐬 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-06-30 15:52:58 +02:00
Micaela Verucchi 61aa24c6b7 yolov4tiny works on CUDNN 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-06-30 15:19:03 +02:00
Micaela Verucchi 082920f3f5 Shelfnet works, also visualization. Postprocessing need to be parallelized 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-06-23 20:01:47 +02:00
Micaela Verucchi 94e558003d Shelfnet works on tensorRT (shortcut need to be fixed) 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-06-23 12:50:24 +02:00
Micaela Verucchi 6fd261f628 Shelfnet works on cuDNN. To test everything else 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-06-22 18:11:19 +02:00
Micaela Verucchi 9f1e30eaa9 Add shelfnet. Resnet18backbone works 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-06-19 18:55:42 +02:00
Francesco Gatti 1dfc69ba89 viz yolo3 preprocess 2020-06-16 13:19:30 +02:00
Francesco Gatti 3d3a2427c9 viz yolo3 2020-06-16 12:36:45 +02:00