crater-rcnn / config.json
wdoppenberg's picture
Push model using huggingface_hub.
5d3ab32 verified
raw
history blame contribute delete
920 Bytes
{
"backbone_name": "resnet50",
"bbox_reg_weights": [
1.0,
1.0,
1.0,
1.0
],
"box_batch_size_per_image": 512,
"box_bg_iou_thresh": 0.5,
"box_detections_per_img": 100,
"box_fg_iou_thresh": 0.5,
"box_nms_thresh": 0.5,
"box_positive_fraction": 0.25,
"box_score_thresh": 0.05,
"ellipse_head": null,
"ellipse_loss_normalize": false,
"ellipse_loss_scale": 1.0,
"ellipse_predictor": null,
"ellipse_roi_pool": null,
"image_mean": null,
"image_std": null,
"max_size": 512,
"min_size": 256,
"num_classes": 2,
"rpn_anchor_generator": null,
"rpn_batch_size_per_image": 256,
"rpn_bg_iou_thresh": 0.3,
"rpn_fg_iou_thresh": 0.7,
"rpn_head": null,
"rpn_nms_thresh": 0.7,
"rpn_positive_fraction": 0.5,
"rpn_post_nms_top_n_test": 1000,
"rpn_post_nms_top_n_train": 2000,
"rpn_pre_nms_top_n_test": 1000,
"rpn_pre_nms_top_n_train": 2000,
"rpn_score_thresh": 0.0
}