Upload 6 files
Browse files- 2024-04-24T011302_CellViT-unireplknet-fold1-final/checkpoints/latest_checkpoint.pth +3 -0
- 2024-04-24T011302_CellViT-unireplknet-fold1-final/checkpoints/model_best.pth +3 -0
- 2024-04-24T011302_CellViT-unireplknet-fold1-final/config.yaml +156 -0
- 2024-04-24T013604_CellViT-unireplknet-fold1-final/checkpoints/latest_checkpoint.pth +3 -0
- 2024-04-24T013604_CellViT-unireplknet-fold1-final/checkpoints/model_best.pth +3 -0
- 2024-04-24T013604_CellViT-unireplknet-fold1-final/config.yaml +156 -0
2024-04-24T011302_CellViT-unireplknet-fold1-final/checkpoints/latest_checkpoint.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:03626e7d3db8e6a13971db2fde5946e70523f33ac604089eff2ff7801f678fec
|
3 |
+
size 1515916420
|
2024-04-24T011302_CellViT-unireplknet-fold1-final/checkpoints/model_best.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:2fd6ae772f870a349351f25c531a7718e32070c121f4a27229f90a46259400b2
|
3 |
+
size 1515875610
|
2024-04-24T011302_CellViT-unireplknet-fold1-final/config.yaml
ADDED
@@ -0,0 +1,156 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
logging:
|
2 |
+
log_dir: /data3/ziweicui/cell-unreplknet-small/2024-04-24T011302_CellViT-unireplknet-fold1
|
3 |
+
mode: online
|
4 |
+
project: Cell-Segmentation
|
5 |
+
notes: CellViT-256
|
6 |
+
log_comment: CellViT-unireplknet-fold1
|
7 |
+
tags:
|
8 |
+
- Fold-1
|
9 |
+
- ViT256
|
10 |
+
wandb_dir: /home/ziweicui/cell_unireplknet/results
|
11 |
+
level: Debug
|
12 |
+
group: CellViT256
|
13 |
+
run_id: przm46zv
|
14 |
+
wandb_file: przm46zv
|
15 |
+
random_seed: 19
|
16 |
+
gpu: 0
|
17 |
+
data:
|
18 |
+
dataset: PanNuke
|
19 |
+
dataset_path: /data3/ziweicui/PanNuke/cellvit-png
|
20 |
+
train_folds:
|
21 |
+
- 2
|
22 |
+
val_folds:
|
23 |
+
- 1
|
24 |
+
test_folds:
|
25 |
+
- 0
|
26 |
+
num_nuclei_classes: 6
|
27 |
+
num_tissue_classes: 19
|
28 |
+
model:
|
29 |
+
backbone: default
|
30 |
+
pretrained_encoder: /data3/ziweicui/cell-unreplknet-small/unireplknet_s_in1k_224_acc83.91.pth
|
31 |
+
shared_skip_connections: true
|
32 |
+
loss:
|
33 |
+
nuclei_binary_map:
|
34 |
+
focaltverskyloss:
|
35 |
+
loss_fn: FocalTverskyLoss
|
36 |
+
weight: 1
|
37 |
+
dice:
|
38 |
+
loss_fn: dice_loss
|
39 |
+
weight: 1
|
40 |
+
hv_map:
|
41 |
+
mse:
|
42 |
+
loss_fn: mse_loss_maps
|
43 |
+
weight: 2.5
|
44 |
+
msge:
|
45 |
+
loss_fn: msge_loss_maps
|
46 |
+
weight: 8
|
47 |
+
nuclei_type_map:
|
48 |
+
bce:
|
49 |
+
loss_fn: xentropy_loss
|
50 |
+
weight: 0.5
|
51 |
+
dice:
|
52 |
+
loss_fn: dice_loss
|
53 |
+
weight: 0.2
|
54 |
+
mcfocaltverskyloss:
|
55 |
+
loss_fn: MCFocalTverskyLoss
|
56 |
+
weight: 0.5
|
57 |
+
args:
|
58 |
+
num_classes: 6
|
59 |
+
tissue_types:
|
60 |
+
ce:
|
61 |
+
loss_fn: CrossEntropyLoss
|
62 |
+
weight: 0.1
|
63 |
+
training:
|
64 |
+
drop_rate: 0
|
65 |
+
attn_drop_rate: 0.1
|
66 |
+
drop_path_rate: 0.1
|
67 |
+
batch_size: 16
|
68 |
+
world_size: 2
|
69 |
+
rank: 0
|
70 |
+
epochs: 100
|
71 |
+
optimizer: AdamW
|
72 |
+
early_stopping_patience: 70
|
73 |
+
scheduler:
|
74 |
+
scheduler_type: cosine
|
75 |
+
hyperparameters:
|
76 |
+
eta_min: 1e-5
|
77 |
+
optimizer_hyperparameter:
|
78 |
+
betas:
|
79 |
+
- 0.85
|
80 |
+
- 0.95
|
81 |
+
lr: 0.001
|
82 |
+
weight_decay: 0.1
|
83 |
+
opt_eps: 8.0e-08
|
84 |
+
layer_decay: 0.999999
|
85 |
+
unfreeze_epoch: 0
|
86 |
+
sampling_gamma: 0.85
|
87 |
+
sampling_strategy: cell+tissue
|
88 |
+
mixed_precision: false
|
89 |
+
transformations:
|
90 |
+
randomrotate90:
|
91 |
+
p: 0.5
|
92 |
+
horizontalflip:
|
93 |
+
p: 0.5
|
94 |
+
verticalflip:
|
95 |
+
p: 0.5
|
96 |
+
downscale:
|
97 |
+
p: 0.15
|
98 |
+
scale: 0.5
|
99 |
+
blur:
|
100 |
+
p: 0.2
|
101 |
+
blur_limit: 10
|
102 |
+
gaussnoise:
|
103 |
+
p: 0.25
|
104 |
+
var_limit: 50
|
105 |
+
colorjitter:
|
106 |
+
p: 0.2
|
107 |
+
scale_setting: 0.25
|
108 |
+
scale_color: 0.1
|
109 |
+
superpixels:
|
110 |
+
p: 0.1
|
111 |
+
zoomblur:
|
112 |
+
p: 0.1
|
113 |
+
randomsizedcrop:
|
114 |
+
p: 0.1
|
115 |
+
elastictransform:
|
116 |
+
p: 0.2
|
117 |
+
normalize:
|
118 |
+
mean:
|
119 |
+
- 0.5
|
120 |
+
- 0.5
|
121 |
+
- 0.5
|
122 |
+
std:
|
123 |
+
- 0.5
|
124 |
+
- 0.5
|
125 |
+
- 0.5
|
126 |
+
eval_checkpoint: latest_checkpoint.pth
|
127 |
+
dataset_config:
|
128 |
+
tissue_types:
|
129 |
+
Adrenal_gland: 0
|
130 |
+
Bile-duct: 1
|
131 |
+
Bladder: 2
|
132 |
+
Breast: 3
|
133 |
+
Cervix: 4
|
134 |
+
Colon: 5
|
135 |
+
Esophagus: 6
|
136 |
+
HeadNeck: 7
|
137 |
+
Kidney: 8
|
138 |
+
Liver: 9
|
139 |
+
Lung: 10
|
140 |
+
Ovarian: 11
|
141 |
+
Pancreatic: 12
|
142 |
+
Prostate: 13
|
143 |
+
Skin: 14
|
144 |
+
Stomach: 15
|
145 |
+
Testis: 16
|
146 |
+
Thyroid: 17
|
147 |
+
Uterus: 18
|
148 |
+
nuclei_types:
|
149 |
+
Background: 0
|
150 |
+
Neoplastic: 1
|
151 |
+
Inflammatory: 2
|
152 |
+
Connective: 3
|
153 |
+
Dead: 4
|
154 |
+
Epithelial: 5
|
155 |
+
run_sweep: false
|
156 |
+
agent: null
|
2024-04-24T013604_CellViT-unireplknet-fold1-final/checkpoints/latest_checkpoint.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:7744227a4ebae5810e9f8b0e7e24fa0591761442864b1925bd40f1cc5c3997cb
|
3 |
+
size 1515916420
|
2024-04-24T013604_CellViT-unireplknet-fold1-final/checkpoints/model_best.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:1e38f4fd67407d420946ed1a401953c14b3d3a055aebe0918d70387e637b0402
|
3 |
+
size 1515875610
|
2024-04-24T013604_CellViT-unireplknet-fold1-final/config.yaml
ADDED
@@ -0,0 +1,156 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
logging:
|
2 |
+
log_dir: /data3/ziweicui/cell-unreplknet-small/2024-04-24T013604_CellViT-unireplknet-fold1
|
3 |
+
mode: online
|
4 |
+
project: Cell-Segmentation
|
5 |
+
notes: CellViT-256
|
6 |
+
log_comment: CellViT-unireplknet-fold1
|
7 |
+
tags:
|
8 |
+
- Fold-1
|
9 |
+
- ViT256
|
10 |
+
wandb_dir: /home/ziweicui/cell_unireplknet/results
|
11 |
+
level: Debug
|
12 |
+
group: CellViT256
|
13 |
+
run_id: xii8r2nj
|
14 |
+
wandb_file: xii8r2nj
|
15 |
+
random_seed: 19
|
16 |
+
gpu: 0
|
17 |
+
data:
|
18 |
+
dataset: PanNuke
|
19 |
+
dataset_path: /data3/ziweicui/PanNuke/cellvit-png
|
20 |
+
train_folds:
|
21 |
+
- 2
|
22 |
+
val_folds:
|
23 |
+
- 1
|
24 |
+
test_folds:
|
25 |
+
- 0
|
26 |
+
num_nuclei_classes: 6
|
27 |
+
num_tissue_classes: 19
|
28 |
+
model:
|
29 |
+
backbone: default
|
30 |
+
pretrained_encoder: /data3/ziweicui/cell-unreplknet-small/unireplknet_s_in1k_224_acc83.91.pth
|
31 |
+
shared_skip_connections: true
|
32 |
+
loss:
|
33 |
+
nuclei_binary_map:
|
34 |
+
focaltverskyloss:
|
35 |
+
loss_fn: FocalTverskyLoss
|
36 |
+
weight: 1
|
37 |
+
dice:
|
38 |
+
loss_fn: dice_loss
|
39 |
+
weight: 1
|
40 |
+
hv_map:
|
41 |
+
mse:
|
42 |
+
loss_fn: mse_loss_maps
|
43 |
+
weight: 2.5
|
44 |
+
msge:
|
45 |
+
loss_fn: msge_loss_maps
|
46 |
+
weight: 8
|
47 |
+
nuclei_type_map:
|
48 |
+
bce:
|
49 |
+
loss_fn: xentropy_loss
|
50 |
+
weight: 0.5
|
51 |
+
dice:
|
52 |
+
loss_fn: dice_loss
|
53 |
+
weight: 0.2
|
54 |
+
mcfocaltverskyloss:
|
55 |
+
loss_fn: MCFocalTverskyLoss
|
56 |
+
weight: 0.5
|
57 |
+
args:
|
58 |
+
num_classes: 6
|
59 |
+
tissue_types:
|
60 |
+
ce:
|
61 |
+
loss_fn: CrossEntropyLoss
|
62 |
+
weight: 0.1
|
63 |
+
training:
|
64 |
+
drop_rate: 0
|
65 |
+
attn_drop_rate: 0.1
|
66 |
+
drop_path_rate: 0.1
|
67 |
+
batch_size: 16
|
68 |
+
world_size: 2
|
69 |
+
rank: 0
|
70 |
+
epochs: 100
|
71 |
+
optimizer: AdamW
|
72 |
+
early_stopping_patience: 70
|
73 |
+
scheduler:
|
74 |
+
scheduler_type: cosine
|
75 |
+
hyperparameters:
|
76 |
+
eta_min: 1e-5
|
77 |
+
optimizer_hyperparameter:
|
78 |
+
betas:
|
79 |
+
- 0.85
|
80 |
+
- 0.95
|
81 |
+
lr: 0.001
|
82 |
+
weight_decay: 0.1
|
83 |
+
opt_eps: 8.0e-08
|
84 |
+
layer_decay: 0.999999
|
85 |
+
unfreeze_epoch: 0
|
86 |
+
sampling_gamma: 0.85
|
87 |
+
sampling_strategy: cell+tissue
|
88 |
+
mixed_precision: false
|
89 |
+
transformations:
|
90 |
+
randomrotate90:
|
91 |
+
p: 0.5
|
92 |
+
horizontalflip:
|
93 |
+
p: 0.5
|
94 |
+
verticalflip:
|
95 |
+
p: 0.5
|
96 |
+
downscale:
|
97 |
+
p: 0.15
|
98 |
+
scale: 0.5
|
99 |
+
blur:
|
100 |
+
p: 0.2
|
101 |
+
blur_limit: 10
|
102 |
+
gaussnoise:
|
103 |
+
p: 0.25
|
104 |
+
var_limit: 50
|
105 |
+
colorjitter:
|
106 |
+
p: 0.2
|
107 |
+
scale_setting: 0.25
|
108 |
+
scale_color: 0.1
|
109 |
+
superpixels:
|
110 |
+
p: 0.1
|
111 |
+
zoomblur:
|
112 |
+
p: 0.1
|
113 |
+
randomsizedcrop:
|
114 |
+
p: 0.1
|
115 |
+
elastictransform:
|
116 |
+
p: 0.2
|
117 |
+
normalize:
|
118 |
+
mean:
|
119 |
+
- 0.5
|
120 |
+
- 0.5
|
121 |
+
- 0.5
|
122 |
+
std:
|
123 |
+
- 0.5
|
124 |
+
- 0.5
|
125 |
+
- 0.5
|
126 |
+
eval_checkpoint: latest_checkpoint.pth
|
127 |
+
dataset_config:
|
128 |
+
tissue_types:
|
129 |
+
Adrenal_gland: 0
|
130 |
+
Bile-duct: 1
|
131 |
+
Bladder: 2
|
132 |
+
Breast: 3
|
133 |
+
Cervix: 4
|
134 |
+
Colon: 5
|
135 |
+
Esophagus: 6
|
136 |
+
HeadNeck: 7
|
137 |
+
Kidney: 8
|
138 |
+
Liver: 9
|
139 |
+
Lung: 10
|
140 |
+
Ovarian: 11
|
141 |
+
Pancreatic: 12
|
142 |
+
Prostate: 13
|
143 |
+
Skin: 14
|
144 |
+
Stomach: 15
|
145 |
+
Testis: 16
|
146 |
+
Thyroid: 17
|
147 |
+
Uterus: 18
|
148 |
+
nuclei_types:
|
149 |
+
Background: 0
|
150 |
+
Neoplastic: 1
|
151 |
+
Inflammatory: 2
|
152 |
+
Connective: 3
|
153 |
+
Dead: 4
|
154 |
+
Epithelial: 5
|
155 |
+
run_sweep: false
|
156 |
+
agent: null
|