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
6a133d8dec
-Replaced ISliceLayer based paddings(reflection,constant and zero) with the IPluginV2 version for TensorRT >= 8.2.0
...
-Fixed demo build issue on Windows
Signed-off-by: perseusdg <harshvardhan.chandira@gmail.com >
2022-01-20 00:52:22 +05:30
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
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
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
367061fea2
fps fix
2021-11-13 18:11:46 +05:30
d488c3bd17
remove duplicated lines
2021-11-13 02:12:50 +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
7dd33cd118
added assert for supportsFormat in plugins
2021-11-10 14:37:16 +05:30
936b680f2f
Remove debug prints
2021-11-10 13:45:44 +05:30
744396fb0e
Update README.md,windows.md and demo.cpp
...
Small fixes in DeformableConvRT.cpp
2021-11-09 13:37:26 +05:30
d2d44e9e92
Fix max elem (remove thrust) for segmentation
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2021-11-05 19:54:02 +05:30
802c01bd8f
windows debug fix
2021-10-30 19:29:21 +05:30
7a89a4a573
ReshapeRT.cpp fix
2021-10-28 23:43:12 +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
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
9fa116ce4a
tkdnn(trt8) runs now minus the detections ,[executionContext.cpp::enqueueInternal::312] Error Code 3: Internal Error (Parameter check failed at: runtime/api/executionContext.cpp::enqueueInternal::312, condition: mDeviceMemorySize == 0 || mExecutionResources->getDeviceMemory() != nullptr - is the error that causes it
2021-09-07 07:20:46 +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
bb902f5b65
fix for downloading weights(utils.cpp) when using docker
2021-07-25 23:53:30 +05:30
1412aa66c4
minor fixes for windows ,for release v0.6
2021-07-24 08:51:44 -07:00
e58ddadb7d
Merge branch 'ceccocats:master' into master
2021-07-24 08:18:17 -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
548e87fc0e
Merge branch 'ceccocats:master' into master
2021-07-21 12:44:24 +05:30
318dfe4b55
merge with master
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2021-07-20 18:38:46 +02:00
4ed247df58
Merge branch 'ceccocats:master' into master
2021-07-20 21:24:24 +05:30
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
710cb54db3
Centernet fix for windows
2021-06-05 22:12:15 +05:30
ba8199a030
fix compile error on old tensorrt and remove useless prints
2021-05-16 22:33:45 +02:00
Harshvardhan Chandirasekar
Micaela Verucchi
perseusdg
perseusdg
Harshvardhan Chandirasekar
Davide Sapienza
Francesco Gatti