1. 22 Mar, 2021 1 commit
  2. 22 Jun, 2020 1 commit
    • Tong He's avatar
      [Model] PointNet and PointNet++ for point cloud (#1510) · 558673e1
      Tong He authored
      
      
      * commit patch
      
      * commit patch
      
      * pointnet basic
      
      * fix data
      
      * reorg
      
      * reorg
      
      * temp status
      
      * remove validate set
      
      * add partseg data and model
      
      * partseg miou
      
      * clean up
      
      * fix loss
      
      * network definition match paper
      
      * fix
      
      * fix miou
      
      * update data format
      
      * fix
      
      * fix
      
      * working pointnet ssg cls
      
      * avoid some pytorch bug
      
      * fix script
      
      * update hyperparams
      
      * add msg module
      
      * try different dataset
      
      * update new dataset info
      
      * quick fix to subgraph
      
      * fix speed
      
      * update training
      
      * update
      
      * fix bs
      
      * update docstring
      
      * update
      
      * update
      
      * remove parallel reduction in fps
      
      * switch to kernel fps, training is 30% faster
      Co-authored-by: default avatarUbuntu <ubuntu@ip-172-31-20-181.us-west-2.compute.internal>
      558673e1
  3. 21 Jun, 2020 1 commit
    • Tong He's avatar
      [Op] Farthest Point Sampler in Cpp and CUDA (#1630) · 3d47693b
      Tong He authored
      * working framework without actual algorithm logic
      
      * rename
      
      * fix
      
      * fps passes compilation
      
      * correct algorithm
      
      * add cuda implementation
      
      * update random start
      
      * before refactor
      
      * pass compilation but cuda not working
      
      * working
      
      * code working, will add docstring
      
      * add mxnet support
      
      * update docstring
      
      * update doc and test
      
      * cpplint
      
      * cpcplint
      
      * pylint
      
      * temporary fix
      
      * fix for win64
      
      * fix unitetest
      
      * fix
      
      * fix
      
      * remove comment
      
      * move to geometry package
      
      * remove redundant include
      
      * add docstrings and comments
      
      * add proof
      
      * add validity check
      3d47693b