Add script used to convert ONNX to fp32/fp16/int8/mixed engine 92676db Luigi commited on May 17, 2024
Use ONNXruntime instead of ONNX.checker.check_model to detect ONNX model 57a8c6b Luigi commited on May 17, 2024
Add model format (ONNX/Engine) & input size detection (based on file header, not on filename) for RTMO 58a44cf Luigi commited on May 15, 2024
Re-quantize models in FP16 in keep positional encoding in FP32 to avoid bad accuracy f9a6075 verified Luigi commited on Mar 27, 2024
Add appropriate pre- to input image and postprocessing to validate function 6dfe441 Luigi commited on Mar 26, 2024
Add comparison between joint coordinated from original and converted models ab55aa6 Luigi commited on Mar 25, 2024
Add 'rtol', 'atol' arguments to control tolerance in model validation 684ebde Luigi commited on Mar 25, 2024
Load ./libmmdeploy_tensorrt_ops.so for TensorRT EP if available 1c5dc58 Luigi commited on Mar 19, 2024
Remove models with fixed batch size as dynamic batch is generally supported ae1686f Luigi commited on Mar 19, 2024
Bugfix: correct batch dimension fixation then regenerate ONNX models e4e03fd Luigi commited on Mar 16, 2024
Add option --yolo_nas_pose, used to read YOLO NAS Pose model instead of RTMO 1f0f5d8 Luigi commited on Mar 8, 2024
Bugfix for batch demo: add padding if remaning number of frames is not equal to batch size 12e168d Luigi commited on Mar 7, 2024
Make ONNX models compatible with ONNXruntime's TensorrtExecutionProvider b75f05d Luigi commited on Mar 7, 2024
Add new class 'RTMO_GPU_Batch' that can perform inference on batch of images 8072759 Luigi commited on Mar 5, 2024