Hi,
Using the default config provided in the samples dir I get this error:
ValueError: second_stage_batch_size should be no greater than first_stage_max_proposals.
Edit: Using the new faster_rcnn_nas pre trained model.
Thanks
Matt
Adding
second_stage_batch_size: 49
to the config fixed this issue
@mattryles “Adding second_stage_batch_size: 49 to the config fixed this issue”
Adding to the config what position?
}
first_stage_nms_score_threshold: 0.0
first_stage_nms_iou_threshold: 0.7
first_stage_max_proposals: 50
second_stage_batch_size: 49 ----------------------here?
first_stage_localization_loss_weight: 2.0
first_stage_objectness_loss_weight: 1.0
initial_crop_size: 17
maxpool_kernel_size: 1
maxpool_stride: 1
second_stage_box_predictor {
mask_rcnn_box_predictor {
after second_stage_classification_loss_weight: 1.0
I am getting another error even after i add this line.
TypeError: separable_convolution2d() got an unexpected keyword argument 'data_format'
Try to reduce the height and width of image_resizer, in my case, I reduced them from 1200 to 600.
Most helpful comment
after second_stage_classification_loss_weight: 1.0