Commit Graph

157 Commits

Author SHA1 Message Date
2f52521b9a Using bayer encoding for images. 2025-01-13 16:50:08 +01:00
23f8df32d5 Merge branch 'master' into jetson 2025-01-13 15:01:06 +01:00
53543368c4 Merge branch 'trt' into 'master'
Using TensorRT for speedups

See merge request Percipiote/RapidPoseTriangulation!3
2025-01-13 13:47:10 +00:00
ba234628eb Small docker fix. 2025-01-13 14:43:26 +01:00
5187310d16 Update of skelda submodule. 2024-12-19 16:20:21 +01:00
0dd0f84e19 Updated results. 2024-12-19 16:02:18 +01:00
c5c9266264 Added jetson results. 2024-12-19 15:55:14 +01:00
d412645880 Merge remote-tracking branch 'origin/trt' into jetson 2024-12-19 15:16:16 +01:00
f31d6464d7 Update readme. 2024-12-19 14:56:18 +01:00
b483269697 Renamed new wrapper script. 2024-12-19 14:23:41 +01:00
7b073249d6 Deleted old 2D pose wrapper. 2024-12-19 14:22:48 +01:00
5bc7bd9094 Splitting 2D and 3D predictions into single loops to prevent CPU priority issues. 2024-12-19 14:13:12 +01:00
db47a8af6a Updated dockerfile. 2024-12-19 14:01:07 +01:00
2ad20549f2 Small updates. 2024-12-19 13:52:44 +01:00
19bb1a6372 Use fp32 model for whole-body poses. 2024-12-19 11:12:13 +01:00
1bb7dd966e Converted wb model as well. 2024-12-18 16:53:32 +01:00
f8984f9408 Some mixed updates. 2024-12-18 16:28:13 +01:00
07426fac2f Optional batched pose processing. 2024-12-18 16:22:08 +01:00
7b8d209601 Cache onnx-trt models, they are faster than using two trt-engines. Removed trt-runtime again. 2024-12-18 14:28:25 +01:00
b26ec998b3 Using tensorrt runtime directly. 2024-12-18 12:10:45 +01:00
6e8f6a22ba Running mmdeploy benchmark. 2024-12-18 11:51:05 +01:00
21e93661f5 Updated dataset configs and results. 2024-12-16 14:43:54 +01:00
17ff41010a Some small updates. 2024-12-16 14:34:44 +01:00
07f75c53ee Renamed a parameter. 2024-12-09 14:17:35 +01:00
23108dd594 Further small improvements. 2024-12-06 18:15:08 +01:00
ee8b9bafb3 Updated easypose scripts. 2024-12-06 17:35:49 +01:00
0f55599da2 Switched to onnx backend. 2024-12-06 16:51:49 +01:00
108937d96c Some mixed improvements. 2024-12-06 13:58:34 +01:00
7a253cd615 Moved image cropping out of the graph again. 2024-12-06 12:32:51 +01:00
39bb132a98 Moved some files. 2024-12-05 10:48:31 +01:00
8a12e8bdfd Small readme updates. 2024-12-05 10:41:59 +01:00
acf1d19b64 Improved box cutting with always fixed tensor shapes. 2024-12-04 17:54:57 +01:00
6452d20ec8 Some small updates. 2024-12-04 11:46:06 +01:00
97ff32b9ce Moved pose pre/post-processing into onnx graph. 2024-12-03 17:59:31 +01:00
2e5c3399ed Workaround problem of tensorrt hanging sometimes. 2024-12-03 17:39:45 +01:00
742d2386c7 Moved detector pre/post-processing into onnx graph. 2024-12-03 11:52:55 +01:00
36781e616b Use rgb input for both models. 2024-12-02 17:28:34 +01:00
dc44a71b2c Some further speedups. 2024-12-02 17:00:04 +01:00
c8c48e4bf2 Using TensorRT backend. 2024-12-02 13:50:35 +01:00
bf837fbca9 Adding first results. 2024-12-02 13:46:17 +01:00
b44d4134d9 Move fp16 casting into onnx models. 2024-12-02 13:45:26 +01:00
9cc63532d3 Extracted onnx tools into a single script. 2024-12-02 12:32:53 +01:00
d50364cda7 Some further optimizations. 2024-11-29 18:22:48 +01:00
ad1ef42b99 Export pose postprocessing steps as well. 2024-11-29 17:59:15 +01:00
19015c6326 Exporting fp16 onnx models. 2024-11-29 17:37:44 +01:00
7b9505ca02 Moved image normalization into onnx models. 2024-11-29 17:05:43 +01:00
1b5e0c44e3 Speed up preprocessing. 2024-11-29 16:19:06 +01:00
93d4611a91 Fixed running custom onnx models. 2024-11-29 15:18:57 +01:00
f6d13ea5a7 Testing onnx runtime with easypose. 2024-11-29 14:25:57 +01:00
ae21a3cf29 Updating base image. 2024-11-29 13:53:47 +01:00