facebookresearch / ContrastiveSceneContexts

Code for CVPR 2021 oral paper "Exploring Data-Efficient 3D Scene Understanding with Contrastive Scene Contexts"
MIT License
221 stars 28 forks source link

Confuse about compute_partitions_fast in shape context.py #6

Closed mahaoxiang822 closed 3 years ago

mahaoxiang822 commented 3 years ago

pretrain/contrastive_scene_contexts/lib/shape_context.py Line 89-90

maskUp = rel_trans[:,:,2] > self.r1 maskDown = rel_trans[:,:,2] > self.r1

maskUp and maskDown are same. Is there existing some mistakes?

Sekunde commented 3 years ago

Hi, Haoxiang, Thanks for pointing it out. It was due to some copy-paste error during the code refactoring, and we will fix it soon.