forked from ildoonet/data-science-bowl-2018
-
Notifications
You must be signed in to change notification settings - Fork 0
/
commons.py
88 lines (84 loc) · 6.68 KB
/
commons.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
def chunker(seq, size):
return (seq[pos:pos + size] for pos in range(0, len(seq), size))
ensemble_models = {
'stage1_test': {
'rcnn': [
],
'unet': [
"/data/public/rw/kaggle-data-science-bowl/submissions/lb525_ensemble_s80px(0.5-2.0)xflip/submission_lb525.pkl"
]
},
'stage1_unet': {
'rcnn': [
"/data/public/rw/datasets/dsb2018/pickles/pickled_mask_info_full.pkl"
],
'unet': [
"/data/public/rw/kaggle-data-science-bowl/submissions/lb525_ensemble_s80px(0.5-2.0)xflip/submission_lb525.pkl",
"/data/public/rw/kaggle-data-science-bowl/submissions/stage1_folds7/unetv1_fold1_thick_unet_lr=0.00010000_epoch=600_bs=16/submission.pkl",
"/data/public/rw/kaggle-data-science-bowl/submissions/stage1_folds7/unetv1_fold2_thick_unet_lr=0.00010000_epoch=600_bs=16/submission.pkl",
"/data/public/rw/kaggle-data-science-bowl/submissions/stage1_folds7/unetv1_fold3_thick_unet_lr=0.00010000_epoch=600_bs=16/submission.pkl",
"/data/public/rw/kaggle-data-science-bowl/submissions/stage1_folds7/unetv1_fold4_thick_unet_lr=0.00010000_epoch=600_bs=16/submission.pkl",
"/data/public/rw/kaggle-data-science-bowl/submissions/stage1_folds7/unetv1_fold5_thick_unet_lr=0.00010000_epoch=600_bs=16/submission.pkl",
"/data/public/rw/kaggle-data-science-bowl/submissions/stage1_folds7/unetv1_fold6_thick_unet_lr=0.00010000_epoch=600_bs=16/submission.pkl",
"/data/public/rw/kaggle-data-science-bowl/submissions/stage1_folds7/unetv1_fold7_thick_unet_lr=0.00010000_epoch=600_bs=16/submission.pkl",
]
},
'stage2_unetv1': {
'rcnn': [
"/data/public/rw/datasets/dsb2018/pickles/pickled_mask_info_full_stage2.pkl"
],
'unet': [
"/data/public/rw/kaggle-data-science-bowl/submissions/stage2_w/unetv1_best496_unet_lr=0.00010000_epoch=600_bs=16/submission.pkl",
"/data/public/rw/kaggle-data-science-bowl/submissions/stage2_w/unetv1_fold1_ext2thick_unet_lr=0.00010000_epoch=600_bs=16/submission.pkl",
"/data/public/rw/kaggle-data-science-bowl/submissions/stage2_w/unetv1_fold2_ext2thick_unet_lr=0.00010000_epoch=600_bs=16/submission.pkl",
"/data/public/rw/kaggle-data-science-bowl/submissions/stage2_w/unetv1_fold3_ext2thick_unet_lr=0.00010000_epoch=600_bs=16/submission.pkl",
"/data/public/rw/kaggle-data-science-bowl/submissions/stage2_w/unetv1_fold4_ext2thick_unet_lr=0.00010000_epoch=600_bs=16/submission.pkl",
"/data/public/rw/kaggle-data-science-bowl/submissions/stage2_w/unetv1_fold5_ext2thick_unet_lr=0.00010000_epoch=600_bs=16/submission.pkl",
"/data/public/rw/kaggle-data-science-bowl/submissions/stage2_w/unetv1_fold6_ext2thick_unet_lr=0.00010000_epoch=600_bs=16/submission.pkl",
"/data/public/rw/kaggle-data-science-bowl/submissions/stage2_w/unetv1_fold7_ext2thick_unet_lr=0.00010000_epoch=600_bs=16/submission.pkl",
]
},
'stage2_unetv1_norcnn': {
'rcnn': [
],
'unet': [
"/data/public/rw/kaggle-data-science-bowl/submissions/stage2_w/unetv1_best496_unet_lr=0.00010000_epoch=600_bs=16/submission.pkl",
"/data/public/rw/kaggle-data-science-bowl/submissions/stage2_w/unetv1_fold1_ext2thick_unet_lr=0.00010000_epoch=600_bs=16/submission.pkl",
"/data/public/rw/kaggle-data-science-bowl/submissions/stage2_w/unetv1_fold2_ext2thick_unet_lr=0.00010000_epoch=600_bs=16/submission.pkl",
"/data/public/rw/kaggle-data-science-bowl/submissions/stage2_w/unetv1_fold3_ext2thick_unet_lr=0.00010000_epoch=600_bs=16/submission.pkl",
"/data/public/rw/kaggle-data-science-bowl/submissions/stage2_w/unetv1_fold4_ext2thick_unet_lr=0.00010000_epoch=600_bs=16/submission.pkl",
"/data/public/rw/kaggle-data-science-bowl/submissions/stage2_w/unetv1_fold5_ext2thick_unet_lr=0.00010000_epoch=600_bs=16/submission.pkl",
"/data/public/rw/kaggle-data-science-bowl/submissions/stage2_w/unetv1_fold6_ext2thick_unet_lr=0.00010000_epoch=600_bs=16/submission.pkl",
"/data/public/rw/kaggle-data-science-bowl/submissions/stage2_w/unetv1_fold7_ext2thick_unet_lr=0.00010000_epoch=600_bs=16/submission.pkl",
]
},
'stage2_unetv2': {
'rcnn': [
"/data/public/rw/datasets/dsb2018/pickles/pickled_mask_info_full_stage2.pkl"
],
'unet': [
"/data/public/rw/kaggle-data-science-bowl/submissions/stage2_w/unetv2_best539_unet_lr=0.00010000_epoch=600_bs=16/submission.pkl",
"/data/public/rw/kaggle-data-science-bowl/submissions/stage2_w/unetv2_fold1_unet_lr=0.00010000_epoch=600_bs=16/submission.pkl",
"/data/public/rw/kaggle-data-science-bowl/submissions/stage2_w/unetv2_fold2_unet_lr=0.00010000_epoch=600_bs=16/submission.pkl",
"/data/public/rw/kaggle-data-science-bowl/submissions/stage2_w/unetv2_fold3_unet_lr=0.00010000_epoch=600_bs=16/submission.pkl",
"/data/public/rw/kaggle-data-science-bowl/submissions/stage2_w/unetv2_fold4_unet_lr=0.00010000_epoch=600_bs=16/submission.pkl",
"/data/public/rw/kaggle-data-science-bowl/submissions/stage2_w/unetv2_fold5_unet_lr=0.00010000_epoch=600_bs=16/submission.pkl",
"/data/public/rw/kaggle-data-science-bowl/submissions/stage2_w/unetv2_fold6_unet_lr=0.00010000_epoch=600_bs=16/submission.pkl",
"/data/public/rw/kaggle-data-science-bowl/submissions/stage2_w/unetv2_fold7_unet_lr=0.00010000_epoch=600_bs=16/submission.pkl",
]
},
'stage2_unetv2_norcnn': {
'rcnn': [
],
'unet': [
"/data/public/rw/kaggle-data-science-bowl/submissions/stage2_w/unetv2_best539_unet_lr=0.00010000_epoch=600_bs=16/submission.pkl",
"/data/public/rw/kaggle-data-science-bowl/submissions/stage2_w/unetv2_fold1_unet_lr=0.00010000_epoch=600_bs=16/submission.pkl",
"/data/public/rw/kaggle-data-science-bowl/submissions/stage2_w/unetv2_fold2_unet_lr=0.00010000_epoch=600_bs=16/submission.pkl",
"/data/public/rw/kaggle-data-science-bowl/submissions/stage2_w/unetv2_fold3_unet_lr=0.00010000_epoch=600_bs=16/submission.pkl",
"/data/public/rw/kaggle-data-science-bowl/submissions/stage2_w/unetv2_fold4_unet_lr=0.00010000_epoch=600_bs=16/submission.pkl",
"/data/public/rw/kaggle-data-science-bowl/submissions/stage2_w/unetv2_fold5_unet_lr=0.00010000_epoch=600_bs=16/submission.pkl",
"/data/public/rw/kaggle-data-science-bowl/submissions/stage2_w/unetv2_fold6_unet_lr=0.00010000_epoch=600_bs=16/submission.pkl",
"/data/public/rw/kaggle-data-science-bowl/submissions/stage2_w/unetv2_fold7_unet_lr=0.00010000_epoch=600_bs=16/submission.pkl",
]
},
}