sanghoon / pva-faster-rcnn

Demo code for PVANet
https://arxiv.org/abs/1611.08588
Other
651 stars 241 forks source link

"faster_rcnn_train_test_ft_rcnn_only_plus_comp.pt " some trianing layer missing? #55

Open runningJ opened 7 years ago

runningJ commented 7 years ago

when used this file to train my datasets. I found that some layer missing. layer { 6439 name: 'rpn-data' 6440 type: 'Python' 6441 bottom: 'rpn_cls_score' 6442 bottom: 'gt_boxes' 6443 bottom: 'im_info' 6444 bottom: 'data' 6445 top: 'rpn_labels' 6446 top: 'rpn_bbox_targets' 6447 top: 'rpn_bbox_inside_weights' 6448 top: 'rpn_bbox_outside_weights' 6449 include { phase: TRAIN } 6450 python_param { 6451 module: 'rpn.anchor_target_layer' 6452 layer: 'AnchorTargetLayer' 6453 param_str: "{'feat_stride': 16, 'ratios': [0.333, 0.5, 0.667, 1, 1.5, 2, 3], 'scales': [2, 3, 5, 9, 16, 32]}" 6454 } 6455 } 6456 layer { 6457 name: "rpn_loss_cls" 6458 type: "SoftmaxWithLoss" 6459 bottom: "rpn_cls_score_reshape" 6460 bottom: "rpn_labels" 6461 propagate_down: 1 6462 propagate_down: 0 6463 top: "rpn_loss_cls" 6464 include { phase: TRAIN } 6465 loss_weight: 1 6466 loss_param { ignore_label: -1 normalize: true } 6467 } layer { 6469 name: "rpn_loss_bbox" 6470 type: "SmoothL1Loss" 6471 bottom: "rpn_bbox_pred" 6472 bottom: "rpn_bbox_targets" 6473 bottom: "rpn_bbox_inside_weights" 6474 bottom: "rpn_bbox_outside_weights" 6475 top: "rpn_loss_bbox" 6476 include { phase: TRAIN } 6477 loss_weight: 1 6478 smooth_l1_loss_param { sigma: 3.0 } 6479 }