Small updates.

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

View File

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

View File

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