blaz-r commited on
Commit
4e6fbf2
·
verified ·
1 Parent(s): f826285

Push model using huggingface_hub.

Browse files
Files changed (3) hide show
  1. README.md +10 -0
  2. config.json +125 -0
  3. model.safetensors +3 -0
README.md ADDED
@@ -0,0 +1,10 @@
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ tags:
3
+ - model_hub_mixin
4
+ - pytorch_model_hub_mixin
5
+ ---
6
+
7
+ This model has been pushed to the Hub using the [PytorchModelHubMixin](https://huggingface.co/docs/huggingface_hub/package_reference/mixins#huggingface_hub.PyTorchModelHubMixin) integration:
8
+ - Code: https://github.com/blaz-r/BTC-change-detection
9
+ - Paper: https://arxiv.org/abs/2507.03367
10
+ - Docs: https://github.com/blaz-r/BTC-change-detection
config.json ADDED
@@ -0,0 +1,125 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "config": {
3
+ "ckpt_path": null,
4
+ "config_tag": "BTC-B",
5
+ "data": {
6
+ "batch_size": 32,
7
+ "data_path": "./datasets",
8
+ "dataset": "levir",
9
+ "img_size": 256,
10
+ "load_in_mem": null,
11
+ "num_workers": 8,
12
+ "pin_memory": true,
13
+ "use_hf": true
14
+ },
15
+ "decoder": {
16
+ "train": {
17
+ "class_path": "models.modules.upernet.UperNetHead",
18
+ "init_args": {
19
+ "hidden_size": 512,
20
+ "load_weights": null,
21
+ "lr": null,
22
+ "out_channels": 1,
23
+ "transfer": true,
24
+ "use_auxfcn": false,
25
+ "weight_decay": null
26
+ }
27
+ }
28
+ },
29
+ "dev": false,
30
+ "devices": "auto",
31
+ "diff": {
32
+ "class_path": "models.modules.simple_diff.SubDiff",
33
+ "init_args": {
34
+ "lr": null,
35
+ "norm": false,
36
+ "transfer": true,
37
+ "use_in_pretrain": false,
38
+ "weight_decay": null
39
+ }
40
+ },
41
+ "encoder": {
42
+ "class_path": "models.modules.HFBackbone",
43
+ "init_args": {
44
+ "lr": null,
45
+ "lr_adjustments": false,
46
+ "model_name": "facebook/mask2former-swin-base-IN21k-cityscapes-semantic",
47
+ "pretrain_args": null,
48
+ "return_layers": [
49
+ 1,
50
+ 2,
51
+ 3,
52
+ 4
53
+ ],
54
+ "train_args": {
55
+ "finetune": true,
56
+ "pretrained": true
57
+ },
58
+ "transfer": true,
59
+ "weight_decay": null
60
+ }
61
+ },
62
+ "eval_only": true,
63
+ "out_proc": null,
64
+ "pre_diff": null,
65
+ "res_path": "./results",
66
+ "seed": 42,
67
+ "tag": "",
68
+ "train": {
69
+ "base_lr": 0.0001,
70
+ "ckpt": null,
71
+ "epochs": 100,
72
+ "grad_clip_val": null,
73
+ "layer_decay": 0.99,
74
+ "loss": [
75
+ "dice"
76
+ ],
77
+ "lr_scheduler": {
78
+ "gamma": null,
79
+ "ratio_t0": 1.0,
80
+ "type": "cosine",
81
+ "warmup_epoch_ratio": null,
82
+ "warmup_lr_ratio": null
83
+ },
84
+ "transforms": [
85
+ {
86
+ "VerticalFlip": {
87
+ "p": 0.3
88
+ }
89
+ },
90
+ {
91
+ "HorizontalFlip": {
92
+ "p": 0.3
93
+ }
94
+ },
95
+ {
96
+ "Rotate": {
97
+ "limit": [
98
+ -90,
99
+ 90
100
+ ],
101
+ "p": 0.3
102
+ }
103
+ },
104
+ {
105
+ "Normalize": {
106
+ "mean": [
107
+ 0.485,
108
+ 0.456,
109
+ 0.406
110
+ ],
111
+ "std": [
112
+ 0.229,
113
+ 0.224,
114
+ 0.225
115
+ ]
116
+ }
117
+ }
118
+ ],
119
+ "val_freq": 5,
120
+ "weight_decay": 0.0001
121
+ },
122
+ "vis_path": true,
123
+ "wandb_proj": "None"
124
+ }
125
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:02c14a31adce5acfa29cfa0e019a9c94a1c8a66b305abe5500ff260dac0b0395
3
+ size 484579612