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