【睿慕课点云处理】第九章-基于深度学习的点云分割方法

最新推荐文章于 2024-06-14 22:00:50 发布

Tech沉思录

最新推荐文章于 2024-06-14 22:00:50 发布

阅读量988

点赞数 1

分类专栏：深度学习 PointCloud 文章标签：深度学习 dgcnn

本文链接：https://blog.youkuaiyun.com/suyunzzz/article/details/107570868

版权

课程

课程汇总

作业

在这里插入图片描述

代码注释（tensorflow版本）

github

在这里插入图片描述

EdgeConv修改

增加余弦距离

def get_edge_feature(point_cloud, nn_idx, k=20):
  """Construct edge feature for each point
  Args:
    point_cloud: (batch_size, num_points, 1, num_dims)
    nn_idx: (batch_size, num_points, k)
    k: int

  Returns:
    edge features: (batch_size, num_points, k, num_dims)
  """
  og_batch_size = point_cloud.get_shape().as_list()[0]
  point_cloud = tf.squeeze(point_cloud)         # B*N*num_dims


  if og_batch_size == 1:
    point_cloud = tf.expand_dims(point_cloud