Skip to content

Commit

Permalink
other resnet_config + dataset_2D crops to input size
Browse files Browse the repository at this point in the history
  • Loading branch information
Jeroen committed Jun 21, 2016
1 parent 32c9e62 commit 8f3a399
Show file tree
Hide file tree
Showing 2 changed files with 60 additions and 0 deletions.
53 changes: 53 additions & 0 deletions config/jeroen_resnet.ini
Original file line number Diff line number Diff line change
@@ -0,0 +1,53 @@
[info]
experiment: ResNet default settings
name: resnet

[dataset]
pixels: 64
n_classes: 2
channels: 1
subset: None
filenames_train: ./../../data/cadV2_0.5mm_64x64_xy_xz_yz/subset[0-7]/*/*.pkl.gz
filenames_validation:./../../data/cadV2_0.5mm_64x64_xy_xz_yz/subset[8]/*/*.pkl.gz
data_folder: None ;not used

[network]
architecture: resnet
input_size: 48
depth: 1
branching_factor: 8
batch_normalization: False
batch_normalization_alpha: 0.3
dropout: 0.25
spatial_dropout: 0.5
gaussian_noise: 0.05

[updates]
optimization: -
learning_rate: 0 ;not used, see schedule
momentum: 0.9
l2_lambda: 0.00005
batch_size_train: 128
batch_size_validation: 256
n_epochs: 100

[preprocessing] ;Not used
random_crop: 0 ;no cropping
erode_segmentation: 11

[normalization]
zero_center: True
mean_pixel: 0.2606236106512

[augmentation]
augment: True
flip: True
zoom: 0.08 ;Only if OpenCV2 is available
rotation: 20
translation: 3


[misc]
multiprocess_load_augmentation: False
save_every_n_epoch: 2
n_workers_load_augmentation: 6
7 changes: 7 additions & 0 deletions src/deep/dataset_2D.py
Original file line number Diff line number Diff line change
Expand Up @@ -17,6 +17,13 @@ def load_images (image_paths, deterministic=False):
x,y,z = xy_xz_yz
if P.AUGMENT and not deterministic:
x,y,z = augment.augment(xy_xz_yz)

offset = (len(x) - P.INPUT_SIZE) / 2

x = x[offset:-offset,offset:-offset]
y = y[offset:-offset,offset:-offset]
z = z[offset:-offset,offset:-offset]

X.append([x])
X.append([y])
X.append([z])
Expand Down

0 comments on commit 8f3a399

Please sign in to comment.