修改部分备注

yizt · Apr 14, 2019 · 22bdee3 · 22bdee3
1 parent 6b962f8
commit 22bdee3
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/ctpn/layers/losses.py b/ctpn/layers/losses.py
@@ -14,9 +14,9 @@ def ctpn_cls_loss(predict_cls_ids, true_cls_ids, indices):
     ctpn分类损失
     :param predict_cls_ids: 预测的anchors类别，(batch_num,anchors_num,2) fg or bg
     :param true_cls_ids:实际的anchors类别，(batch_num,rpn_train_anchors,(class_id,tag))
-             tag 1：正样本，0：负样本，-1 padding
+             tag 1：正负样本，0 padding
     :param indices: 正负样本索引，(batch_num,rpn_train_anchors,(idx,tag))，
-             idx:指定anchor索引位置，tag 1：正样本，0：负样本，-1 padding
+             idx:指定anchor索引位置，tag 1：正样本，-1：负样本，0 padding
     :return:
     """
     # 去除padding

diff --git a/ctpn/layers/target.py b/ctpn/layers/target.py
@@ -98,7 +98,7 @@ def ctpn_target_graph(gt_boxes, gt_cls, anchors, valid_anchors_indices, train_an
     :param positive_ratios
     :param max_gt_num
     :return:
-    deltas:[train_anchors_num, (dy,dh,dx,tag)],anchor边框回归目标,tag=1为正样本,tag=0为padding
+    deltas:[train_anchors_num, (dy,dh,dx,tag)],anchor边框回归目标,tag=1为正负样本,tag=0为padding
     class_id:[train_anchors_num,(class_id,tag)]
     indices: [train_anchors_num,(anchors_index,tag)] tag=1为正样本,tag=0为padding,-1为负样本
     """