joshvm commited on
Commit
25f922c
·
verified ·
1 Parent(s): b4a97e0

Push model using huggingface_hub.

Browse files
Files changed (2) hide show
  1. config.json +57 -7
  2. model.safetensors +1 -1
config.json CHANGED
@@ -1,11 +1,61 @@
1
  {
2
- "config_args": null,
3
- "existing_train_dataloader": null,
4
- "existing_val_dataloader": null,
5
- "label_dict": {
6
- "Tree": 0
 
 
 
7
  },
8
- "model": null,
9
  "num_classes": 1,
10
- "transforms": null
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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:926612e876513d9a8b8984f92a03961c6a9735e5d269ce17e5e6416f5324e0d8
3
  size 160684940
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bbe49294868f65432ef086b847ce6845e66f95f60928ec91ff8fd22263359fbd
3
  size 160684940