Hugging Face's logo

anforsm
/
3DEM_TACC_LSD_AFF

Transformers
PyTorch
Safetensors
pytorch_model_hub_mixin
model_hub_mixin
Inference Endpoints
Model card Files Community
anforsm commited on
Commit
a30f6be
•
1 Parent(s): ca60ee6

Push model using huggingface_hub.

Browse files
Files changed (3) hide show
  1. README.md +9 -0
  2. config.json +477 -0
  3. model.safetensors +3 -0
README.md ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ tags:
3
+ - pytorch_model_hub_mixin
4
+ - model_hub_mixin
5
+ ---
6
+
7
+ This model has been pushed to the Hub using ****:
8
+ - Repo: [More Information Needed]
9
+ - Docs: [More Information Needed]
config.json ADDED
@@ -0,0 +1,477 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model": {
3
+ "hf_path": "anforsm/3DEM_TACC_LSD_AFF",
4
+ "hyperparameters": {
5
+ "downsample_factors": [
6
+ [
7
+ 1,
8
+ 2,
9
+ 2
10
+ ],
11
+ [
12
+ 1,
13
+ 2,
14
+ 2
15
+ ],
16
+ [
17
+ 2,
18
+ 2,
19
+ 2
20
+ ]
21
+ ],
22
+ "fmap_inc_factor": 5,
23
+ "in_channels": 1,
24
+ "kernel_size_down": [
25
+ [
26
+ [
27
+ 3,
28
+ 3,
29
+ 3
30
+ ],
31
+ [
32
+ 3,
33
+ 3,
34
+ 3
35
+ ]
36
+ ],
37
+ [
38
+ [
39
+ 3,
40
+ 3,
41
+ 3
42
+ ],
43
+ [
44
+ 3,
45
+ 3,
46
+ 3
47
+ ]
48
+ ],
49
+ [
50
+ [
51
+ 3,
52
+ 3,
53
+ 3
54
+ ],
55
+ [
56
+ 3,
57
+ 3,
58
+ 3
59
+ ]
60
+ ],
61
+ [
62
+ [
63
+ 1,
64
+ 3,
65
+ 3
66
+ ],
67
+ [
68
+ 1,
69
+ 3,
70
+ 3
71
+ ]
72
+ ]
73
+ ],
74
+ "kernel_size_up": [
75
+ [
76
+ [
77
+ 3,
78
+ 3,
79
+ 3
80
+ ],
81
+ [
82
+ 3,
83
+ 3,
84
+ 3
85
+ ]
86
+ ],
87
+ [
88
+ [
89
+ 3,
90
+ 3,
91
+ 3
92
+ ],
93
+ [
94
+ 3,
95
+ 3,
96
+ 3
97
+ ]
98
+ ],
99
+ [
100
+ [
101
+ 3,
102
+ 3,
103
+ 3
104
+ ],
105
+ [
106
+ 3,
107
+ 3,
108
+ 3
109
+ ]
110
+ ]
111
+ ],
112
+ "output_shapes": [
113
+ 10,
114
+ 3
115
+ ]
116
+ },
117
+ "name": "3D Model With LSD",
118
+ "path": "out"
119
+ },
120
+ "pipeline": {
121
+ "_order": [
122
+ "source",
123
+ "normalize",
124
+ "augment",
125
+ "lsd_target",
126
+ "target"
127
+ ],
128
+ "_outputs": [
129
+ "RAW",
130
+ "LABELS",
131
+ "GT_AFFS",
132
+ "AFFS_WEIGHTS",
133
+ "GT_LSDS",
134
+ "LSDS_WEIGHTS",
135
+ "GT_AFFS_MASK",
136
+ "LABELS_MASK"
137
+ ],
138
+ "augment": [
139
+ {
140
+ "elastic_augment": {
141
+ "control_point_spacing": [
142
+ 1,
143
+ 50,
144
+ 50
145
+ ],
146
+ "jitter_sigma": [
147
+ 0,
148
+ 5,
149
+ 5
150
+ ],
151
+ "rotation_interval": [
152
+ 0,
153
+ 1.570796
154
+ ],
155
+ "scale_interval": [
156
+ 0.5,
157
+ 2
158
+ ],
159
+ "subsample": 4
160
+ }
161
+ },
162
+ {
163
+ "simple_augment": {
164
+ "transpose_only": [
165
+ 1,
166
+ 2
167
+ ]
168
+ }
169
+ },
170
+ {
171
+ "noise_augment": {
172
+ "array": "RAW"
173
+ }
174
+ },
175
+ {
176
+ "intensity_augment": {
177
+ "array": "RAW",
178
+ "scale_max": 1.1,
179
+ "scale_min": 0.9,
180
+ "shift_max": 0.1,
181
+ "shift_min": -0.1
182
+ }
183
+ },
184
+ {
185
+ "smooth_array": {
186
+ "array": "RAW"
187
+ }
188
+ },
189
+ {
190
+ "grow_boundary": {
191
+ "labels": "LABELS",
192
+ "only_xy": true
193
+ }
194
+ }
195
+ ],
196
+ "lsd_target": [
197
+ {
198
+ "add_local_shape_descriptor": {
199
+ "descriptor": "GT_LSDS",
200
+ "downsample": 2,
201
+ "labels_mask": "LABELS_MASK",
202
+ "lsds_mask": "LSDS_WEIGHTS",
203
+ "segmentation": "LABELS",
204
+ "sigma": 40
205
+ }
206
+ }
207
+ ],
208
+ "normalize": [
209
+ {
210
+ "normalize": {
211
+ "array": "RAW"
212
+ }
213
+ }
214
+ ],
215
+ "source": [
216
+ [
217
+ [
218
+ {
219
+ "zarr_source": {
220
+ "array_specs": {
221
+ "_labels": {
222
+ "array_spec": {
223
+ "interpolatable": false
224
+ }
225
+ },
226
+ "_labels_mask": {
227
+ "array_spec": {
228
+ "interpolatable": false
229
+ }
230
+ },
231
+ "_raw": {
232
+ "array_spec": {
233
+ "interpolatable": true
234
+ }
235
+ }
236
+ },
237
+ "datasets": {
238
+ "_labels": "labels/s0",
239
+ "_labels_mask": "labels_mask/s0",
240
+ "_raw": "raw/s0"
241
+ },
242
+ "store": "SynapseWeb/kh2015/oblique"
243
+ }
244
+ },
245
+ {
246
+ "pad": {
247
+ "key": "RAW",
248
+ "size": null
249
+ }
250
+ },
251
+ {
252
+ "pad": {
253
+ "key": "LABELS",
254
+ "size": {
255
+ "coordinate": {
256
+ "_positional": [
257
+ 450,
258
+ 290,
259
+ 290
260
+ ]
261
+ }
262
+ }
263
+ }
264
+ },
265
+ {
266
+ "pad": {
267
+ "key": "LABELS_MASK",
268
+ "size": {
269
+ "coordinate": {
270
+ "_positional": [
271
+ 450,
272
+ 290,
273
+ 290
274
+ ]
275
+ }
276
+ }
277
+ }
278
+ },
279
+ {
280
+ "random_location": {
281
+ "mask": "LABELS_MASK",
282
+ "min_masked": 0.1
283
+ }
284
+ }
285
+ ],
286
+ [
287
+ {
288
+ "zarr_source": {
289
+ "array_specs": {
290
+ "_labels": {
291
+ "array_spec": {
292
+ "interpolatable": false
293
+ }
294
+ },
295
+ "_labels_mask": {
296
+ "array_spec": {
297
+ "interpolatable": false
298
+ }
299
+ },
300
+ "_raw": {
301
+ "array_spec": {
302
+ "interpolatable": true
303
+ }
304
+ }
305
+ },
306
+ "datasets": {
307
+ "_labels": "labels/s0",
308
+ "_labels_mask": "labels_mask/s0",
309
+ "_raw": "raw/s0"
310
+ },
311
+ "store": "SynapseWeb/kh2015/spine"
312
+ }
313
+ },
314
+ {
315
+ "pad": {
316
+ "key": "RAW",
317
+ "size": null
318
+ }
319
+ },
320
+ {
321
+ "pad": {
322
+ "key": "LABELS",
323
+ "size": {
324
+ "coordinate": {
325
+ "_positional": [
326
+ 450,
327
+ 290,
328
+ 290
329
+ ]
330
+ }
331
+ }
332
+ }
333
+ },
334
+ {
335
+ "pad": {
336
+ "key": "LABELS_MASK",
337
+ "size": {
338
+ "coordinate": {
339
+ "_positional": [
340
+ 450,
341
+ 290,
342
+ 290
343
+ ]
344
+ }
345
+ }
346
+ }
347
+ },
348
+ {
349
+ "random_location": {
350
+ "mask": "LABELS_MASK",
351
+ "min_masked": 0.1
352
+ }
353
+ }
354
+ ]
355
+ ],
356
+ {
357
+ "random_provider": {}
358
+ }
359
+ ],
360
+ "target": [
361
+ {
362
+ "add_affinities": {
363
+ "affinities": "GT_AFFS",
364
+ "affinities_mask": "GT_AFFS_MASK",
365
+ "affinity_neighborhood": [
366
+ [
367
+ -1,
368
+ 0,
369
+ 0
370
+ ],
371
+ [
372
+ 0,
373
+ -1,
374
+ 0
375
+ ],
376
+ [
377
+ 0,
378
+ 0,
379
+ -1
380
+ ]
381
+ ],
382
+ "labels": "LABELS",
383
+ "labels_mask": "LABELS_MASK"
384
+ }
385
+ },
386
+ {
387
+ "balance_labels": {
388
+ "labels": "GT_AFFS",
389
+ "mask": "GT_AFFS_MASK",
390
+ "scales": "AFFS_WEIGHTS"
391
+ }
392
+ },
393
+ {
394
+ "intensity_scale_shift": {
395
+ "array": "RAW",
396
+ "scale": 2,
397
+ "shift": -1
398
+ }
399
+ },
400
+ {
401
+ "unsqueeze": {
402
+ "arrays": [
403
+ "RAW"
404
+ ],
405
+ "axis": 0
406
+ }
407
+ }
408
+ ]
409
+ },
410
+ "predict": {
411
+ "model": {
412
+ "path": "out/latest_model3.pt"
413
+ },
414
+ "output": [
415
+ {
416
+ "dataset": "pred_affs",
417
+ "path": "out_zarr.zarr",
418
+ "stacked": true
419
+ }
420
+ ],
421
+ "source": [
422
+ {
423
+ "dataset": "raw/s0",
424
+ "path": "SynapseWeb/kh2015/oblique"
425
+ }
426
+ ]
427
+ },
428
+ "training": {
429
+ "batch_outputs": [
430
+ "raw",
431
+ "labels",
432
+ "gt_affs",
433
+ "affs_weights",
434
+ "gt_lsds",
435
+ "lsds_weights"
436
+ ],
437
+ "logging": {
438
+ "wandb": false
439
+ },
440
+ "loss": {
441
+ "_inputs": [
442
+ "lsds",
443
+ "gt_lsds",
444
+ "lsds_weights",
445
+ "affs",
446
+ "gt_affs",
447
+ "affs_weights"
448
+ ],
449
+ "weighted_m_s_e_loss_double": {}
450
+ },
451
+ "model_inputs": [
452
+ "raw"
453
+ ],
454
+ "model_outputs": [
455
+ "lsds",
456
+ "affs"
457
+ ],
458
+ "multi_gpu": true,
459
+ "save_every": 1000,
460
+ "train_dataloader": {
461
+ "batch_size": 2,
462
+ "input_image_shape": [
463
+ 36,
464
+ 212,
465
+ 212
466
+ ],
467
+ "num_workers": 4,
468
+ "output_image_shape": [
469
+ 12,
470
+ 120,
471
+ 120
472
+ ],
473
+ "parallel": true,
474
+ "precache_per_worker": 4
475
+ }
476
+ }
477
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f60e6dd2f9be6795b4c284a4a2a3813bfb0ab3048710935565a5f7d321ecc01f
3
+ size 180239916