SPengLiang / WeakM3D

[ICLR 2022] WeakM3D: Towards Weakly Supervised Monocular 3D Object Detection.
32 stars 3 forks source link

'''orient'''计算 #4

Closed aqr1961 closed 1 year ago

aqr1961 commented 1 year ago

作者您好: 在dataloader/kitti_dataloader.py文件中,大约190行左右的位置,KITTI3D_Object_Dataset_Raw_RoI_Lidar(KITTI3D_Object_Dataset)类中有这样的一段代码,作用应该是计算一组样本点之间的斜率。 '''orient''' orient_set = [(i[1] - j[1]) / (i[0] - j[0]) for j in depth_points_np_xz for i in depth_points_np_xz] 但是,这里为什么不区分i和j是同一个点的情况呢,会不会出现除0错误呢?请问是否应该将原始代码这样修改呢? orient_set = [] for j_idx, j in enumerate(depth_points_np_xz): for i_idx, i in enumerate(depth_points_np_xz): if i_idx != j_idx: # 避免选择同一对点 if i[0] != j[0]: # 避免除以零错误 orient = (i[1] - j[1]) / (i[0] - j[0]) orient_set.append(orient)