76c978a200
Added TensorRT8 support
...
- Migrated to TRT8 API, but I didn't really try maintaining old compatiblity. Lowest support now could be TRT6.
- Refractored some stuff that are bad C++ practices and made coding really hard, like including headers in namespaces or not using an include guard.
- Needed to move the yolo container outside tkdnn object, which means we now only have one and global. Deserialization in TRT8 doesn't happen in your object, but in the plugin itself so it couldn't access the yolo objects. I think the need to hold onto yolo layers itself is flawed and shouldn't be nessesarry.
2021-10-28 01:02:26 +03:00
eca10ac0a8
Update weights
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2021-09-19 00:16:31 +02:00
0292e21c33
Merge remote-tracking branch 'origin/master' into cnet
2021-07-22 17:03:23 +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
f4e71a5a28
Add yolo4_berkeley_f1
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2021-07-21 12:56:00 +02:00
89b1bb7bee
Add missing files
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2021-07-20 18:39:25 +02:00
318dfe4b55
merge with master
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2021-07-20 18:38:46 +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
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
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
f12ec3c935
added yolo4x and yolo4-csp from the github repo and download file corrections
2021-03-26 12:18:46 +05:30
4b3731928c
Add yolo4_320_coco2 (pedestrian and stop sign)
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2021-02-25 09:57:10 +01: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
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
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
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
86478f9384
Add yolo4_mmr test
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-10-23 11:40:55 +02:00
a68f45cb4e
Merge with master, add yolov4tiny_512
...
Signed-off-by: tk <micaelaverucchi@gmail.com >
2020-07-06 17:57:50 +02:00
b4c8c2bbad
Add stats
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-07-03 18:45:20 +02:00
7c2155decf
Add weights download link for csresnext50-panet-spp_berkeley ( fix #63 )
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-07-01 11:56:46 +02:00
a5cc4e3eda
Add berkeley test, add weights for shelfnets
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-07-01 10:53:24 +02:00
fe2e4eae92
yolo4tiny works on tensorRT 🐬 🐬 🐬 🐬
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-06-30 15:52:58 +02:00
61aa24c6b7
yolov4tiny works on CUDNN
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-06-30 15:19:03 +02:00
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
94e558003d
Shelfnet works on tensorRT (shortcut need to be fixed)
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-06-23 12:50:24 +02:00
6fd261f628
Shelfnet works on cuDNN. To test everything else
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-06-22 18:11:19 +02:00
9f1e30eaa9
Add shelfnet. Resnet18backbone works
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-06-19 18:55:42 +02:00
1dfc69ba89
viz yolo3 preprocess
2020-06-16 13:19:30 +02:00
3d3a2427c9
viz yolo3
2020-06-16 12:36:45 +02:00
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
6dff675db7
fix wrong commit
2020-06-14 13:03:48 +02:00
cbfc8ea4f2
serialize fix
2020-06-14 13:01:45 +02:00
8d08f5aade
New yolo4_512 download link
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-06-11 20:54:17 +02:00
e094a3e0fc
Add script for inference FPS
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-06-11 16:09:01 +02:00
c4e955eab5
Add different yolov4 size tests
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-06-11 12:43:34 +02:00
c8ed6d782a
all test ok
2020-06-01 16:15:29 +02:00
d4e0d07e09
merge
2020-06-01 15:34:59 +02:00
a0e4e9f209
Modify yolov4_berkeley link for download, clean to merge with master
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-06-01 15:17:53 +02:00
18794d52c2
darknet parser tested
2020-06-01 14:55:08 +02:00
d2e2669b6d
darknet parse all net to be tested
2020-06-01 12:22:55 +02:00
d8fbee58d8
Add names files
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-06-01 10:42:06 +02:00
826fcc97c8
Read classes' names from file
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-06-01 10:34:48 +02:00
6d81473b2a
yolo3 parsed ok
2020-05-30 22:11:00 +02:00
0ae96a6bc4
darknet parser to be tested on yolo3
2020-05-30 19:38:26 +02:00
e5e6654b1d
Add fields to darknetParseFields
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-05-30 18:44:17 +02:00
298487d5c5
parse yolo layers
2020-05-30 18:44:02 +02:00
041a5cf65d
Add some fields
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-05-30 18:00:22 +02:00
Harijs Grinbergs
Micaela Verucchi
Davide Sapienza
Francesco Gatti
hchandirasekar