Commit Graph

536 Commits

Author SHA1 Message Date
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
perseusdg eba78e7e78 Updates to build libkernel.so under TensorRT 8 2021-08-16 13:59:19 +05:30
perseusdg d2ce966c46 fix return type for bool functions 2021-08-04 12:37:58 +05:30
Harshvardhan Chandirasekar bb902f5b65 fix for downloading weights(utils.cpp) when using docker 2021-07-25 23:53:30 +05:30
perseusdg 1412aa66c4 minor fixes for windows ,for release v0.6 2021-07-24 08:51:44 -07:00
Harshvardhan Chandirasekar e58ddadb7d Merge branch 'ceccocats:master' into master 2021-07-24 08:18:17 -07:00
Micaela Verucchi a992c9feb5 Update README.md 2021-07-23 16:50:15 +02:00
Micaela Verucchi 09080709a9 Remove Issues.md
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2021-07-23 15:23:47 +02:00
Micaela Verucchi 7521d10ba7 Update README with gifs
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2021-07-23 15:07:52 +02:00
Micaela Verucchi ab349083bf Merge with cnet works, all tests passed.
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
v0.6
2021-07-23 14:37:04 +02:00
Micaela Verucchi 4f9f27152a Update READMEs
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2021-07-23 10:31:34 +02:00
Davide Sapienza 0292e21c33 Merge remote-tracking branch 'origin/master' into cnet 2021-07-22 17:03:23 +02:00
Davide Sapienza 2bb70859da Add FPS results for CenterTrack 2D/3D
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
2021-07-22 16:54:25 +02:00
Davide Sapienza b9f82510d3 Add demo_utils.
This commit splits the utils file into two different files:
utils and demo_utils, to overcome some opencv problems caused
by demo_utils includes.

Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
2021-07-22 10:45:18 +02:00
Davide Sapienza 1216e8bb74 Add camera calibration file reading for demo3d and demoTracker.
Some minor fixes.

Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
2021-07-21 16:14:53 +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
Davide Sapienza f78f7bfddc Fix link in readme files
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
2021-07-21 15:35:17 +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
Harshvardhan Chandirasekar 548e87fc0e Merge branch 'ceccocats:master' into master 2021-07-21 12:44:24 +05:30
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
Harshvardhan Chandirasekar 4ed247df58 Merge branch 'ceccocats:master' into master 2021-07-20 21:24:24 +05:30
Micaela Verucchi 946b1fe5f5 Update READMEs
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2021-07-20 17:30:32 +02:00
Micaela Verucchi 69256992a3 Add colormap for each segmentation dataset
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2021-07-20 15:29:26 +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 4ca69836e9 Update README_seg
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2021-07-20 11:39:16 +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
Harshvardhan Chandirasekar 7f9b10ca71 Merge branch 'ceccocats:master' into master 2021-06-18 06:16:13 -07:00
Francesco Gatti c306b36860 Update CMakeLists.txt 2021-06-17 15:31:25 +02:00
Francesco Gatti 9b78f143cb Update CMakeLists.txt 2021-06-16 14:53:46 +02:00
Francesco Gatti d847a4b852 Update CMakeLists.txt 2021-06-15 20:33:41 +02:00
Francesco Gatti 8df7d5fd1b Update CMakeLists.txt 2021-06-15 20:31:58 +02:00
Francesco Gatti 6d7c456f72 cmake version fix 2021-06-15 19:39:34 +02:00
Francesco Gatti b0fdeb4127 install targets 2021-06-15 19:38:13 +02:00
perseusdg 710cb54db3 Centernet fix for windows 2021-06-05 22:12:15 +05:30
Harshvardhan Chandirasekar 23fbcd1850 Merge branch 'ceccocats:master' into master 2021-06-05 21:23:40 +05:30
Francesco Gatti ba8199a030 fix compile error on old tensorrt and remove useless prints 2021-05-16 22:33:45 +02:00
Francesco Gatti 6b8ae1e27c remove wrong cuda arch that cause big performance gap
fix #226
2021-05-16 22:16:21 +02:00
Davide Sapienza 25d02ef3ea Update exporting weights README
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
2021-05-11 17:30:24 +02:00
Davide Sapienza fd56e64938 Update the README and split it into several files.
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
2021-05-11 17:02:43 +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 0dc96d2a9e Improve CenterTrack.
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
2021-05-04 17:59:20 +02:00
Fabio Bagni f8327e2dac Merge with cnet 2021-05-04 11:50:08 +02:00
Fabio Bagni 10f39d1055 Fix tracker for batch size > 1 2021-05-04 11:21:05 +02:00
Davide Sapienza d1ae1791d9 Fix a bug in the draw function of CenterTrack.
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
2021-05-03 18:56:07 +02:00
Davide Sapienza ff6e0e010a Fix a bug with batch > 1
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
2021-04-30 22:26:33 +02:00
Davide Sapienza 2367519799 Add the calibration matrix reading for CenterTrack
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
2021-04-30 17:10:51 +02:00
Davide Sapienza be6ad27c11 Batch size > 1 for the 3D demo.
This commit lets to use differtent batch size for 3D CenterNet
and CenterTrack.

Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
2021-04-29 11:13:24 +02:00