crosstyan/RapidPoseTriangulation
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Moved image normalization into onnx models.

Browse Source
This commit is contained in:
Daniel
2024-11-29 17:05:43 +01:00
parent 1b5e0c44e3
commit 7b9505ca02
6 changed files with 95 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
scripts/utils_2d_pose_ep.py
View File

@ -16,9 +16,9 @@ def load_model():
# model = ep.TopDown("rtmpose_m", "SimCC", "rtmdet_s")
model = ep.TopDown(
"/RapidPoseTriangulation/extras/mmdeploy/exports/rtmpose-m_384.onnx",
"/RapidPoseTriangulation/extras/mmdeploy/exports/rtmpose-m_384x288_with-norm.onnx",
"SimCC",
"/RapidPoseTriangulation/extras/mmdeploy/exports/rtmdet_nano_320.onnx",
"/RapidPoseTriangulation/extras/mmdeploy/exports/rtmdet-nano_320x320_with-norm.onnx",
conf_threshold=0.3,
iou_threshold=0.3,
warmup=10,