04de9908a6
Add yolo4-csp for crowdhuman dataset, add shelfnet for coco-stuff dataset, fix minor in demo.cpp
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2021-11-29 17:33:57 +01: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
Micaela Verucchi
Davide Sapienza
Francesco Gatti
hchandirasekar