hxwork / H2ONet_Pytorch

[CVPR 2023] H2ONet: Hand-Occlusion-and-Orientation-aware Network for Real-time 3D Hand Mesh Reconstruction, Pytorch implementation.
MIT License
54 stars 1 forks source link

where can i download these two files in test.py ? #8

Open chouchunan opened 2 weeks ago

chouchunan commented 2 weeks ago

template_fp = "model/h2onet/template/template.ply" transform_fp = "model/h2onet/template/transform.pkl"

hxwork commented 2 weeks ago

These files can be generated by this function.

chouchunan commented 2 weeks ago

(h2onet_py39) user@user-System-Product-Name:/media/user/test/H2ONet_Pytorch-main$ python test.py --model_dir=./experiment/multi_frame_dexycb/stage_2 --resume=./experiment/multi_frame_dexycb/stage_2/dexycb_h2onet_mf.pth /media/user/test/H2ONet_Pytorch-main/common/utils/transforms.py:3: UserWarning: A NumPy version >=1.22.4 and <2.3.0 is required for this version of SciPy (detected version 1.21.6) from scipy.linalg import orthogonal_procrustes 2024-06-18 21:28:44 root Output and logs will be saved to ./experiment/multi_frame_dexycb/stage_2/test.log 2024-06-18 21:28:44 root Using GPU ids: [0] 2024-06-18 21:28:44 data_loader.data_loader Dataset: DEX_YCB_MF 2024-06-18 21:28:44 data_loader.transforms Train transforms: ToTensor 2024-06-18 21:28:44 data_loader.transforms Val/Test transforms: ToTensor loading annotations into memory... Done (t=35.34s) creating index... index created! loading annotations into memory... Done (t=5.83s) creating index... index created! /home/user/anaconda3/envs/h2onet_py39/lib/python3.9/site-packages/torch/utils/data/dataloader.py:560: UserWarning: This DataLoader will create 20 worker processes in total. Our suggested max number of worker in current system is 12, which is smaller than what this DataLoader is going to create. Please be aware that excessive worker creation might get DataLoader running slow or even freeze, lower the worker number to avoid potential slowness/freeze if necessary. warnings.warn(_create_warning_msg( 2024-06-18 21:30:02 model.modules.models.densestack Load pre-trained weight: densestack.pth Generating transform matrices... Traceback (most recent call last): File "/media/user/test/H2ONet_Pytorch-main/test.py", line 107, in main(cfg) File "/media/user/test/H2ONet_Pytorch-main/test.py", line 81, in main model = fetch_model(cfg) File "/media/user/test/H2ONet_Pytorch-main/model/model.py", line 215, in fetch_model model = H2ONet_MF(cfg) File "/media/user/test/H2ONet_Pytorch-main/model/model.py", line 121, in init spiralindices, , up_transform, tmp = spiral_tramsform(transform_fp, template_fp, ds_factors=[2, 2, 2, 2], seq_length=[9, 9, 9, 9], dilation=[1, 1, 1, 1]) File "/media/user/test/H2ONet_Pytorch-main/model/modules/utils/read.py", line 37, in spiral_tramsform mesh = Mesh(filename=template_fp) File "/home/user/anaconda3/envs/h2onet_py39/lib/python3.9/site-packages/psbody/mesh/mesh.py", line 67, in init self.load_from_file(filename) File "/home/user/anaconda3/envs/h2onet_py39/lib/python3.9/site-packages/psbody/mesh/mesh.py", line 461, in load_from_file serialization.load_from_file(self, filename) File "/home/user/anaconda3/envs/h2onet_py39/lib/python3.9/site-packages/psbody/mesh/serialization/serialization.py", line 412, in load_from_file self.load_from_ply(filename) File "/home/user/anaconda3/envs/h2onet_py39/lib/python3.9/site-packages/psbody/mesh/mesh.py", line 464, in load_from_ply serialization.load_from_ply(self, filename) File "/home/user/anaconda3/envs/h2onet_py39/lib/python3.9/site-packages/psbody/mesh/serialization/serialization.py", line 431, in load_from_ply from psbody.mesh.serialization import plyutils ImportError: cannot import name 'plyutils' from 'psbody.mesh.serialization' (/home/user/anaconda3/envs/h2onet_py39/lib/python3.9/site-packages/psbody/mesh/serialization/init.py)

how can i solve this problem? thanks