search

naver / dust3r

DUSt3R: Geometric 3D Vision Made Easy
https://dust3r.europe.naverlabs.com/
Other
5.03k stars 547 forks source link

preprocess_co3d.py can't work #162

Open lulcai opened 1 month ago

lulcai commented 1 month ago

When I run the following command python3 datasets_preprocess/preprocess_co3d.py --co3d_dir data/co3d_subset --output_dir data/co3d_subset_processed --single_sequence_subset error occurred ValueError: The truth value of an array with more than one element is ambiguous. Use a.any() or a.all()

dream-in-night commented 1 month ago

@lulcai 在resize的时候，传入的参数应该是分辨率的元组，改一下就行了，加个tuple就OK了 https://github.com/naver/dust3r/blob/69192aa322d279438390c109b8b85d5b859b5cdd/dust3r/datasets/utils/cropping.py#L75 如下，加一个tuple 🤞 image = image.resize(tuple(output_resolution), resample=lanczos if scale_final < 1 else bicubic)