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