LiWentomng/OrientedRepPoints

quality assessment measure

Opened this issue · 5 comments

想请教一下论文中关于质量评估度量Q的计算,在代码中具体是在哪个py文件里

主要实现 体现在orientedreppoints_head.py 脚本中的
def points_quality_assessment(self, pts_feats, cls_score, pts_pred_init, pts_pred_refine, label, obbox_gt, label_weight, obox_weight, pos_inds):

请问,在orientedReppoint_head.py文件中,pts_coordinate_preds_init_image和 pts_coordinate_preds_refine_image有什么区别吗?都是干什么用的呢?论文中的Q_poc好像在代码中没有体现?
image
十分感谢。

@kuazhangxiaoai pts_coordinate_preds_init_imagepts_coordinate_preds_refine_image 分别是intial stage 和refine stage 预测的点集在图像中实际位置的坐标。 Q_poc 的相关代码,请看我们现在更新的代码。

请问pts_preds_init和pts_preds_refine在哪里计算得到的呢? 有点儿找不到. 它的channel为什么是18呢?
谢谢

@kuazhangxiaoai pts_preds_initpts_preds_refineforward函数中网络的预测,分别对应pts_out_initpts_out_refine,代表着9个点 x, y上的 offset,故channel为18.