Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion .github/workflows/conda-test.yml
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,7 @@ jobs:
shell: bash -el {0}
strategy:
matrix:
python-version: ["3.10", "3.12"]
python-version: ["3.10", "3.13"]
steps:
- uses: actions/checkout@v4
with:
Expand Down
2 changes: 1 addition & 1 deletion .github/workflows/pip-test.yml
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@ jobs:
runs-on: ubuntu-latest
strategy:
matrix:
python-version: ["3.10", "3.12"]
python-version: ["3.10", "3.13"]
steps:
- uses: actions/checkout@v4
with:
Expand Down
2 changes: 1 addition & 1 deletion .github/workflows/uv-test.yml
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@ jobs:
runs-on: ubuntu-latest
strategy:
matrix:
python-version: ["3.10", "3.12"]
python-version: ["3.10", "3.13"]
steps:
- uses: actions/checkout@v4
with:
Expand Down
2 changes: 1 addition & 1 deletion happypose/toolbox/datasets/augmentations.py
Original file line number Diff line number Diff line change
Expand Up @@ -533,7 +533,7 @@ def __call__(self, obs: SceneObservation) -> SceneObservation:
new_obs.rgb = np.array(rgb_pil, dtype=np.uint8)
new_obs.segmentation = np.array(segmentation_pil, dtype=np.int32)
if depth_pil is not None:
new_obs.depth = np.array(depth_pil, dtype=np.float_)
new_obs.depth = np.array(depth_pil, dtype=np.float64)

# Update modal object bounding boxes
dets_gt = make_detections_from_segmentation(new_obs.segmentation[None])[0]
Expand Down
6 changes: 3 additions & 3 deletions happypose/toolbox/lib3d/symmetries.py
Original file line number Diff line number Diff line change
Expand Up @@ -33,15 +33,15 @@ class ContinuousSymmetry:
See https://github.com/thodan/bop_toolkit/blob/master/bop_toolkit_lib/misc.py
"""

offset: npt.NDArray[np.float_]
axis: npt.NDArray[np.float_]
offset: npt.NDArray[np.float64]
axis: npt.NDArray[np.float64]


@dataclass
class DiscreteSymmetry:
"""pose: (4, 4) homogeneous matrix."""

pose: npt.NDArray[np.float_]
pose: npt.NDArray[np.float64]


def make_symmetries_poses(
Expand Down
8 changes: 4 additions & 4 deletions pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -65,12 +65,12 @@ version = "0.1.0"

[project.optional-dependencies]
cpu = [
"torch>=2.5.1",
"torchvision>=0.20.1"
"torch>=2.6",
"torchvision>=0.21"
]
cu124 = [
"torch>=2.5.1",
"torchvision>=0.20.1"
"torch>=2.6",
"torchvision>=0.21"
]
evaluation = ["cmeel-teaser-plusplus>=2.0"]
pypi = [
Expand Down
147 changes: 75 additions & 72 deletions requirements/base.txt
Original file line number Diff line number Diff line change
Expand Up @@ -5,131 +5,134 @@
absl-py==2.1.0
addict==2.4.0
antlr4-python3-runtime==4.9.3
anyio==4.6.2.post1
asttokens==2.4.1
attrs==24.2.0
beautifulsoup4==4.12.3
bokeh==3.6.1
bop-toolkit-lib @ git+https://github.com/agimus-project/bop_toolkit@53b6e2e335c4e6aa76e5db34e809506ee3f4cf16
anyio==4.8.0
asttokens==3.0.0 ; python_full_version < '3.11'
attrs==25.3.0
beautifulsoup4==4.13.3
bokeh==3.7.0
bop-toolkit-lib @ git+https://github.com/agimus-project/bop_toolkit@7c28057c741afed432b521735e184741e25b8d73
braceexpand==0.1.7
certifi==2024.8.30
cffi==1.17.1 ; (implementation_name == 'pypy' and os_name != 'nt') or (implementation_name == 'pypy' and platform_machine == 'aarch64' and sys_platform == 'linux') or (implementation_name != 'pypy' and platform_machine == 'aarch64' and sys_platform == 'linux' and extra == 'extra-9-happypose-cpu' and extra == 'extra-9-happypose-cu124') or (implementation_name == 'pypy' and sys_platform == 'darwin') or (implementation_name != 'pypy' and sys_platform == 'darwin' and extra == 'extra-9-happypose-cpu' and extra == 'extra-9-happypose-cu124') or (os_name == 'nt' and platform_machine != 'aarch64' and sys_platform == 'linux') or (os_name == 'nt' and sys_platform != 'darwin' and sys_platform != 'linux') or (os_name != 'nt' and extra == 'extra-9-happypose-cpu' and extra == 'extra-9-happypose-cu124')
certifi==2025.1.31
cffi==1.17.1 ; implementation_name == 'pypy'
cfgv==3.4.0
charset-normalizer==3.4.0
colorama==0.4.6 ; sys_platform == 'win32' or (extra == 'extra-9-happypose-cpu' and extra == 'extra-9-happypose-cu124')
charset-normalizer==3.4.1
colorama==0.4.6 ; python_full_version < '3.11' and sys_platform == 'win32'
contourpy==1.3.1
coverage==7.6.8
coverage==7.6.12
cycler==0.12.1
cython==3.0.11
decorator==5.1.1
cython==3.0.12
decorator==5.2.1 ; python_full_version < '3.11'
distlib==0.3.9
exceptiongroup==1.2.2 ; python_full_version < '3.11' or (extra == 'extra-9-happypose-cpu' and extra == 'extra-9-happypose-cu124')
executing==2.1.0
filelock==3.16.1
fonttools==4.55.0
exceptiongroup==1.2.2 ; python_full_version < '3.11'
executing==2.2.0 ; python_full_version < '3.11'
filelock==3.18.0
fonttools==4.56.0
freetype-py==2.5.1
fsspec==2024.10.0
grpcio==1.68.0
fsspec==2025.3.0
grpcio==1.71.0
h11==0.14.0
hsluv==5.0.4
httpcore==1.0.7
httpx==0.27.2
identify==2.6.2
httpx==0.28.1
identify==2.6.9
idna==3.10
imageio==2.36.0
imageio==2.37.0
iniconfig==2.0.0
ipython==8.29.0
jedi==0.19.2
jinja2==3.1.4
ipython==8.34.0 ; python_full_version < '3.11'
ipython==9.0.2 ; python_full_version >= '3.11'
ipython-pygments-lexers==1.1.1 ; python_full_version >= '3.11'
jedi==0.19.2 ; python_full_version < '3.11'
jinja2==3.1.6
joblib==1.4.2
jsonpatch==1.33
jsonpointer==3.0.0
kiwisolver==1.4.7
kiwisolver==1.4.8
lazy-loader==0.4
markdown==3.7
markupsafe==3.0.2
matplotlib==3.9.2
matplotlib-inline==0.1.7
matplotlib==3.10.1
matplotlib-inline==0.1.7 ; python_full_version < '3.11'
meshcat==0.3.2
mypy-extensions==1.0.0
narwhals==1.30.0
networkx==3.4.2
nodeenv==1.9.1
numpy==1.26.4
numpy==2.2.3
omegaconf==2.3.0
outcome==1.3.0.post0
packaging==24.2
panda3d==1.10.15
pandas==2.2.3
parso==0.8.4
pexpect==4.9.0 ; (sys_platform != 'emscripten' and sys_platform != 'win32') or (sys_platform == 'emscripten' and extra == 'extra-9-happypose-cpu' and extra == 'extra-9-happypose-cu124') or (sys_platform == 'win32' and extra == 'extra-9-happypose-cpu' and extra == 'extra-9-happypose-cu124')
pillow==11.0.0
parso==0.8.4 ; python_full_version < '3.11'
pexpect==4.9.0 ; python_full_version < '3.11' and sys_platform != 'emscripten' and sys_platform != 'win32'
pillow==11.1.0
platformdirs==4.3.6
pluggy==1.5.0
plyfile==1.1
pre-commit==4.0.1
prompt-toolkit==3.0.48
protobuf==5.28.3
psutil==6.1.0
ptyprocess==0.7.0 ; (sys_platform != 'emscripten' and sys_platform != 'win32') or (sys_platform == 'emscripten' and extra == 'extra-9-happypose-cpu' and extra == 'extra-9-happypose-cu124') or (sys_platform == 'win32' and extra == 'extra-9-happypose-cpu' and extra == 'extra-9-happypose-cu124')
pure-eval==0.2.3
pre-commit==4.1.0
prompt-toolkit==3.0.50 ; python_full_version < '3.11'
protobuf==6.30.1
psutil==7.0.0
ptyprocess==0.7.0 ; python_full_version < '3.11' and sys_platform != 'emscripten' and sys_platform != 'win32'
pure-eval==0.2.3 ; python_full_version < '3.11'
pybind11==2.13.6
pybullet==3.2.6
pycparser==2.22 ; (implementation_name == 'pypy' and os_name != 'nt') or (implementation_name == 'pypy' and platform_machine == 'aarch64' and sys_platform == 'linux') or (implementation_name != 'pypy' and platform_machine == 'aarch64' and sys_platform == 'linux' and extra == 'extra-9-happypose-cpu' and extra == 'extra-9-happypose-cu124') or (implementation_name == 'pypy' and sys_platform == 'darwin') or (implementation_name != 'pypy' and sys_platform == 'darwin' and extra == 'extra-9-happypose-cpu' and extra == 'extra-9-happypose-cu124') or (os_name == 'nt' and platform_machine != 'aarch64' and sys_platform == 'linux') or (os_name == 'nt' and sys_platform != 'darwin' and sys_platform != 'linux') or (os_name != 'nt' and extra == 'extra-9-happypose-cpu' and extra == 'extra-9-happypose-cu124')
pygments==2.18.0
pyngrok==7.2.1
pybullet==3.2.7
pycparser==2.22 ; implementation_name == 'pypy'
pygments==2.19.1 ; python_full_version < '3.11'
pyngrok==7.2.3
pyopengl==3.1.0
pyparsing==3.2.0
pyparsing==3.2.1
pypng==0.20220715.0
pyre-extensions==0.0.32
pysocks==1.7.1
pytest==8.3.3
pytest==8.3.5
pytest-order==1.3.0
python-dateutil==2.9.0.post0
pytz==2024.2
pytz==2025.1
pyyaml==6.0.2
pyzmq==26.2.0
pyzmq==26.3.0
requests==2.32.3
roma==1.5.1
ruff==0.8.0
scikit-image==0.24.0
scikit-learn==1.5.2
scipy==1.14.1
roma==1.5.2.1
ruff==0.11.0
scikit-image==0.25.2
scikit-learn==1.6.1
scipy==1.15.2
seaborn==0.13.2
selenium==4.26.1
setuptools==75.6.0
simplejson==3.19.3
six==1.16.0
selenium==4.29.0
setuptools==76.0.0
simplejson==3.20.1
six==1.17.0
sniffio==1.3.1
sortedcontainers==2.4.0
soupsieve==2.6
stack-data==0.6.3
stack-data==0.6.3 ; python_full_version < '3.11'
tabulate==0.9.0
tensorboard==2.18.0
tensorboard==2.19.0
tensorboard-data-server==0.7.2
threadpoolctl==3.5.0
tifffile==2024.9.20
tomli==2.1.0 ; python_full_version <= '3.11' or (extra == 'extra-9-happypose-cpu' and extra == 'extra-9-happypose-cu124')
threadpoolctl==3.6.0
tifffile==2025.3.13
tomli==2.2.1 ; python_full_version < '3.11'
torchnet==0.0.4
torchtnt==0.2.4
tornado==6.4.2
tornado==6.4.2 ; sys_platform != 'emscripten'
tqdm==4.67.1
traitlets==5.14.3
traitlets==5.14.3 ; python_full_version < '3.11'
transforms3d==0.4.2
trimesh==4.5.2
trio==0.27.0
trio-websocket==0.11.1
trimesh==4.6.4
trio==0.29.0
trio-websocket==0.12.2
typing-extensions==4.12.2
typing-inspect==0.9.0
tzdata==2024.2
tzdata==2025.1
u-msgpack-python==2.8.0
urllib3==2.2.3
virtualenv==20.27.1
urllib3==2.3.0
virtualenv==20.29.3
visdom==0.2.4
vispy==0.14.3
wcwidth==0.2.13
webdataset==0.2.100
wcwidth==0.2.13 ; python_full_version < '3.11'
webdataset==0.2.111
websocket-client==1.8.0
werkzeug==3.1.3
wsproto==1.2.0
xarray==2024.11.0
xyzservices==2024.9.0
xarray==2025.1.2
xyzservices==2025.1.0
Loading