rtmo

3 contributors

History: 18 commits

Luigi

Reperform shape inference on all models

e7b68a5 8 months ago

video
Add a sample video 9 months ago
.gitattributes

1.66 kB

Add a sample video 9 months ago
README.md

1.47 kB

Update README.md 9 months ago
convert_to_fp16.py

1.11 kB

Fix FP16 models having poor accuracy 9 months ago
convert_to_fp16_all.py

1.36 kB

Bugfix: correct batch dimension fixation then regenerate ONNX models 8 months ago
demo.sh

63 Bytes

Fix typo in model names mentioned in demo.sh demo_batch.sh 9 months ago
demo_batch.sh

71 Bytes

Fix typo in model names mentioned in demo.sh demo_batch.sh 9 months ago
fix_batch_dimension_all.sh

898 Bytes

Bugfix: correct batch dimension fixation then regenerate ONNX models 8 months ago
rtmo-l.fp16.onnx

88 MB
LFS

Reperform shape inference on all models 8 months ago
rtmo-l.fp16_batch1.onnx

88 MB
LFS

Reperform shape inference on all models 8 months ago
rtmo-l.fp16_batch2.onnx

88 MB
LFS

Reperform shape inference on all models 8 months ago
rtmo-l.fp16_batch4.onnx

88 MB
LFS

Reperform shape inference on all models 8 months ago
rtmo-l.onnx

176 MB
LFS

Make ONNX models compatible with ONNXruntime's TensorrtExecutionProvider 9 months ago
rtmo-l_batch1.onnx

176 MB
LFS

Reperform shape inference on all models 8 months ago
rtmo-l_batch2.onnx

176 MB
LFS

Reperform shape inference on all models 8 months ago
rtmo-l_batch4.onnx

176 MB
LFS

Reperform shape inference on all models 8 months ago
rtmo-m.fp16.onnx

44.7 MB
LFS

Reperform shape inference on all models 8 months ago
rtmo-m.fp16_batch1.onnx

44.7 MB
LFS

Reperform shape inference on all models 8 months ago
rtmo-m.fp16_batch2.onnx

44.7 MB
LFS

Reperform shape inference on all models 8 months ago
rtmo-m.fp16_batch4.onnx

44.7 MB
LFS

Reperform shape inference on all models 8 months ago
rtmo-m.onnx

89.3 MB
LFS

Make ONNX models compatible with ONNXruntime's TensorrtExecutionProvider 9 months ago
rtmo-m_batch1.onnx

89.3 MB
LFS

Reperform shape inference on all models 8 months ago
rtmo-m_batch2.onnx

89.3 MB
LFS

Reperform shape inference on all models 8 months ago
rtmo-m_batch4.onnx

89.3 MB
LFS

Reperform shape inference on all models 8 months ago
rtmo-s.fp16.onnx

19.9 MB
LFS

Reperform shape inference on all models 8 months ago
rtmo-s.fp16_batch1.onnx

19.9 MB
LFS

Reperform shape inference on all models 8 months ago
rtmo-s.fp16_batch2.onnx

19.9 MB
LFS

Reperform shape inference on all models 8 months ago
rtmo-s.fp16_batch4.onnx

19.9 MB
LFS

Reperform shape inference on all models 8 months ago
rtmo-s.onnx

39.7 MB
LFS

Make ONNX models compatible with ONNXruntime's TensorrtExecutionProvider 9 months ago
rtmo-s_batch1.onnx

39.6 MB
LFS

Reperform shape inference on all models 8 months ago
rtmo-s_batch2.onnx

39.6 MB
LFS

Reperform shape inference on all models 8 months ago
rtmo-s_batch4.onnx

39.6 MB
LFS

Reperform shape inference on all models 8 months ago
rtmo-t.fp16.onnx

13.7 MB
LFS

Reperform shape inference on all models 8 months ago
rtmo-t.fp16_batch1.onnx

13.7 MB
LFS

Reperform shape inference on all models 8 months ago
rtmo-t.fp16_batch2.onnx

13.7 MB
LFS

Reperform shape inference on all models 8 months ago
rtmo-t.fp16_batch4.onnx

13.7 MB
LFS

Reperform shape inference on all models 8 months ago
rtmo-t.onnx

27.4 MB
LFS

Make ONNX models compatible with ONNXruntime's TensorrtExecutionProvider 9 months ago
rtmo-t_batch1.onnx

27.4 MB
LFS

Reperform shape inference on all models 8 months ago
rtmo-t_batch2.onnx

27.4 MB
LFS

Reperform shape inference on all models 8 months ago
rtmo-t_batch4.onnx

27.4 MB
LFS

Reperform shape inference on all models 8 months ago
rtmo_demo.py

2.18 kB

Add option --yolo_nas_pose, used to read YOLO NAS Pose model instead of RTMO 9 months ago
rtmo_demo_batch.py

3.15 kB

Bugfix for batch demo: add padding if remaning number of frames is not equal to batch size 9 months ago
rtmo_gpu.py

17.8 kB

Add option --yolo_nas_pose, used to read YOLO NAS Pose model instead of RTMO 9 months ago
symbolic_shape_infer.py

138 kB

Make ONNX models compatible with ONNXruntime's TensorrtExecutionProvider 9 months ago
symbolic_shape_infer_all.sh

135 Bytes

Make ONNX models compatible with ONNXruntime's TensorrtExecutionProvider 9 months ago