wdoppenberg commited on
Commit
5d3ab32
·
verified ·
1 Parent(s): d78cd0b

Push model using huggingface_hub.

Browse files
Files changed (3) hide show
  1. README.md +10 -0
  2. config.json +38 -0
  3. model.safetensors +3 -0
README.md ADDED
@@ -0,0 +1,10 @@
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: ellipse-rcnn
3
+ tags:
4
+ - model_hub_mixin
5
+ - pytorch_model_hub_mixin
6
+ ---
7
+
8
+ This model has been pushed to the Hub using the [PytorchModelHubMixin](https://huggingface.co/docs/huggingface_hub/package_reference/mixins#huggingface_hub.PyTorchModelHubMixin) integration:
9
+ - Library: https://github.com/wdoppenberg/ellipse-rcnn
10
+ - Docs: https://github.com/wdoppenberg/ellipse-rcnn
config.json ADDED
@@ -0,0 +1,38 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "backbone_name": "resnet50",
3
+ "bbox_reg_weights": [
4
+ 1.0,
5
+ 1.0,
6
+ 1.0,
7
+ 1.0
8
+ ],
9
+ "box_batch_size_per_image": 512,
10
+ "box_bg_iou_thresh": 0.5,
11
+ "box_detections_per_img": 100,
12
+ "box_fg_iou_thresh": 0.5,
13
+ "box_nms_thresh": 0.5,
14
+ "box_positive_fraction": 0.25,
15
+ "box_score_thresh": 0.05,
16
+ "ellipse_head": null,
17
+ "ellipse_loss_normalize": false,
18
+ "ellipse_loss_scale": 1.0,
19
+ "ellipse_predictor": null,
20
+ "ellipse_roi_pool": null,
21
+ "image_mean": null,
22
+ "image_std": null,
23
+ "max_size": 512,
24
+ "min_size": 256,
25
+ "num_classes": 2,
26
+ "rpn_anchor_generator": null,
27
+ "rpn_batch_size_per_image": 256,
28
+ "rpn_bg_iou_thresh": 0.3,
29
+ "rpn_fg_iou_thresh": 0.7,
30
+ "rpn_head": null,
31
+ "rpn_nms_thresh": 0.7,
32
+ "rpn_positive_fraction": 0.5,
33
+ "rpn_post_nms_top_n_test": 1000,
34
+ "rpn_post_nms_top_n_train": 2000,
35
+ "rpn_pre_nms_top_n_test": 1000,
36
+ "rpn_pre_nms_top_n_train": 2000,
37
+ "rpn_score_thresh": 0.0
38
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f925f688fc0a16c7c5a243e2dc649288523ce178fe602e1574470ecab0b72c7f
3
+ size 166679588