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

Small updates.

Browse Source
This commit is contained in:
Daniel
2024-12-19 13:52:44 +01:00
parent 19bb1a6372
commit 2ad20549f2
2 changed files with 8 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
README.md
View File

@ -27,6 +27,11 @@ Fast triangulation of multiple persons from multiple camera views.
./run_container.sh ./run_container.sh
``` ```
- Build triangulator:
```bash
cd /RapidPoseTriangulation/swig/ && make all && cd ../tests/ && python3 test_interface.py && cd ..
```
- Test with samples: - Test with samples:
```bash ```bash
python3 /RapidPoseTriangulation/scripts/test_triangulate.py python3 /RapidPoseTriangulation/scripts/test_triangulate.py
@ -37,11 +42,3 @@ Fast triangulation of multiple persons from multiple camera views.
export CUDA_VISIBLE_DEVICES=0 export CUDA_VISIBLE_DEVICES=0
python3 /RapidPoseTriangulation/scripts/test_skelda_dataset.py python3 /RapidPoseTriangulation/scripts/test_skelda_dataset.py
``` ```
<br>
## Debugging
```bash
cd /RapidPoseTriangulation/swig/ && make all && cd ../tests/ && python3 test_interface.py && cd ..
```

3
scripts/utils_2d_pose_ort.py
View File

@ -49,8 +49,9 @@ class BaseModel(ABC):
}, },
) )
) )
if usegpu and "CUDAExecutionProvider" in providers: elif usegpu and "CUDAExecutionProvider" in providers:
self.providers.append("CUDAExecutionProvider") self.providers.append("CUDAExecutionProvider")
else:
self.providers.append("CPUExecutionProvider") self.providers.append("CPUExecutionProvider")
print("Using providers:", self.providers) print("Using providers:", self.providers)