Push model using huggingface_hub.
Browse files- config.json +57 -7
- model.safetensors +1 -1
config.json
CHANGED
@@ -1,11 +1,61 @@
|
|
1 |
{
|
2 |
-
"
|
3 |
-
"
|
4 |
-
"
|
5 |
-
"
|
6 |
-
|
|
|
|
|
|
|
7 |
},
|
8 |
-
"
|
9 |
"num_classes": 1,
|
10 |
-
"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
11 |
}
|
|
|
1 |
{
|
2 |
+
"accelerator": "auto",
|
3 |
+
"annotations_xml": null,
|
4 |
+
"architecture": "deformable_detr",
|
5 |
+
"batch_size": 1,
|
6 |
+
"devices": "auto",
|
7 |
+
"model": {
|
8 |
+
"name": "joshvm/deepforest-detr",
|
9 |
+
"revision": "main"
|
10 |
},
|
11 |
+
"nms_thresh": 0.05,
|
12 |
"num_classes": 1,
|
13 |
+
"patch_overlap": 0.05,
|
14 |
+
"patch_size": 400,
|
15 |
+
"path_to_raster": null,
|
16 |
+
"path_to_rgb": null,
|
17 |
+
"predict": {
|
18 |
+
"pin_memory": false
|
19 |
+
},
|
20 |
+
"rgb_dir": null,
|
21 |
+
"score_thresh": 0.25,
|
22 |
+
"train": {
|
23 |
+
"csv_file": null,
|
24 |
+
"epochs": 1,
|
25 |
+
"fast_dev_run": false,
|
26 |
+
"lr": 0.001,
|
27 |
+
"preload_images": false,
|
28 |
+
"root_dir": null,
|
29 |
+
"scheduler": {
|
30 |
+
"params": {
|
31 |
+
"T_max": 10,
|
32 |
+
"cooldown": 0,
|
33 |
+
"eps": 1e-08,
|
34 |
+
"eta_min": 1e-05,
|
35 |
+
"factor": 0.1,
|
36 |
+
"gamma": 0.1,
|
37 |
+
"lr_lambda": "0.95 ** epoch",
|
38 |
+
"milestones": [
|
39 |
+
50,
|
40 |
+
100
|
41 |
+
],
|
42 |
+
"min_lr": 0,
|
43 |
+
"mode": "min",
|
44 |
+
"patience": 10,
|
45 |
+
"step_size": 30,
|
46 |
+
"threshold": 0.0001,
|
47 |
+
"threshold_mode": "rel"
|
48 |
+
},
|
49 |
+
"type": null
|
50 |
+
}
|
51 |
+
},
|
52 |
+
"validation": {
|
53 |
+
"csv_file": null,
|
54 |
+
"iou_threshold": 0.4,
|
55 |
+
"preload_images": false,
|
56 |
+
"root_dir": null,
|
57 |
+
"size": null,
|
58 |
+
"val_accuracy_interval": 20
|
59 |
+
},
|
60 |
+
"workers": 0
|
61 |
}
|
model.safetensors
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
size 160684940
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:bbe49294868f65432ef086b847ce6845e66f95f60928ec91ff8fd22263359fbd
|
3 |
size 160684940
|