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
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
2e92944f1d
Opencv cuda fix
2021-03-31 03:44:51 -07:00
f12ec3c935
added yolo4x and yolo4-csp from the github repo and download file corrections
2021-03-26 12:18:46 +05:30
78859fe191
timer fix
2021-03-25 16:38:53 +05:30
e94e1f7622
tkdnn first patch for windows
2021-03-24 22:26:46 +05:30
06787a931f
minor migrations
2021-03-17 23:06:50 +05:30
304ab49897
shared_ptr migrations
2021-03-17 22:26:26 +05:30
4a90314333
minor fixes in test.h
2021-01-26 23:11:58 +05:30
2d4dececb6
Builds on windows successfully,issues with deserialization and downloading weights
2021-01-26 20:17:43 +04:00
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
512acd8cba
minor fixes
2021-01-21 09:29:27 +04:00
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
1cfa199ee6
Add pre-processing and post-processing stats
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-12-09 19:43:51 +01:00
48ecebe6dd
Add CenterTrack pre, post, visualization and demo.
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-12-07 18:45:29 +01:00
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
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
a17e7800b9
Add computation of #parameters, #MACC, and max feature map size in the tests
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-11-23 13:07:54 +01:00
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
04b4a69107
Merge with master
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-09-14 17:30:02 +02:00
fa2b3d26cb
Fix typos ( #107 )
...
Signed-off-by: micaela <micaelaverucchi@gmail.com >
2020-09-11 09:13:59 +02:00
f778e1aa99
Fixed boxes to float, add conf thresh as param
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-08-05 19:55:10 +02:00
286e777300
Improved segmentation results, removed resize, code to reorder
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-07-22 19:21:46 +02:00
perseusdg
Harshvardhan Chandirasekar
Davide Sapienza
Micaela Verucchi
Francesco Gatti
Fabio Bagni