Training cosypose #228
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
name: Tests with pip | |
on: [ push, pull_request ] | |
jobs: | |
test: | |
runs-on: ubuntu-latest | |
strategy: | |
matrix: | |
python-version: ["3.9", "3.10"] | |
steps: | |
- uses: actions/checkout@v4 | |
with: | |
submodules: 'recursive' | |
- name: Install EGL mesa - required for Panda3D renderer | |
run: sudo apt-get update && sudo apt-get install -qqy libegl1-mesa libegl1-mesa-dev | |
- name: Caching of the happypose installation and data | |
uses: actions/cache@v4 | |
with: | |
path: local_data | |
key: data | |
- uses: actions/setup-python@v5 | |
with: | |
python-version: ${{ matrix.python-version }} | |
- name: Update pip | |
run: pip install -U pip | |
- name: Install happypose | |
run: pip install ".[cpu,evaluation,pypi]" --extra-index-url https://download.pytorch.org/whl/cpu | |
- name: Download pre-trained models required for tests | |
run: | | |
mkdir -p local_data | |
python -m happypose.toolbox.utils.download \ | |
--megapose_models \ | |
--examples \ | |
barbecue-sauce \ | |
--cosypose_models \ | |
detector-bop-hope-pbr--15246 \ | |
coarse-bop-hope-pbr--225203 \ | |
refiner-bop-hope-pbr--955392 | |
- name: Run tests | |
run: python -m unittest |