def _add_fast_rcnn_head(model, add_roi_box_head_func, blob_in, dim_in,
spatial_scale_in):
"""Add a Fast R-CNN head to the model."""
blob_frcn, dim_frcn = add_roi_box_head_func(model, blob_in, dim_in,
spatial_scale_in)
fast_rcnn_heads.add_fast_rcnn_outputs(model, blob_frcn, dim_frcn)
if model.train:
loss_gradients = fast_rcnn_heads.add_fast_rcnn_losses(model)
else:
loss_gradients = None
return loss_gradients
def _add_fast_rcnn_head(
model, add_roi_box_head_func, blob_in, dim_in, spatial_scale_in
):
"""Add a Fast R-CNN head to the model."""
blob_frcn, dim_frcn = add_roi_box_head_func(
model, blob_in, dim_in, spatial_scale_in
)
fast_rcnn_heads.add_fast_rcnn_outputs(model, blob_frcn, dim_frcn)
if model.train:
loss_gradients = fast_rcnn_heads.add_fast_rcnn_losses(model)
else:
loss_gradients = None
return loss_gradients
def _add_fast_rcnn_head(model, add_roi_box_head_func, blob_in, dim_in,
spatial_scale_in):
"""Add a Fast R-CNN head to the model."""
# Add the fast-rcnn head
blob_frcn, dim_frcn = add_roi_box_head_func(model, blob_in, dim_in,
spatial_scale_in)
# Add the fast-rcnn output
fast_rcnn_heads.add_fast_rcnn_outputs(model, blob_frcn, dim_frcn)
# Add the fast-rcnn loss
''' seg-eve '''
if model.train and not cfg.TRAIN.TRAIN_MASK_HEAD_ONLY:
''' seg-eve '''
loss_gradients = fast_rcnn_heads.add_fast_rcnn_losses(model)
else:
loss_gradients = None
return loss_gradients
