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

127 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 eba78e7e78 Updates to build libkernel.so under TensorRT 8 2021-08-16 13:59:19 +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 84ff978ccb Merge with master, all tests passed 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2021-07-20 12:48:41 +02: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
Francesco Gatti 6b8ae1e27c remove wrong cuda arch that cause big performance gap 
fix #226
 2021-05-16 22:16:21 +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
Fabio Bagni f8327e2dac Merge with cnet 2021-05-04 11:50:08 +02:00
Davide Sapienza 28fab9c3e1 Merge remote-tracking branch 'origin/master' into cnet 2021-04-27 14:43:39 +02:00
Harshvardhan Chandirasekar f90ee8ab7d Update CMakeLists.txt 2021-04-09 13:20:44 +05:30
perseusdg 1de804f98d Code cleanup and readme fixes 2021-04-09 13:17:41 +05:30
hchandirasekar 44b71ae6f3 ReadMe.md windows changes 2021-03-25 20:44:43 +05:30
hchandirasekar 78859fe191 timer fix 2021-03-25 16:38:53 +05:30
Harshvardhan Chandirasekar 304ab49897 shared_ptr migrations 2021-03-17 22:26:26 +05:30
hchandirasekar 6aa8666be5 Commits for msvc 16.9 2021-03-10 12:59:04 +05:30
hchandirasekar 4a90314333 minor fixes in test.h 2021-01-26 23:11:58 +05:30
hchandirasekar 2d4dececb6 Builds on windows successfully,issues with deserialization and downloading weights 2021-01-26 20:17:43 +04:00
hchandirasekar fb52444cdc Replaced dynamic arrays with std::vector ,works on linux ..needs to be tested on windows after clearing up the lnk2019 error 2021-01-25 04:12:05 +05:30
perseusdg 56feb54377 able to build kernels as shared object file(dll),and minor changes to lstm.cpp and utils.cpp to overcome minor msvc build errors 2021-01-21 00:22:15 +04: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 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 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 9f1e30eaa9 Add shelfnet. Resnet18backbone works 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-06-19 18:55:42 +02:00
Micaela Verucchi 285c77d6dd Imptove relative paths 
Signed-off-by:  Micaela Verucchi <micaelaverucchi@gmail.com>
		Francesco Gatti <gattifrancesco@hotmail.it>
 2020-06-15 17:15:37 +02:00
Francesco Gatti 567dc0f75d cmake cudnn fix 2020-06-14 12:57:29 +02:00
Francesco Gatti d2e2669b6d darknet parse all net to be tested 2020-06-01 12:22:55 +02:00
Davide Sapienza 7a677d5c10 Add CenterNet based on Resnet101 for 3D, CUDNN and TensorRT work 
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2020-05-29 14:54:36 +02:00
Davide Sapienza 6bdf47bae6 Add 3D demo program 
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2020-05-27 18:05:38 +02:00
Davide Sapienza 64098ad244 Add CenterNet based on DLA34 for 3D, CUDNN and TensorRT work 
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2020-05-22 12:51:25 +02:00
Francesco Gatti 85bbbf42a1 merge 2020-05-03 15:58:00 +02:00
Micaela Verucchi db0f8a4d99 Support yolov4 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-04-28 20:35:30 +02:00
Micaela Verucchi f418a1f9b2 Add bdd new tests, add BDD100K_val download, fix yolo3-512 link 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-04-24 12:39:57 +02:00
Francesco Gatti e3e5a133f5 pull from repos 2020-04-12 21:47:25 +02:00
Francesco Gatti f9e3f17c0a cmake dont download test data 2020-04-09 20:00:29 +02:00
Francesco Gatti 2635d24855 Merge branch 'master' into cnet 2020-04-09 00:57:43 +02:00
Francesco Gatti 1bb9f73a0b remove deprecated and unused warning 2020-04-09 00:55:07 +02:00
Micaela Verucchi 0951f4644f Merge with master works 
Signed-off-by:  Micaela Verucchi <micaelaverucchi@gmail.com>
		Davide Sapienza <sapienza.dav@gmail.com>
 2020-04-08 14:59:42 +02:00
Micaela Verucchi af13e7c954 Add getMemoryUsage function, detection update moved in abstract lass, splitted execution time in pre-inf-post, other minors. 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-04-01 19:18:43 +02:00
Micaela Verucchi 7c55dcb708 Add tp tests for yolo3512 and yolo3tiny512 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-03-26 15:06:49 +01:00
Micaela Verucchi 1d388ca51a Add mobilenetv2ssdlite512 test. 
Works for CuDNN, not for tensorRT.
Modified channels in second convolution for classification headers from 126 to 486, when changing size from 300 to 512.
Added size 512 SSD specs and support to COCO dataset (81 classes, first BACKGROUND due to repo for training).
Refactoring class names.

Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-03-17 10:35:02 +01:00
Micaela Verucchi 9c25d15ff2 Add yolov3_512 (size 512) test 2020-03-16 11:29:15 +01:00