061bc79a69
Added BatchNorm to NetworkRT (conver_layer)
2022-01-10 22:48:16 +05:30
bcf0c4eab3
Added BatchNorm Layer (Testing still needs to be done)
2022-01-10 22:48:06 +05:30
e1eac2d42a
Added a seperate layer file for batchnorm,to support an independent batchnorm class in order to pass necessary parameters read from the bin file.
2022-01-10 22:47:59 +05:30
3b58fbcb82
Added ConstantPadding plugin for TensorRT < 8.2
2022-01-07 13:05:09 +05:30
f189efcbb1
Reflection Padding native plugin fix ,forgot to added input_dim.c in the plugin creator
2022-01-07 05:41:26 +00:00
b75cecb105
Added constant and zero padding with ISliceLayer,need to add them for tensorrt versions less than 8.2 using IPluginV2Ext instead of ISliceLayer since they dont seem to support reflect and zero
2022-01-04 09:03:04 +05:30
ba022663f1
completed padding migrations from github
2022-01-03 19:49:32 +05:30
6837644eb2
add reflection padding from github
2022-01-03 18:58:13 +05:30
dcf4054bc6
performance improvements for yolo based networks,significant reduction in inference time can be seen yolo4tiny ,yolo4 and minor reduction in inference time can be seen in yolo4_berkeley_f1 and yolo4_berkeley - tested with a batchsize of 1 and 2 and on gtx 1070,it is possible that the performance improvement is more signficant in newer hardware
2022-01-03 13:52:16 +05:30
a4781244f4
Fix merge problem
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2021-11-24 18:16:15 +01:00
0707c26bbd
Merge of perseusdg-tensorrt8 inside tkDNN
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2021-11-24 17:59:27 +01:00
9cecc5051a
-Modified Dockerfile.base to cudagl
...
-Changed demo to take in input from demoConfig.yaml file
-Readme changes for demo.md
2021-11-24 03:39:57 +05:30
a8c98e3c31
Signed-off-by: perseusdg <43143075+perseusdg@users.noreply.github.com>
...
removed all TRT8_DEPRACTED functions
2021-11-19 22:40:52 +05:30
6f936096ae
Added num of layers in net and netRT,suggestions from pull request 270 on main repo
2021-11-14 07:13:42 +05:30
ee5000ccca
bug fixes for dla networks and ported optimization from different pull request
2021-11-13 02:00:49 +05:30
9e328c0daa
Migrate tkDNN max pooling plugin creation to pluginRegistry from the default method
2021-11-10 20:09:57 +05:30
c5e66c6bf6
TRT8 works with almost every nerual network now!!!!(including demo3d)
2021-10-28 23:35:37 +05:30
8c36dd0431
mobilenet works with trt7(IPluginV2IOExt) ,need to test it with trt8
2021-10-27 18:00:33 +05:30
54e7af11ed
tensorrt7 support for ipluginv2
2021-10-18 18:07:57 +05:30
18cc6abbb6
DeformableConvRT.h fixes ,cmake cuda arch auto detection
2021-09-07 23:57:52 +05:30
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
65ba5c9844
tkDNN can now deserialize tensorrt-8 engine (both through test_* and trtexec)
...
but demo has issues in yolo::computeDetections
2021-09-05 01:06:56 +05:30
de83ae5d25
update tensorrt8 branch
2021-08-30 19:04:26 +05:30
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
3d8b1ac494
Yolo3Detection.cpp doesn't build yet,issues with dependecies of preprocessing and postprocessing on IPluginFactory
2021-08-21 22:29:36 +05:30
eba78e7e78
Updates to build libkernel.so under TensorRT 8
2021-08-16 13:59:19 +05:30
d2ce966c46
fix return type for bool functions
2021-08-04 12:37:58 +05:30
1412aa66c4
minor fixes for windows ,for release v0.6
2021-07-24 08:51:44 -07:00
0292e21c33
Merge remote-tracking branch 'origin/master' into cnet
2021-07-22 17:03:23 +02:00
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
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
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
318dfe4b55
merge with master
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2021-07-20 18:38:46 +02:00
69256992a3
Add colormap for each segmentation dataset
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2021-07-20 15:29:26 +02:00
84ff978ccb
Merge with master, all tests passed
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2021-07-20 12:48:41 +02:00
b3c44a86a8
Fix shelfnet berkeley, update readme, minors
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2021-07-20 10:57:53 +02:00
b86a93e85d
ActivationLeaky IpluginV2
2021-06-28 23:23:15 +02:00
ba8199a030
fix compile error on old tensorrt and remove useless prints
2021-05-16 22:33:45 +02:00
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
0dc96d2a9e
Improve CenterTrack.
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2021-05-04 17:59:20 +02:00
f8327e2dac
Merge with cnet
2021-05-04 11:50:08 +02:00
10f39d1055
Fix tracker for batch size > 1
2021-05-04 11:21:05 +02:00
ff6e0e010a
Fix a bug with batch > 1
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2021-04-30 22:26:33 +02:00
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
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
28fab9c3e1
Merge remote-tracking branch 'origin/master' into cnet
2021-04-27 14:43:39 +02:00
6611a91201
merge perseusdg windows10 support
2021-04-24 18:55:26 +02:00
1de804f98d
Code cleanup and readme fixes
2021-04-09 13:17:41 +05:30
37b2a5bd98
minor modifications
2021-04-08 00:57:58 +05:30
cc594f09ef
merge from gitlab
2021-04-08 00:28:00 +05:30
perseusdg
perseusdg
Micaela Verucchi
perseusdg
Harshvardhan Chandirasekar
Davide Sapienza
Francesco Gatti
Fabio Bagni