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
59b0f434a7
Fix a bug in the 3D bounding boxes.
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-12-14 15:32:35 +01: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
9e1d7b3bb4
Update demo3d with show flag
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-12-09 18:05:54 +01:00
dbc052865c
Fix a wrong path in CenternetDetection3DTrack.cpp
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-12-09 18:04:42 +01:00
4543df8533
Update readme.
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-12-07 18:56:07 +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
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
8fb5772ad9
Update README_seg.md
2020-11-24 13:03:02 +01:00
43d213ba36
Update README_seg.md
2020-11-24 12:44:36 +01:00
f137bcb694
Update README_seg
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-11-24 12:39:37 +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
b8855b9599
Update README
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-11-23 11:34:06 +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
a0e7f05a50
Update README.md
2020-10-10 13:03:01 +02:00
d3372aad31
Update README with Xavier NX FPS results 15W4Core
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-09-15 14:09:02 +02:00
be818d5e3a
Update README with Xavier NX FPS results
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-09-15 10:50:24 +02: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
38106a9495
Merge branch 'master' of https://github.com/ceccocats/tkDNN
2020-08-06 15:54:09 +02:00
df5443e017
Fix boxes also for Centernet
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-08-06 15:53:57 +02:00
8bef544bae
Update README.md
2020-08-06 11:05:26 +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
a5d2d4792a
fix coords convert
2020-07-27 13:52:39 +02:00
3a0802d70c
Resolve detection objects pick by prob threshold.
...
Before this it will only pick the last object with prob > thresh wich is absolutely wrong
Now it picks all the objects with prob > thesh.
fixes #94
2020-07-27 13:45:42 +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
f4970d1e6f
Update README.md
2020-07-17 14:37:10 +02:00
6a68f19b2c
Fix patch from @ahmedius2 , tkDNN now supports CUDNN 8.0.1 ( Fix #74 )
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
Francesco Gatti <gattifrancesco@hotmail.it >
2020-07-16 19:06:54 +02:00
c4aad7fe95
Patch for CUDNN 8.0.1
...
Signed-off-by: tk <micaelaverucchi@gmail.com >
2020-07-16 18:16:09 +02:00
b2df9fc110
Update README.md
2020-07-13 19:51:09 +02:00
b12cf0d7c2
docker
2020-07-13 19:50:00 +02:00
594947f301
Update file names map demo
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-07-08 11:27:09 +02:00
e7779ad773
Add conf_thresh parameter to detectors, add batches handling in map demo
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-07-08 09:52:36 +02:00
65e2074dda
Merge branch 'eval' of https://github.com/ceccocats/tkDNN into eval
2020-07-06 17:58:08 +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
79cd96de6f
Add resize to original size, writing of segmentation
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-07-02 12:14: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
a4dca23111
Minor
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-07-01 10:22:31 +02:00
04602f3952
Yolo4-tiny batched
...
fix #59
2020-06-30 19:37:16 +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
3bf9547502
Improve preprocessing
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-06-29 15:11:52 +02:00
6d9beb1ec5
Merge pull request #53 from omaralvarez/master
...
Fix error when parsing label files with unexpected chars
2020-06-25 10:07:41 +02:00
d25803d438
Post-processing improved
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-06-24 18:46:25 +02:00
perseusdg
Davide Sapienza
Micaela Verucchi
Francesco Gatti