ylabbe / cosypose

Code for "CosyPose: Consistent multi-view multi-object 6D pose estimation", ECCV 2020.
MIT License
301 stars 89 forks source link

I can't run some scripts in README. #63

Closed a1n4s8 closed 3 years ago

a1n4s8 commented 3 years ago

1.

python -m cosypose.scripts.convert_models_to_urdf --models=ycbv

It did not work, so I fix model in command to ycbv.bop, following to make_object_dataset.py than it work well.

Can I fix like that?

2.

export CUDA_VISIBLE_DEVICES=0
python -m cosypose.scripts.run_dataset_recording --config tless --local
python -m cosypose.scripts.run_dataset_recording --config ycbv --local

below is --config tless log

distributed.worker - WARNING - Compute Failed
Function:  record_chunk
args:      ()
kwargs:    {'ds_dir': PosixPath('/cosypose/local_data/synt_datasets/tless-1M'), 'seed': 174, 'n_frames': 100, 'scene_cls': 'cosypose.recording.bop_recording_scene.BopRecordingScene', 'scene_kwargs': {'gpu_renderer': True, 'texture_ds': 'shapenet', 'domain_randomization': True, 'n_objects_interval': (3, 9), 'proba_falling': 0.5, 'border_check': False, 'n_textures_cache': 100, 'objects_xyz_interval': ((-0.25, -0.25, 0.0), (0.25, 0.25, 0.25)), 'focal_interval': ((1060, 1060), (1080, 1080)), 'urdf_ds': 'tless.cad', 'resolution': (720, 540), 'textures_on_objects': True, 'camera_distance_interval': (0.65, 0.94), 'seed': 174}}
Exception: AssertionError()

At first, ycbv was FileNotFoundError, than I fix dir name /local_data/urdfs/ycbv.bop to /local_data/urdfs/ycbv.

than it same as tless.

How can I do?

a1n4s8 commented 3 years ago

solved 2. missing export CUDA_VISIBLE_DEVICES=0