lib/roi_data/fast_rcnn_rel.py [500:518]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
        sbj_pos_starts=pos_starts_sbj,
        obj_pos_starts=pos_starts_obj,
        rel_pos_starts=pos_starts_rel,
        sbj_pos_ends=pos_ends_sbj,
        obj_pos_ends=pos_ends_obj,
        rel_pos_ends=pos_ends_rel,
        sbj_neg_starts=neg_starts_sbj,
        obj_neg_starts=neg_starts_obj,
        rel_neg_starts=neg_starts_rel,
        sbj_neg_ends=neg_ends_sbj,
        obj_neg_ends=neg_ends_obj,
        rel_neg_ends=neg_ends_rel)
    if cfg.TRAIN.ADD_LOSS_WEIGHTS:
        blob['rel_pos_weights'] = rel_pos_weights
    if cfg.TRAIN.ADD_LOSS_WEIGHTS_SO:
        blob['sbj_pos_weights'] = sbj_pos_weights
        blob['obj_pos_weights'] = obj_pos_weights

    return blob
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



lib/roi_data/fast_rcnn_rel.py [707:725]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
        sbj_pos_starts=pos_starts_sbj,
        obj_pos_starts=pos_starts_obj,
        rel_pos_starts=pos_starts_rel,
        sbj_pos_ends=pos_ends_sbj,
        obj_pos_ends=pos_ends_obj,
        rel_pos_ends=pos_ends_rel,
        sbj_neg_starts=neg_starts_sbj,
        obj_neg_starts=neg_starts_obj,
        rel_neg_starts=neg_starts_rel,
        sbj_neg_ends=neg_ends_sbj,
        obj_neg_ends=neg_ends_obj,
        rel_neg_ends=neg_ends_rel)
    if cfg.TRAIN.ADD_LOSS_WEIGHTS:
        blob['rel_pos_weights'] = rel_pos_weights
    if cfg.TRAIN.ADD_LOSS_WEIGHTS_SO:
        blob['sbj_pos_weights'] = sbj_pos_weights
        blob['obj_pos_weights'] = obj_pos_weights

    return blob
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



