Commit History

Increase logging level to extra verbose
a9584a8

Luigi commited on

Increase logging level to verbose on onnx to engine script
3deadf3

Luigi commited on

trim sample video to keep only important part
f5e3a24

luigi commited on

Make 2 distinct ONNX2TRT conversion scripts, one for JetPack 4.6, the other for JetPack 5.1
06f3435

Luigi commited on

adjust visualisation style, allow draw_bbox() to show person id
a9e21bc

Luigi commited on

Add method to free buffers
b79a8af

Luigi commited on

Implement pycuda backend for inference with TensorRT engine
889281f

Luigi commited on

Add 'get_model_format_and_input_shape' helper function
2f2f685

Luigi commited on

Return and show bounding box confidence
634d4ff

Luigi commited on

Deal with multiple cameras
98d66d4

Luigi commited on

Show bounding box on screen too
0bf1eb7

Luigi commited on

Internalize input frame buffer
846e714

Luigi commited on

Unify batched and non-batched versions
42892de

Luigi commited on

Show FPS on demo video
15801f5

Luigi commited on

Support Inference over batch with TensorRT Engine Model
0cdc9a7

Luigi commited on

Use dynamic batch size by default
09ccc6e

Luigi commited on

Add script used to convert ONNX to fp32/fp16/int8/mixed engine
92676db

Luigi commited on

Use ONNXruntime instead of ONNX.checker.check_model to detect ONNX model
57a8c6b

Luigi commited on

Resize in keeping aspect ratio in visualization
35b2a45

Luigi commited on

fix postprocess_batch method in RTMO_GPU_Batch class
6008f96

Luigi commited on

Add model format (ONNX/Engine) & input size detection (based on file header, not on filename) for RTMO
58a44cf

Luigi commited on

Add support TensorRT engine support for RTMO
bbf20b6

Luigi commited on

Re-quantize models in FP16 in keep positional encoding in FP32 to avoid bad accuracy
f9a6075
verified

Luigi commited on

Improve Debug Message Layout in Validate Funciton
e8f08e3

Luigi commited on

Add appropriate pre- to input image and postprocessing to validate function
6dfe441

Luigi commited on

Validate only on pose result, add visual check
b58a63b

Luigi commited on

Add comparison between joint coordinated from original and converted models
ab55aa6

Luigi commited on

Add 'rtol', 'atol' arguments to control tolerance in model validation
684ebde

Luigi commited on

Fix Input Type Error raised in infer()
18eb8e3

Luigi commited on

Add utility to convert ONNX model in FP32/16 mixed precision
9f60e86

Luigi commited on

Load ./libmmdeploy_tensorrt_ops.so for TensorRT EP if available
1c5dc58

Luigi commited on

Remove fix_batch_dimension_all.sh
01042ef

Luigi commited on

Remove models with fixed batch size as dynamic batch is generally supported
ae1686f

Luigi commited on

Apply constant folding
d474b05

Luigi commited on

Reperform shape inference on all models
e7b68a5

Luigi commited on

Bugfix: correct batch dimension fixation then regenerate ONNX models
e4e03fd

Luigi commited on

Update README.md
bd45a81
verified

Luigi commited on

Update README.md
0a36dd1
verified

Luigi commited on

Add option --yolo_nas_pose, used to read YOLO NAS Pose model instead of RTMO
1f0f5d8

Luigi commited on

Fix typo in model names mentioned in demo.sh demo_batch.sh
9133777

Luigi commited on

Repair DEMO to run with TRT EP
c4f84a9

Luigi commited on

Fix FP16 models having poor accuracy
25f1444

Luigi commited on

Bugfix for batch demo: add padding if remaning number of frames is not equal to batch size
12e168d

Luigi commited on

Make ONNX models compatible with ONNXruntime's TensorrtExecutionProvider
b75f05d

Luigi commited on

Use RTMO_GPU_Batch to demo
3939a91

Luigi commited on

Add new class 'RTMO_GPU_Batch' that can perform inference on batch of images
8072759

Luigi commited on

Optimization for CUDAExecutionProvider
43a489d

Luigi commited on

INT8 models obtained via dynamic quantization
e1ce29b
verified

Luigi commited on

Update README.md
c4cff5e
verified

Luigi commited on

Add a sample video
2e05515
verified

Luigi commited on